Home
last modified time | relevance | path

Searched refs:TFE_TensorHandleCopySharingTensor (Results 1 – 5 of 5) sorted by relevance

/aosp_15_r20/external/tensorflow/tensorflow/c/eager/parallel_device/
H A Dparallel_device.cc99 components.emplace_back(TFE_TensorHandleCopySharingTensor( in ExecuteWithSpecialOps()
136 TFE_TensorHandleCopySharingTensor(t->tensor(i), status)); in ExecuteWithSpecialOps()
299 return TFE_TensorHandleCopySharingTensor(parallel_tensor->tensor(0), in CopyTensorFromParallelDevice()
/aosp_15_r20/external/tensorflow/tensorflow/c/eager/
H A Dc_api.h172 TF_CAPI_EXPORT extern TFE_TensorHandle* TFE_TensorHandleCopySharingTensor(
H A Dc_api.cc344 TF_CAPI_EXPORT extern TFE_TensorHandle* TFE_TensorHandleCopySharingTensor( in TFE_TensorHandleCopySharingTensor() function
H A Dc_api_test.cc1527 TFE_TensorHandleCopySharingTensor(h, status.get()); in TEST()
/aosp_15_r20/external/tensorflow/tensorflow/dtensor/cc/
H A Ddtensor_device.cc631 TFE_TensorHandleCopySharingTensor(t->get_tensor(output_index), status); in Unpack()
2010 return TFE_TensorHandleCopySharingTensor(typed_input->get_tensor(0), status); in CopyFromDTensorDevice()