Home
last modified time | relevance | path

Searched refs:tensor_handle (Results 1 – 11 of 11) sorted by relevance

/external/tensorflow/tensorflow/core/distributed_runtime/eager/
Deager_service_impl_test.cc203 tensorflow::TensorHandle* tensor_handle; in TEST_F() local
205 response.context_id(), RemoteTensorHandleInternal(2, 0), &tensor_handle)); in TEST_F()
210 TF_ASSERT_OK(tensor_handle->Tensor(&t)); in TEST_F()
274 tensorflow::TensorHandle* tensor_handle; in TEST_F() local
276 response.context_id(), RemoteTensorHandleInternal(2, 0), &tensor_handle)); in TEST_F()
277 TF_ASSERT_OK(tensor_handle->Tensor(&t)); in TEST_F()
340 tensorflow::TensorHandle* tensor_handle; in TEST_F() local
342 response.context_id(), RemoteTensorHandleInternal(2, 0), &tensor_handle)); in TEST_F()
343 TF_ASSERT_OK(tensor_handle->Tensor(&t)); in TEST_F()
345 Device* device = tensor_handle->device(); in TEST_F()
Dremote_tensor_handle.h26 explicit RemoteTensorHandleInternal(const RemoteTensorHandle& tensor_handle) in RemoteTensorHandleInternal()
27 : op_id(tensor_handle.op_id()), output_num(tensor_handle.output_num()) {} in RemoteTensorHandleInternal()
DBUILD40 "//tensorflow/core/common_runtime/eager:tensor_handle",
64 "//tensorflow/core/common_runtime/eager:tensor_handle",
90 "//tensorflow/core/common_runtime/eager:tensor_handle",
Deager_service_impl.cc288 TensorHandle* tensor_handle = in SendTensor() local
292 TF_RETURN_IF_ERROR(EagerCopyToDevice(tensor_handle, context->Context(), in SendTensor()
296 tensor_handle->Unref(); in SendTensor()
/external/tensorflow/tensorflow/core/common_runtime/eager/
DBUILD87 ":tensor_handle",
92 name = "tensor_handle",
94 "tensor_handle.cc",
97 "tensor_handle.h",
130 ":tensor_handle",
218 ":tensor_handle",
Dexecute.cc326 for (TensorHandle* tensor_handle : op->Inputs()) { in AddInputDevicesToCacheKey()
328 if (tensor_handle->dtype == DT_RESOURCE) { in AddInputDevicesToCacheKey()
332 TF_RETURN_IF_ERROR(tensor_handle->Tensor(&tensor)); in AddInputDevicesToCacheKey()
340 } else if (MTypeFromDType(tensor_handle->dtype) == HOST_MEMORY) { in AddInputDevicesToCacheKey()
343 Device* device = tensor_handle->device(); in AddInputDevicesToCacheKey()
825 TensorHandle* tensor_handle = op->Inputs()[i]; in MaybeUpdateOpDevice() local
826 if (tensor_handle->dtype == DT_RESOURCE) { in MaybeUpdateOpDevice()
827 Device* resource_device = tensor_handle->resource_device(); in MaybeUpdateOpDevice()
829 << DataTypeString(tensor_handle->dtype) in MaybeUpdateOpDevice()
850 Device* input_device = tensor_handle->device(); in MaybeUpdateOpDevice()
[all …]
/external/tensorflow/tensorflow/c/eager/
DBUILD44 "//tensorflow/core/common_runtime/eager:tensor_handle",
101 "//tensorflow/core/common_runtime/eager:tensor_handle",
185 "//tensorflow/core/common_runtime/eager:tensor_handle",
Dc_api_test.cc1565 TFE_TensorHandle* tensor_handle = TFE_NewTensorHandle(tensor, status); in TEST() local
1567 TFE_OpAddInput(op, tensor_handle, status); in TEST()
1569 TFE_DeleteTensorHandle(tensor_handle); in TEST()
/external/tensorflow/tensorflow/java/src/main/native/
Doperation_builder_jni.cc208 jlong tensor_handle) { in Java_org_tensorflow_OperationBuilder_setAttrTensor() argument
211 TF_Tensor* t = requireTensor(env, tensor_handle); in Java_org_tensorflow_OperationBuilder_setAttrTensor()
/external/tensorflow/tensorflow/python/client/
Dsession.py1371 for deleter_key, tensor_handle in enumerate(tensors_to_delete):
1373 self.graph, deleter_key, tensor_handle)
1374 feeds[holder] = tensor_handle
/external/tensorflow/tensorflow/lite/delegates/flex/
DBUILD163 "//tensorflow/core/common_runtime/eager:tensor_handle",