/aosp_15_r20/external/tensorflow/tensorflow/java/src/main/native/ |
H A D | eager_operation_jni.cc | 64 TFE_TensorHandle* tensor_handle = requireTensorHandle(env, handle); in Java_org_tensorflow_EagerOperation_resolveTensorHandle() local 112 TFE_TensorHandle* tensor_handle = requireTensorHandle(env, handle); in Java_org_tensorflow_EagerOperation_dataType() local 120 TFE_TensorHandle* tensor_handle = requireTensorHandle(env, handle); in Java_org_tensorflow_EagerOperation_numDims() local 136 TFE_TensorHandle* tensor_handle = requireTensorHandle(env, handle); in Java_org_tensorflow_EagerOperation_dim() local
|
H A D | eager_operation_builder_jni.cc | 132 TFE_TensorHandle* tensor_handle = requireTensorHandle(env, input_handle); in Java_org_tensorflow_EagerOperationBuilder_addInput() local 259 jlong tensor_handle) { in Java_org_tensorflow_EagerOperationBuilder_setAttrTensor()
|
H A D | graph_operation_builder_jni.cc | 212 jlong tensor_handle) { in Java_org_tensorflow_GraphOperationBuilder_setAttrTensor()
|
/aosp_15_r20/external/tensorflow/tensorflow/core/runtime_fallback/runtime/ |
H A D | runtime_fallback_tensor.cc | 179 OwnedTensorHandle tensor_handle{ in MoveDHTToRuntimeFallbackTensor() local 193 OwnedTensorHandle tensor_handle{ in CopyRefDHTToRuntimeFallbackTensor() local 203 OwnedTensorHandle tensor_handle{ in CopySHTToRuntimeFallbackTensor() local
|
H A D | runtime_fallback_kernels.cc | 875 auto& tensor_handle = tensorhandle_args[i]->get<tfrt::TensorHandle>(); in CoreRTTensorHandleToFallbackTensorInternal() local 1132 OwnedTensorHandle tensor_handle{tensorflow::TensorHandle::CreateLocalHandle( in RuntimeFallbackExecuteOp() local 1220 OwnedTensorHandle tensor_handle, HostContext* host) { in ConvertTFTensorHandleToTFRTTensor()
|
/aosp_15_r20/external/tensorflow/tensorflow/core/distributed_runtime/eager/ |
H A D | eager_service_impl_test.cc | 445 tensorflow::TensorHandle* tensor_handle; in TEST_F() local 553 tensorflow::TensorHandle* tensor_handle; in TestFunction() local 646 tensorflow::TensorHandle* tensor_handle; in TestComponentFunction() local 817 tensorflow::TensorHandle* tensor_handle; in CheckOutputsAndClose() local 924 tensorflow::TensorHandle* tensor_handle; in TEST_F() local 1101 tensorflow::TensorHandle* tensor_handle; in TEST_F() local
|
H A D | eager_service_impl.cc | 723 TensorHandle* tensor_handle = TensorHandle::CreateLocalHandle( in SendTensor() local
|
/aosp_15_r20/external/ComputeLibrary/src/graph/detail/ |
H A D | CrossLayerMemoryManagerHelpers.cpp | 138 ITensorHandle *tensor_handle = input_edge->tensor()->handle()->parent_handle(); in get_transition_handles() local 151 ITensorHandle *tensor_handle = output_tensor->handle()->parent_handle(); in get_transition_handles() local
|
/aosp_15_r20/external/pytorch/torch/csrc/inductor/aoti_runtime/ |
H A D | model_container.h | 297 AtenTensorHandle tensor_handle; in update_constant_buffer() local 315 AtenTensorHandle tensor_handle = it->second; in update_constant_buffer() local
|
H A D | model.h | 229 AtenTensorHandle tensor_handle = nullptr; in load_constants() local
|
/aosp_15_r20/external/tensorflow/tensorflow/cc/experimental/base/public/ |
H A D | tensorhandle.h | 86 TFE_TensorHandle* tensor_handle = TFE_NewTensorHandleFromTensor( in FromTensor() local
|
/aosp_15_r20/external/tensorflow/tensorflow/core/runtime_fallback/conversion/ |
H A D | conversion.cc | 51 OwnedTensorHandle tensor_handle{tensorflow::TensorHandle::CreateLocalHandle( in ConvertKernelFallbackToRuntimeFallbackTensor() local
|
/aosp_15_r20/external/tensorflow/tensorflow/core/common_runtime/eager/ |
H A D | placement_utils.cc | 148 TensorHandle* tensor_handle = (*inputs)[i]; in MaybePinToResourceDevice() local
|
H A D | execute.cc | 322 TensorHandle* tensor_handle, Device** result) { in GetDeviceForInput()
|
/aosp_15_r20/external/tensorflow/tensorflow/core/runtime_fallback/runtime/gpu/ |
H A D | conversion_function.cc | 55 OwnedTensorHandle tensor_handle{tensorflow::TensorHandle::CreateLocalHandle( in CopyRefGpuTensorToRuntimeFallbackTensor() local
|
/aosp_15_r20/external/tensorflow/tensorflow/core/tfrt/utils/ |
H A D | tensor_util.cc | 175 const TensorHandle& tensor_handle) { in CreateTFTensorFromTensorHandle()
|
/aosp_15_r20/external/tensorflow/tensorflow/c/eager/ |
H A D | c_api_experimental_test.cc | 367 void Deleter(void* data, size_t unused, void* tensor_handle) { in Deleter()
|
H A D | c_api_test.cc | 1412 TFE_TensorHandle* tensor_handle = TFE_NewTensorHandle(tensor, status); in TEST() local 1471 TFE_TensorHandle* tensor_handle = TFE_NewTensorHandle(tensor, status); in TEST() local
|
/aosp_15_r20/external/ComputeLibrary/arm_compute/graph/backends/ |
H A D | FunctionHelpers.h | 69 ITensorHandle *tensor_handle = tensor->handle(); in get_backing_tensor() local
|