Home
last modified time | relevance | path

Searched defs:tensor_handle (Results 1 – 19 of 19) sorted by relevance

/aosp_15_r20/external/tensorflow/tensorflow/java/src/main/native/
H A Deager_operation_jni.cc64 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 Deager_operation_builder_jni.cc132 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 Dgraph_operation_builder_jni.cc212 jlong tensor_handle) { in Java_org_tensorflow_GraphOperationBuilder_setAttrTensor()
/aosp_15_r20/external/tensorflow/tensorflow/core/runtime_fallback/runtime/
H A Druntime_fallback_tensor.cc179 OwnedTensorHandle tensor_handle{ in MoveDHTToRuntimeFallbackTensor() local
193 OwnedTensorHandle tensor_handle{ in CopyRefDHTToRuntimeFallbackTensor() local
203 OwnedTensorHandle tensor_handle{ in CopySHTToRuntimeFallbackTensor() local
H A Druntime_fallback_kernels.cc875 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 Deager_service_impl_test.cc445 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 Deager_service_impl.cc723 TensorHandle* tensor_handle = TensorHandle::CreateLocalHandle( in SendTensor() local
/aosp_15_r20/external/ComputeLibrary/src/graph/detail/
H A DCrossLayerMemoryManagerHelpers.cpp138 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 Dmodel_container.h297 AtenTensorHandle tensor_handle; in update_constant_buffer() local
315 AtenTensorHandle tensor_handle = it->second; in update_constant_buffer() local
H A Dmodel.h229 AtenTensorHandle tensor_handle = nullptr; in load_constants() local
/aosp_15_r20/external/tensorflow/tensorflow/cc/experimental/base/public/
H A Dtensorhandle.h86 TFE_TensorHandle* tensor_handle = TFE_NewTensorHandleFromTensor( in FromTensor() local
/aosp_15_r20/external/tensorflow/tensorflow/core/runtime_fallback/conversion/
H A Dconversion.cc51 OwnedTensorHandle tensor_handle{tensorflow::TensorHandle::CreateLocalHandle( in ConvertKernelFallbackToRuntimeFallbackTensor() local
/aosp_15_r20/external/tensorflow/tensorflow/core/common_runtime/eager/
H A Dplacement_utils.cc148 TensorHandle* tensor_handle = (*inputs)[i]; in MaybePinToResourceDevice() local
H A Dexecute.cc322 TensorHandle* tensor_handle, Device** result) { in GetDeviceForInput()
/aosp_15_r20/external/tensorflow/tensorflow/core/runtime_fallback/runtime/gpu/
H A Dconversion_function.cc55 OwnedTensorHandle tensor_handle{tensorflow::TensorHandle::CreateLocalHandle( in CopyRefGpuTensorToRuntimeFallbackTensor() local
/aosp_15_r20/external/tensorflow/tensorflow/core/tfrt/utils/
H A Dtensor_util.cc175 const TensorHandle& tensor_handle) { in CreateTFTensorFromTensorHandle()
/aosp_15_r20/external/tensorflow/tensorflow/c/eager/
H A Dc_api_experimental_test.cc367 void Deleter(void* data, size_t unused, void* tensor_handle) { in Deleter()
H A Dc_api_test.cc1412 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 DFunctionHelpers.h69 ITensorHandle *tensor_handle = tensor->handle(); in get_backing_tensor() local