diff --git a/tensorflow/core/common_runtime/eager/kernel_and_device.cc b/tensorflow/core/common_runtime/eager/kernel_and_device.cc index bdb29c1035a..0c7f6d402f7 100644 --- a/tensorflow/core/common_runtime/eager/kernel_and_device.cc +++ b/tensorflow/core/common_runtime/eager/kernel_and_device.cc @@ -33,7 +33,6 @@ limitations under the License. #include "tensorflow/core/lib/core/errors.h" #include "tensorflow/core/lib/core/refcount.h" #include "tensorflow/core/lib/gtl/map_util.h" -#include "tensorflow/core/lib/gtl/stl_util.h" #include "tensorflow/core/lib/random/random.h" #include "tensorflow/core/platform/fingerprint.h" #include "tensorflow/core/platform/mutex.h" @@ -279,7 +278,7 @@ Status KernelAndDeviceOp::Run(ScopedStepContainer* step_container, params.op_kernel = kernel_.get(); params.resource_manager = device_->resource_manager(); params.input_alloc_attrs = &in_attrs; - params.output_attr_array = gtl::vector_as_array(&out_attrs); + params.output_attr_array = out_attrs.data(); params.function_library = flr_; params.slice_reader_cache = &slice_reader_cache_; params.rendezvous = rendez_; diff --git a/tensorflow/core/common_runtime/gpu/gpu_event_mgr_test.cc b/tensorflow/core/common_runtime/gpu/gpu_event_mgr_test.cc index 43ac015000b..1ed15d41f2f 100644 --- a/tensorflow/core/common_runtime/gpu/gpu_event_mgr_test.cc +++ b/tensorflow/core/common_runtime/gpu/gpu_event_mgr_test.cc @@ -28,7 +28,6 @@ limitations under the License. #include "tensorflow/core/graph/node_builder.h" #include "tensorflow/core/lib/core/notification.h" #include "tensorflow/core/lib/core/status_test_util.h" -#include "tensorflow/core/lib/gtl/stl_util.h" #include "tensorflow/core/platform/stream_executor.h" #include "tensorflow/core/platform/test.h" #include "tensorflow/core/platform/test_benchmark.h" @@ -417,7 +416,7 @@ class EMBenchmarkHelper { attr.set_on_host(on_host); attrs->push_back(attr); } - params->output_attr_array = gtl::vector_as_array(attrs); + params->output_attr_array = attrs->data(); params->forward_from_array = {}; } diff --git a/tensorflow/core/kernels/data/dataset_test_base.cc b/tensorflow/core/kernels/data/dataset_test_base.cc index 67b18724b28..aecdef33160 100644 --- a/tensorflow/core/kernels/data/dataset_test_base.cc +++ b/tensorflow/core/kernels/data/dataset_test_base.cc @@ -532,7 +532,7 @@ Status DatasetOpsTestBase::CreateOpKernelContext( attr.set_on_host(on_host); allocator_attrs_.emplace_back(attr); } - params_->output_attr_array = gtl::vector_as_array(&allocator_attrs_); + params_->output_attr_array = allocator_attrs_.data(); *context = absl::make_unique(params_.get()); return Status::OK(); diff --git a/tensorflow/core/kernels/ops_testutil.h b/tensorflow/core/kernels/ops_testutil.h index 77c072c0260..6baf463eb63 100644 --- a/tensorflow/core/kernels/ops_testutil.h +++ b/tensorflow/core/kernels/ops_testutil.h @@ -61,7 +61,7 @@ inline void SetOutputAttrs(OpKernelContext::Params* params, attr.set_on_host(on_host); attrs->push_back(attr); } - params->output_attr_array = gtl::vector_as_array(attrs); + params->output_attr_array = attrs->data(); } } // namespace test