Replace LiteralUtil to Literal in compiler/plugin/executor

lukeiwanski · Jun 21, 2017 · a54d43f · a54d43f
1 parent 62fb49d
commit a54d43f
Show file tree

Hide file tree

Showing 2 changed files with 7 additions and 7 deletions.
diff --git a/tensorflow/compiler/plugin/executor/executable.cc b/tensorflow/compiler/plugin/executor/executable.cc
@@ -36,7 +36,7 @@ static se::DeviceMemoryBase AllocateSingleOutput(sep::ExecutorExecutor* executor
                                                  const Literal& literal) {
   int64 size(xla::ShapeUtil::ByteSizeOf(literal.shape()));
   void* buf = executor->Allocate(size);
-  const void* src = LiteralUtil::InternalData(literal);
+  const void* src = Literal::InternalData(literal);
   memcpy(buf, src, size);
   return se::DeviceMemoryBase(buf, size);
 }
@@ -86,11 +86,11 @@ StatusOr<se::DeviceMemoryBase> ExecutorExecutable::ExecuteOnStream(
   for (int64 p = 0; p < computation->num_parameters(); p++) {
     // Create the input literal for the parameter
     HloInstruction* param = computation->parameter_instruction(p);
-    arg_literals.emplace_back(LiteralUtil::CreateFromShape(param->shape()));
+    arg_literals.emplace_back(Literal::CreateFromShape(param->shape()));
     arg_literals_ptrs.push_back(arg_literals.back().get());
 
     // Copy in the data from the stream_executor buffers
-    void* buffer = LiteralUtil::MutableInternalData(arg_literals.back().get());
+    void* buffer = Literal::MutableInternalData(arg_literals.back().get());
     memcpy(buffer, arguments[p].opaque(),
            ShapeUtil::ByteSizeOf(param->shape()));
   }

diff --git a/tensorflow/compiler/plugin/executor/transfer_manager.cc b/tensorflow/compiler/plugin/executor/transfer_manager.cc
@@ -70,13 +70,13 @@ Status ExecutorTransferManager::TransferLiteralFromDevice(
   }
 
   *literal->mutable_shape() = device_shape;
-  LiteralUtil::Reserve(ShapeUtil::ElementsIn(device_shape), literal);
+  Literal::Reserve(ShapeUtil::ElementsIn(device_shape), literal);
   TF_RETURN_IF_ERROR(TransferBufferFromDevice(
       executor, source, ShapeUtil::ByteSizeOf(device_shape),
-      LiteralUtil::MutableInternalData(literal)));
+      Literal::MutableInternalData(literal)));
   if (!ShapeUtil::Equal(literal_shape, device_shape)) {
     literal->Swap(
-        LiteralUtil::Relayout(*literal, literal_shape.layout()).get());
+        Literal::Relayout(*literal, literal_shape.layout()).get());
   }
   TF_RET_CHECK(ShapeUtil::Equal(literal_shape, literal->shape()));
   return Status::OK();
@@ -134,7 +134,7 @@ Status ExecutorTransferManager::TransferLiteralToDevice(
   }
 
   return TransferBufferToDevice(executor, GetByteSizeRequirement(shape),
-                                LiteralUtil::InternalData(literal),
+                                Literal::InternalData(literal),
                                 destination);
 }