diff options
author | 2018-08-23 19:45:28 -0700 | |
---|---|---|
committer | 2018-08-23 19:49:55 -0700 | |
commit | 792ee96ea19492e59d09d809ce4a993e2e771e14 (patch) | |
tree | 38e19a15a64394b67cf33e320718d2e611c289a6 /tensorflow/compiler/jit | |
parent | 9d51d5914084f4cd65ea18c4da45dfb64f2945b6 (diff) |
Use absl::make_unique in tf2xla.
PiperOrigin-RevId: 210042392
Diffstat (limited to 'tensorflow/compiler/jit')
-rw-r--r-- | tensorflow/compiler/jit/BUILD | 1 | ||||
-rw-r--r-- | tensorflow/compiler/jit/resource_operation_safety_analysis.cc | 6 |
2 files changed, 5 insertions, 2 deletions
diff --git a/tensorflow/compiler/jit/BUILD b/tensorflow/compiler/jit/BUILD index 1c09ee8fb1..aa01b1fd52 100644 --- a/tensorflow/compiler/jit/BUILD +++ b/tensorflow/compiler/jit/BUILD @@ -320,6 +320,7 @@ cc_library( "//tensorflow/core:graph", "//tensorflow/core:lib", "//tensorflow/core:protos_all_cc", + "@com_google_absl//absl/memory", "@com_google_absl//absl/strings", ], ) diff --git a/tensorflow/compiler/jit/resource_operation_safety_analysis.cc b/tensorflow/compiler/jit/resource_operation_safety_analysis.cc index f18d988a5c..2479e57cde 100644 --- a/tensorflow/compiler/jit/resource_operation_safety_analysis.cc +++ b/tensorflow/compiler/jit/resource_operation_safety_analysis.cc @@ -82,6 +82,7 @@ limitations under the License. #include "tensorflow/compiler/jit/resource_operation_safety_analysis.h" +#include "absl/memory/memory.h" #include "absl/strings/str_join.h" #include "tensorflow/core/framework/node_def.pb.h" #include "tensorflow/core/graph/algorithm.h" @@ -300,7 +301,7 @@ class ResourceOpSet { void EnsureIsCopied() { if (storage_ == nullptr) { - storage_ = MakeUnique<Impl>(); + storage_ = absl::make_unique<Impl>(); for (ResourceOp op : *this) { storage_->insert(op); } @@ -349,7 +350,8 @@ Status ComputeIncompatibleResourceOperationPairs( return !edge.src()->IsNextIteration(); }); - auto resource_op_set_for_node = MakeUnique<ResourceOpSet[]>(g.num_node_ids()); + auto resource_op_set_for_node = + absl::make_unique<ResourceOpSet[]>(g.num_node_ids()); const bool vlog = VLOG_IS_ON(2); |