Home
last modified time | relevance | path

Searched defs:tensor_handle (Results 1 – 8 of 8) 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/core/distributed_runtime/eager/
Deager_service_impl_test.cc330 tensorflow::TensorHandle* tensor_handle; in TEST_F() local
408 tensorflow::TensorHandle* tensor_handle; in TestFunction() local
519 tensorflow::TensorHandle* tensor_handle; in CheckOutputsAndClose() local
683 tensorflow::TensorHandle* tensor_handle; in TEST_F() local
Deager_service_impl.cc529 TensorHandle* tensor_handle = nullptr; in SendTensor() local
/external/tensorflow/tensorflow/core/common_runtime/eager/
Dexecute.cc230 Status GetDeviceForInput(const EagerContext& ctx, TensorHandle* tensor_handle, in GetDeviceForInput()
832 TensorHandle* tensor_handle = op->Inputs()[i]; in MaybeUpdateOpDevice() local
/external/tensorflow/tensorflow/c/eager/
Dc_api_experimental_test.cc498 void Deleter(void* data, size_t unused, void* tensor_handle) { in Deleter()
Dc_api_test.cc1165 TFE_TensorHandle* tensor_handle = TFE_NewTensorHandle(tensor, status); in TEST() local