| /external/tensorflow/tensorflow/java/src/main/native/ |
| D | eager_operation_jni.cc | 64 TFE_TensorHandle* tensor_handle = requireTensorHandle(env, handle); in Java_org_tensorflow_EagerOperation_resolveTensorHandle() local
65 if (tensor_handle == nullptr) return 0; in Java_org_tensorflow_EagerOperation_resolveTensorHandle()
67 TF_Tensor* tensor = TFE_TensorHandleResolve(tensor_handle, status); in Java_org_tensorflow_EagerOperation_resolveTensorHandle()
112 TFE_TensorHandle* tensor_handle = requireTensorHandle(env, handle); in Java_org_tensorflow_EagerOperation_dataType() local
113 if (tensor_handle == nullptr) return 0; in Java_org_tensorflow_EagerOperation_dataType()
114 TF_DataType data_type = TFE_TensorHandleDataType(tensor_handle); in Java_org_tensorflow_EagerOperation_dataType()
120 TFE_TensorHandle* tensor_handle = requireTensorHandle(env, handle); in Java_org_tensorflow_EagerOperation_numDims() local
121 if (tensor_handle == nullptr) return 0; in Java_org_tensorflow_EagerOperation_numDims()
123 int num_dims = TFE_TensorHandleNumDims(tensor_handle, status); in Java_org_tensorflow_EagerOperation_numDims()
136 TFE_TensorHandle* tensor_handle = requireTensorHandle(env, handle); in Java_org_tensorflow_EagerOperation_dim() local
[all …]
|
| D | eager_operation_builder_jni.cc | 132 TFE_TensorHandle* tensor_handle = requireTensorHandle(env, input_handle); in Java_org_tensorflow_EagerOperationBuilder_addInput() local
133 if (tensor_handle == nullptr) return; in Java_org_tensorflow_EagerOperationBuilder_addInput()
135 TFE_OpAddInput(op, tensor_handle, status); in Java_org_tensorflow_EagerOperationBuilder_addInput()
259 jlong tensor_handle) { in Java_org_tensorflow_EagerOperationBuilder_setAttrTensor() argument
262 TF_Tensor* t = requireTensor(env, tensor_handle); in Java_org_tensorflow_EagerOperationBuilder_setAttrTensor()
|
| D | graph_operation_builder_jni.cc | 212 jlong tensor_handle) { in Java_org_tensorflow_GraphOperationBuilder_setAttrTensor() argument
215 TF_Tensor* t = requireTensor(env, tensor_handle); in Java_org_tensorflow_GraphOperationBuilder_setAttrTensor()
|
| /external/tensorflow/tensorflow/core/distributed_runtime/eager/ |
| D | remote_tensor_handle.h | 26 explicit RemoteTensorHandleInternal(const RemoteTensorHandle& tensor_handle) in RemoteTensorHandleInternal()
27 : op_id(tensor_handle.op_id()), output_num(tensor_handle.output_num()) {} in RemoteTensorHandleInternal()
|
| D | BUILD | 43 "//tensorflow/core/common_runtime/eager:tensor_handle",
85 "//tensorflow/core/common_runtime/eager:tensor_handle",
114 "//tensorflow/core/common_runtime/eager:tensor_handle",
146 "//tensorflow/core/common_runtime/eager:tensor_handle",
169 "//tensorflow/core/common_runtime/eager:tensor_handle",
182 "//tensorflow/core/common_runtime/eager:tensor_handle",
215 "//tensorflow/core/common_runtime/eager:tensor_handle",
|
| D | eager_service_impl_test.cc | 330 tensorflow::TensorHandle* tensor_handle; in TEST_F() local
332 context_id, RemoteTensorHandleInternal(2, 0), &tensor_handle)); in TEST_F()
337 TF_ASSERT_OK(tensor_handle->Tensor(&t)); in TEST_F()
408 tensorflow::TensorHandle* tensor_handle; in TestFunction() local
410 context_id, RemoteTensorHandleInternal(2, 0), &tensor_handle)); in TestFunction()
411 TF_ASSERT_OK(tensor_handle->Tensor(&t)); in TestFunction()
519 tensorflow::TensorHandle* tensor_handle; in CheckOutputsAndClose() local
521 context_id_, RemoteTensorHandleInternal(2, 0), &tensor_handle)); in CheckOutputsAndClose()
522 TF_ASSERT_OK(tensor_handle->Tensor(&t)); in CheckOutputsAndClose()
683 tensorflow::TensorHandle* tensor_handle; in TEST_F() local
[all …]
|
| D | eager_service_impl.cc | 529 TensorHandle* tensor_handle = nullptr; in SendTensor() local
530 TF_RETURN_IF_ERROR(TensorHandle::CreateLocalHandle(tensor, &tensor_handle)); in SendTensor()
535 TF_RETURN_IF_ERROR(EagerCopyToDevice(tensor_handle, eager_context, in SendTensor()
539 tensor_handle->Unref(); in SendTensor()
|
| /external/tensorflow/tensorflow/core/common_runtime/eager/ |
| D | BUILD | 109 ":tensor_handle",
141 name = "tensor_handle",
143 "tensor_handle.cc",
146 "tensor_handle.h",
178 ":tensor_handle",
196 ":tensor_handle",
218 ":tensor_handle",
336 ":tensor_handle",
463 "tensor_handle.h",
|
| D | execute.cc | 230 Status GetDeviceForInput(const EagerContext& ctx, TensorHandle* tensor_handle, in GetDeviceForInput() argument
234 if (tensor_handle->IsRemote()) { in GetDeviceForInput()
235 Device* device = tensor_handle->device(); in GetDeviceForInput()
238 } else if (tensor_handle->dtype == DT_RESOURCE) { in GetDeviceForInput()
243 TF_RETURN_IF_ERROR(tensor_handle->Tensor(&tensor)); in GetDeviceForInput()
251 } else if (MTypeFromDType(tensor_handle->dtype) == HOST_MEMORY) { in GetDeviceForInput()
254 Device* device = tensor_handle->device(); in GetDeviceForInput()
832 TensorHandle* tensor_handle = op->Inputs()[i]; in MaybeUpdateOpDevice() local
833 if (tensor_handle->dtype == DT_RESOURCE) { in MaybeUpdateOpDevice()
834 Device* resource_device = tensor_handle->resource_device(); in MaybeUpdateOpDevice()
[all …]
|
| /external/tensorflow/tensorflow/compiler/mlir/tensorflow/utils/ |
| D | eval_util.cc | 143 for (TFE_TensorHandle* tensor_handle : outputs) { in EvaluateOperation()
144 TFE_DeleteTensorHandle(tensor_handle); in EvaluateOperation()
150 for (TFE_TensorHandle* tensor_handle : outputs) { in EvaluateOperation()
151 TF_Tensor* tf_tensor = TFE_TensorHandleResolve(tensor_handle, status); in EvaluateOperation()
|
| /external/tensorflow/tensorflow/c/eager/ |
| D | BUILD | 51 "//tensorflow/core/common_runtime/eager:tensor_handle",
131 "//tensorflow/core/common_runtime/eager:tensor_handle",
231 "//tensorflow/core/common_runtime/eager:tensor_handle",
|
| D | c_api_experimental_test.cc | 498 void Deleter(void* data, size_t unused, void* tensor_handle) { in Deleter() argument
499 TFE_DeleteTensorHandle(static_cast<TFE_TensorHandle*>(tensor_handle)); in Deleter()
|
| D | c_api_test.cc | 1165 TFE_TensorHandle* tensor_handle = TFE_NewTensorHandle(tensor, status); in TEST() local
1167 TFE_OpAddInput(op, tensor_handle, status); in TEST()
1169 TFE_DeleteTensorHandle(tensor_handle); in TEST()
|
| /external/tensorflow/tensorflow/lite/delegates/flex/ |
| D | BUILD | 189 "//tensorflow/core/common_runtime/eager:tensor_handle",
|
| /external/tensorflow/tensorflow/python/client/ |
| D | session.py | 1408 for deleter_key, tensor_handle in enumerate(tensors_to_delete):
1410 self.graph, deleter_key, tensor_handle)
1411 feeds[holder] = tensor_handle
|
| /external/tensorflow/tensorflow/python/eager/ |
| D | BUILD | 40 "//tensorflow/core/common_runtime/eager:tensor_handle",
|