Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/java/src/main/native/
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
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()
Dgraph_operation_builder_jni.cc212 jlong tensor_handle) { in Java_org_tensorflow_GraphOperationBuilder_setAttrTensor()
/external/tensorflow/tensorflow/cc/experimental/base/public/
Dtensorhandle.h86 TFE_TensorHandle* tensor_handle = TFE_NewTensorHandleFromTensor( in FromTensor() local
/external/tensorflow/tensorflow/core/common_runtime/eager/
Dplacement_utils.cc148 TensorHandle* tensor_handle = (*inputs)[i]; in MaybePinToResourceDevice() local
Dexecute.cc268 Status GetDeviceForInput(const EagerContext& ctx, TensorHandle* tensor_handle, in GetDeviceForInput()
/external/tensorflow/tensorflow/core/distributed_runtime/eager/
Deager_service_impl_test.cc439 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
Deager_service_impl.cc684 TensorHandle* tensor_handle = TensorHandle::CreateLocalHandle( in SendTensor() local
/external/tensorflow/tensorflow/c/eager/
Dc_api_experimental_test.cc364 void Deleter(void* data, size_t unused, void* tensor_handle) { in Deleter()
Dc_api_test.cc1299 TFE_TensorHandle* tensor_handle = TFE_NewTensorHandle(tensor, status); in TEST() local
1358 TFE_TensorHandle* tensor_handle = TFE_NewTensorHandle(tensor, status); in TEST() local