Searched defs:tensor_handle (Results 1 – 10 of 10) sorted by relevance
/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 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
|
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()
|
D | graph_operation_builder_jni.cc | 212 jlong tensor_handle) { in Java_org_tensorflow_GraphOperationBuilder_setAttrTensor()
|
/external/tensorflow/tensorflow/cc/experimental/base/public/ |
D | tensorhandle.h | 86 TFE_TensorHandle* tensor_handle = TFE_NewTensorHandleFromTensor( in FromTensor() local
|
/external/tensorflow/tensorflow/core/common_runtime/eager/ |
D | placement_utils.cc | 148 TensorHandle* tensor_handle = (*inputs)[i]; in MaybePinToResourceDevice() local
|
D | execute.cc | 268 Status GetDeviceForInput(const EagerContext& ctx, TensorHandle* tensor_handle, in GetDeviceForInput()
|
/external/tensorflow/tensorflow/core/distributed_runtime/eager/ |
D | eager_service_impl_test.cc | 439 tensorflow::TensorHandle* tensor_handle; in TEST_F() local 547 tensorflow::TensorHandle* tensor_handle; in TestFunction() local 640 tensorflow::TensorHandle* tensor_handle; in TestComponentFunction() local 811 tensorflow::TensorHandle* tensor_handle; in CheckOutputsAndClose() local 918 tensorflow::TensorHandle* tensor_handle; in TEST_F() local 1083 tensorflow::TensorHandle* tensor_handle; in TEST_F() local
|
D | eager_service_impl.cc | 684 TensorHandle* tensor_handle = TensorHandle::CreateLocalHandle( in SendTensor() local
|
/external/tensorflow/tensorflow/c/eager/ |
D | c_api_experimental_test.cc | 364 void Deleter(void* data, size_t unused, void* tensor_handle) { in Deleter()
|
D | c_api_test.cc | 1299 TFE_TensorHandle* tensor_handle = TFE_NewTensorHandle(tensor, status); in TEST() local 1358 TFE_TensorHandle* tensor_handle = TFE_NewTensorHandle(tensor, status); in TEST() local
|