aboutsummaryrefslogtreecommitdiffhomepage
path: root/tensorflow/compiler/xla/python/local_computation_builder.cc
diff options
context:
space:
mode:
Diffstat (limited to 'tensorflow/compiler/xla/python/local_computation_builder.cc')
-rw-r--r--tensorflow/compiler/xla/python/local_computation_builder.cc9
1 files changed, 4 insertions, 5 deletions
diff --git a/tensorflow/compiler/xla/python/local_computation_builder.cc b/tensorflow/compiler/xla/python/local_computation_builder.cc
index c133a20419..00e36c3c86 100644
--- a/tensorflow/compiler/xla/python/local_computation_builder.cc
+++ b/tensorflow/compiler/xla/python/local_computation_builder.cc
@@ -137,8 +137,7 @@ static StatusOr<ScopedShapedBuffer> ToBuffer(LocalClient* client,
/* static */
StatusOr<LocalShapedBuffer*> LocalShapedBuffer::FromLiteral(
- const Literal& argument,
- const tensorflow::gtl::optional<Shape>& shape_with_layout) {
+ const Literal& argument, const absl::optional<Shape>& shape_with_layout) {
LocalClient* client = GetOrCreateLocalClient();
StatusOr<ScopedShapedBuffer> buf = [&] {
if (shape_with_layout) {
@@ -163,7 +162,7 @@ CompiledLocalComputation::CompiledLocalComputation(
StatusOr<std::unique_ptr<Literal>> CompiledLocalComputation::Execute(
const std::vector<Literal>& arguments,
- const std::vector<tensorflow::gtl::optional<Shape>>& shapes_with_layout) {
+ const std::vector<absl::optional<Shape>>& shapes_with_layout) {
LocalClient* client = GetOrCreateLocalClient();
VLOG(1) << "Execution requested with " << GetReplicaCount() << " replicas.";
@@ -194,7 +193,7 @@ StatusOr<std::unique_ptr<Literal>> CompiledLocalComputation::Execute(
scoped_buffers.reserve(arguments.size());
for (int i = 0; i < arguments.size(); ++i) {
const Literal& argument = arguments[i];
- const tensorflow::gtl::optional<Shape>& shape_with_layout =
+ const absl::optional<Shape>& shape_with_layout =
shapes_with_layout[i];
StatusOr<ScopedShapedBuffer> pushed;
@@ -576,7 +575,7 @@ StatusOr<bool> LocalComputationBuilder::IsConstant(const LocalOp& operand) {
}
LocalOp LocalComputationBuilder::Sort(const LocalOp& operand, int64 dimension) {
- return xla::Sort(operand.op(), tensorflow::gtl::nullopt, dimension);
+ return xla::Sort(operand.op(), absl::nullopt, dimension);
}
LocalOp LocalComputationBuilder::SortKeyVal(const LocalOp& keys,