Home
last modified time | relevance | path

Searched refs:TensorHandleFromInterface (Results 1 – 13 of 13) sorted by relevance

/external/tensorflow/tensorflow/c/eager/
Dc_api_remote_test_util.cc165 tensorflow::TensorHandleFromInterface(tensorflow::unwrap(h1_task2)); in TestRemoteExecuteSilentCopies()
178 tensorflow::TensorHandleFromInterface(tensorflow::unwrap(h1_task2)); in TestRemoteExecuteSilentCopies()
Dc_api_debug.cc57 TensorHandleFromInterface(tensorflow::unwrap(h)); in TFE_TensorHandleTensorDebugInfo()
Ddlpack.cc53 tensorflow::TensorHandleFromInterface(tensorflow::unwrap(h)); in GetTensorFromHandle()
Dc_api.cc388 tensorflow::TensorHandleFromInterface(unwrapped_handle); in TFE_TensorHandleDevicePointer()
602 tensorflow::TensorHandleFromInterface(tensorflow::unwrap(h)); in TFE_TensorHandleDeviceMemorySize()
Dc_api_experimental.cc687 tensorflow::TensorHandleFromInterface(unwrapped_handle)); in TFE_CreatePackedTensorHandle()
Dc_api_test.cc422 tensorflow::TensorHandleFromInterface(tensorflow::unwrap(hcpu)); in TensorHandleSilentCopy()
424 tensorflow::TensorHandleFromInterface(tensorflow::unwrap(hgpu)); in TensorHandleSilentCopy()
/external/tensorflow/tensorflow/core/common_runtime/eager/
Dcore.cc118 TensorHandle* input = TensorHandleFromInterface(handle); in CopyTensorHandleToDevice()
Dtensor_handle.h403 inline TensorHandle* TensorHandleFromInterface(T* handle) { in TensorHandleFromInterface() function
/external/tensorflow/tensorflow/core/runtime_fallback/runtime/gpu/
Dconversion_function.cc131 OwnedTensorHandle{TensorHandleFromInterface(tf_tensor_handle->Copy())}; in ConvertRuntimeFallbackTensorToDenseGpuTensor()
/external/tensorflow/tensorflow/core/tfrt/eager/
Dc_api_tfrt.h425 inline TensorHandleInterface* TensorHandleFromInterface(T* handle) { in TensorHandleFromInterface() function
Dc_api_tfrt.cc881 TensorHandle th = tfrt::tf::TensorHandleFromInterface(handle)->Handle(); in TFTensorHandleFromInterface()
1042 TensorHandle src_th = tfrt::tf::TensorHandleFromInterface(handle)->Handle(); in CopyTensorHandleToDevice()
/external/tensorflow/tensorflow/core/runtime_fallback/runtime/
Druntime_fallback_kernels.cc636 TensorHandleFromInterface(result_tensor_handles[i])}, in RuntimeFallbackExecute()
790 OwnedTensorHandle owned_th{TensorHandleFromInterface(retvals[i])}; in RuntimeFallbackKernel()
/external/tensorflow/tensorflow/python/
Dtfe_wrapper.cc443 input_handles.push_back(TensorHandleFromInterface(abstract_tensor_handle)); in InputTFE_OutputTensorHandles()