| /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/common_runtime/eager/ |
| D | placement_utils.cc | 148 TensorHandle* tensor_handle = (*inputs)[i]; in MaybePinToResourceDevice() local
149 if (tensor_handle->dtype == DT_RESOURCE) { in MaybePinToResourceDevice()
150 if (tensor_handle->resource_remote_device_incarnation() != 0) { in MaybePinToResourceDevice()
152 &ctx, tensor_handle->resource_remote_device_incarnation())); in MaybePinToResourceDevice()
154 Device* resource_device = tensor_handle->resource_device(); in MaybePinToResourceDevice()
156 << DataTypeString(tensor_handle->dtype) in MaybePinToResourceDevice()
|
| D | BUILD | 45 ":tensor_handle",
175 ":tensor_handle",
273 ":tensor_handle",
336 name = "tensor_handle",
338 "tensor_handle.cc",
341 "tensor_handle.h",
377 ":tensor_handle",
396 ":tensor_handle",
418 ":tensor_handle",
510 ":tensor_handle",
[all …]
|
| D | execute.cc | 268 Status GetDeviceForInput(const EagerContext& ctx, TensorHandle* tensor_handle, in GetDeviceForInput() argument
272 if (tensor_handle->Type() != TensorHandle::LOCAL) { in GetDeviceForInput()
273 Device* device = tensor_handle->device(); in GetDeviceForInput()
276 } else if (tensor_handle->dtype == DT_RESOURCE) { in GetDeviceForInput()
281 TF_RETURN_IF_ERROR(tensor_handle->Tensor(&tensor)); in GetDeviceForInput()
290 Device* device = tensor_handle->device(); in GetDeviceForInput()
294 is_tpu ? MTypeFromDTypeIntsOnDevice(tensor_handle->dtype) in GetDeviceForInput()
295 : MTypeFromDType(tensor_handle->dtype); in GetDeviceForInput()
|
| /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 | 48 "//tensorflow/core/common_runtime/eager:tensor_handle",
93 "//tensorflow/core/common_runtime/eager:tensor_handle",
121 "//tensorflow/core/common_runtime/eager:tensor_handle",
154 "//tensorflow/core/common_runtime/eager:tensor_handle",
179 "//tensorflow/core/common_runtime/eager:tensor_handle",
195 "//tensorflow/core/common_runtime/eager:tensor_handle",
230 "//tensorflow/core/common_runtime/eager:tensor_handle",
|
| D | eager_service_impl_test.cc | 439 tensorflow::TensorHandle* tensor_handle; in TEST_F() local
441 context_id, RemoteTensorHandleInternal(2, 0), &tensor_handle)); in TEST_F()
446 TF_ASSERT_OK(tensor_handle->Tensor(&t)); in TEST_F()
547 tensorflow::TensorHandle* tensor_handle; in TestFunction() local
549 context_id, RemoteTensorHandleInternal(2, 0), &tensor_handle)); in TestFunction()
550 TF_ASSERT_OK(tensor_handle->Tensor(&t)); in TestFunction()
640 tensorflow::TensorHandle* tensor_handle; in TestComponentFunction() local
643 &tensor_handle)); in TestComponentFunction()
644 TF_ASSERT_OK(tensor_handle->Tensor(&t)); in TestComponentFunction()
811 tensorflow::TensorHandle* tensor_handle; in CheckOutputsAndClose() local
[all …]
|
| D | eager_service_impl.cc | 684 TensorHandle* tensor_handle = TensorHandle::CreateLocalHandle( in SendTensor() local
690 TF_RETURN_IF_ERROR(EagerCopyToDevice(tensor_handle, eager_context, in SendTensor()
694 tensor_handle->Unref(); in SendTensor()
|
| /external/tensorflow/tensorflow/compiler/mlir/tensorflow/utils/ |
| D | eval_util.cc | 127 for (TFE_TensorHandle* tensor_handle : outputs) { in EvaluateOperation()
128 TFE_DeleteTensorHandle(tensor_handle); in EvaluateOperation()
134 for (TFE_TensorHandle* tensor_handle : outputs) { in EvaluateOperation()
135 TF_Tensor* tf_tensor = TFE_TensorHandleResolve(tensor_handle, status); in EvaluateOperation()
|
| /external/tensorflow/tensorflow/cc/experimental/base/public/ |
| D | tensorhandle.h | 86 TFE_TensorHandle* tensor_handle = TFE_NewTensorHandleFromTensor( in FromTensor() local
91 return TensorHandle(tensor_handle); in FromTensor()
|
| /external/tensorflow/tensorflow/c/experimental/saved_model/core/revived_types/ |
| D | BUILD | 48 "//tensorflow/core/common_runtime/eager:tensor_handle",
170 "//tensorflow/core/common_runtime/eager:tensor_handle",
|
| /external/tensorflow/tensorflow/c/eager/ |
| D | BUILD | 79 "//tensorflow/core/common_runtime/eager:tensor_handle",
792 "//tensorflow/core/common_runtime/eager:tensor_handle",
812 "//tensorflow/core/common_runtime/eager:tensor_handle",
970 "//tensorflow/core/common_runtime/eager:tensor_handle",
1135 "//tensorflow/core/common_runtime/eager:tensor_handle",
|
| D | c_api_test.cc | 1299 TFE_TensorHandle* tensor_handle = TFE_NewTensorHandle(tensor, status); in TEST() local
1301 TFE_OpAddInput(op, tensor_handle, status); in TEST()
1303 TFE_DeleteTensorHandle(tensor_handle); in TEST()
1358 TFE_TensorHandle* tensor_handle = TFE_NewTensorHandle(tensor, status); in TEST() local
1360 TFE_OpAddInput(op, tensor_handle, status); in TEST()
1362 TFE_DeleteTensorHandle(tensor_handle); in TEST()
|
| D | c_api_experimental_test.cc | 364 void Deleter(void* data, size_t unused, void* tensor_handle) { in Deleter() argument
365 TFE_DeleteTensorHandle(static_cast<TFE_TensorHandle*>(tensor_handle)); in Deleter()
|
| /external/tensorflow/tensorflow/python/client/ |
| D | session.py | 1416 for deleter_key, tensor_handle in enumerate(tensors_to_delete):
1418 self.graph, deleter_key, tensor_handle)
1419 feeds[holder] = tensor_handle
|
| /external/tensorflow/tensorflow/lite/delegates/flex/ |
| D | BUILD | 118 "//tensorflow/core/common_runtime/eager:tensor_handle",
|
| /external/tensorflow/tensorflow/python/lib/core/ |
| D | BUILD | 229 "//tensorflow/core/common_runtime/eager:tensor_handle",
|
| /external/tensorflow/tensorflow/c/experimental/saved_model/core/ |
| D | BUILD | 259 "//tensorflow/core/common_runtime/eager:tensor_handle",
|
| /external/tensorflow/tensorflow/tools/def_file_filter/ |
| D | symbols_pybind.txt | 386 [//tensorflow/core/common_runtime/eager:tensor_handle] # tfe
|
| /external/tensorflow/tensorflow/python/ |
| D | tfe_wrapper.cc | 332 for (TFE_TensorHandle* tensor_handle : handles) { in InputTFE_OutputTensorHandles()
333 AbstractTensorHandle* abstract_tensor_handle = unwrap(tensor_handle); in InputTFE_OutputTensorHandles()
|
| D | BUILD | 5324 "//tensorflow/core/common_runtime/eager:tensor_handle", # tfe
|
| /external/tensorflow/tensorflow/python/eager/ |
| D | BUILD | 60 "//tensorflow/core/common_runtime/eager:tensor_handle",
|
| /external/tensorflow/tensorflow/compiler/jit/ |
| D | BUILD | 494 "//tensorflow/core/common_runtime/eager:tensor_handle",
|