Searched defs:TensorHandle (Results 1 – 8 of 8) sorted by relevance
| /external/tensorflow/tensorflow/go/ |
| D | tensor_handle.go | 41 type TensorHandle struct { struct
42 c *C.TFE_TensorHandle
58 func (th *TensorHandle) finalizer() {
70 func (th *TensorHandle) DataType() DataType {
75 func (th *TensorHandle) Shape() ([]int64, error) {
91 func (th *TensorHandle) numDims() (int, error) {
99 func (th *TensorHandle) dim(index int) (int64, error) {
114 func (th *TensorHandle) DeviceName() (string, error) {
130 func (th *TensorHandle) BackingDeviceName() (string, error) {
146 func (th *TensorHandle) ToTensor() (*Tensor, error) {
[all …]
|
| /external/tensorflow/tensorflow/core/common_runtime/eager/ |
| D | execute_node.h | 35 const tensorflow::gtl::InlinedVector<TensorHandle*, 4>& inputs, in ExecuteNode()
|
| D | tensor_handle.cc | 48 TensorHandle::TensorHandle(const class Tensor& t, Device* d, Device* op_device, in TensorHandle() function in tensorflow::TensorHandle
62 TensorHandle::TensorHandle(uint64 node_id, Device* d, Device* op_device, in TensorHandle() function in tensorflow::TensorHandle
81 TensorHandle::TensorHandle(int64 op_id, int32 output_num, in TensorHandle() function in tensorflow::TensorHandle
103 TensorHandle::TensorHandle(OutputGraphNode symbolic_tensor, DataType dtype) in TensorHandle() function in tensorflow::TensorHandle
|
| D | execute.cc | 367 gtl::InlinedVector<TensorHandle*, 2>* retvals, in EagerLocalExecute()
894 gtl::InlinedVector<TensorHandle*, 2>* retvals, in EagerExecute()
913 const gtl::InlinedVector<TensorHandle*, 4>& op_inputs, in EagerKernelExecute()
|
| /external/tensorflow/tensorflow/core/distributed_runtime/eager/ |
| D | remote_execute_node.h | 34 const gtl::InlinedVector<TensorHandle*, 4>& inputs, in RemoteExecuteNode()
|
| /external/tensorflow/tensorflow/python/ops/ |
| D | session_ops.py | 42 class TensorHandle(object): class
|
| /external/tensorflow/tensorflow/lite/java/src/main/native/ |
| D | tensor_jni.cc | 32 class TensorHandle { class
34 TensorHandle(tflite::Interpreter* interpreter, int tensor_index) in TensorHandle() function in __anon5dfcb4930111::TensorHandle
|
| /external/tensorflow/tensorflow/c/eager/ |
| D | c_api_test.cc | 441 TEST(CAPI, TensorHandle) { in TEST() argument
|