Home
last modified time | relevance | path

Searched refs:ExtractPackedHandle (Results 1 – 8 of 8) sorted by relevance

/external/tensorflow/tensorflow/core/common_runtime/eager/
Dexecute_node.cc44 TF_RETURN_IF_ERROR(packed_handle->ExtractPackedHandle(i, &h)); in InitPackedHandle()
107 op_inputs[index.index]->ExtractPackedHandle(index.sub_index, &h)); in Init()
Dtensor_handle.h263 Status ExtractPackedHandle(const int index, TensorHandle** handle) const;
367 Status ExtractPackedHandle(const int index, TensorHandle** handle) const;
Dtensor_handle_test.cc202 TF_ASSERT_OK(packed_handle->ExtractPackedHandle(i, &h)); in TEST_F()
245 TF_ASSERT_OK(packed_handle->ExtractPackedHandle(0, &h0)); in TEST_F()
Dtensor_handle.cc160 Status TensorHandle::PackedTensorHandleData::ExtractPackedHandle( in ExtractPackedHandle() function in tensorflow::TensorHandle::PackedTensorHandleData
206 Status TensorHandle::ExtractPackedHandle(const int index, in ExtractPackedHandle() function in tensorflow::TensorHandle
212 return absl::get<PackedTensorHandleData>(data_).ExtractPackedHandle(index, in ExtractPackedHandle()
Dexecute.cc207 TF_RETURN_IF_ERROR(handle->ExtractPackedHandle(j, &h)); in ValidateInputTypeAndPlacement()
771 TF_RETURN_IF_ERROR(handle->ExtractPackedHandle(i, &h)); in MaybePackInputTensor()
/external/tensorflow/tensorflow/c/experimental/saved_model/core/revived_types/
Dvariable.cc109 TF_RETURN_IF_ERROR(packed_handle->ExtractPackedHandle(i, &component)); in CreateUninitialized()
/external/tensorflow/tensorflow/core/distributed_runtime/eager/
Deager_service_impl_test.cc1172 TF_ASSERT_OK(packed_handle->ExtractPackedHandle(0, &handle0)); in TEST_F()
1185 TF_ASSERT_OK(packed_handle->ExtractPackedHandle(1, &handle1)); in TEST_F()
1193 TF_ASSERT_OK(packed_handle->ExtractPackedHandle(2, &handle2)); in TEST_F()
Dremote_copy_node.cc317 TF_RETURN_IF_ERROR(packed_handle->ExtractPackedHandle(i, &h)); in SerializePackedHandle()