| /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 | shape_inference.cc | 30 const gtl::InlinedVector<TensorHandle*, 4>& inputs, in RunShapeInference()
|
| D | execute_node.h | 95 EagerContext* ctx, const absl::InlinedVector<TensorHandle*, 4>& inputs, in ExecuteNode()
151 EagerContext* ctx, const absl::InlinedVector<TensorHandle*, 4>& inputs, in AsyncExecuteNode()
|
| D | custom_device.h | 28 class TensorHandle; variable
|
| D | tensor_handle.cc | 242 TensorHandle::TensorHandle(tensorflow::Tensor&& t, Device* d, Device* op_device, in TensorHandle() function in tensorflow::TensorHandle
258 TensorHandle::TensorHandle(tensorflow::Tensor&& t, Device* d, Device* op_device, in TensorHandle() function in tensorflow::TensorHandle
285 TensorHandle::TensorHandle(Device* d, Device* op_device, in TensorHandle() function in tensorflow::TensorHandle
349 TensorHandle::TensorHandle(std::vector<TensorHandle*>&& handles, Device* device, in TensorHandle() function in tensorflow::TensorHandle
376 TensorHandle::TensorHandle(int64_t op_id, int32_t output_num, in TensorHandle() function in tensorflow::TensorHandle
401 TensorHandle::TensorHandle(int64_t op_id, int32_t output_num, in TensorHandle() function in tensorflow::TensorHandle
|
| D | execute_node.cc | 65 EagerContext* ctx, const gtl::InlinedVector<TensorHandle*, 4>& op_inputs, in Init()
|
| D | eager_operation.cc | 428 const absl::InlinedVector<TensorHandle*, 4>** inputs) const { in TensorHandleInputs() argument
439 absl::InlinedVector<TensorHandle*, 4>** inputs) { in MutableTensorHandleInputs()
|
| D | context.h | 83 class TensorHandle; variable
|
| D | execute.cc | 1535 EagerContext* ctx, const absl::InlinedVector<TensorHandle*, 4>& op_inputs, in EagerKernelExecute()
1747 EagerContext* ctx, const absl::InlinedVector<TensorHandle*, 4>& op_inputs, in EagerKernelExecuteAsync()
|
| /external/tensorflow/tensorflow/compiler/jit/ |
| D | get_compiler_ir.h | 27 class TensorHandle; variable
|
| /external/tensorflow/tensorflow/cc/experimental/base/public/ |
| D | runtime.h | 44 friend class TensorHandle; variable
|
| D | status.h | 61 friend class TensorHandle; variable
|
| D | tensorhandle.h | 54 explicit TensorHandle(TFE_TensorHandle* handle) : handle_(handle) {} in TensorHandle() function
|
| /external/tensorflow/tensorflow/core/distributed_runtime/eager/ |
| D | remote_execute_node.h | 45 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/core/tfrt/eager/ |
| D | c_api_tfrt.h | 61 class TensorHandle; variable
|
| /external/tensorflow/tensorflow/lite/java/src/main/native/ |
| D | tensor_jni.cc | 42 class TensorHandle { class
44 TensorHandle(Interpreter* interpreter, int tensor_index, int subgraph_index) in TensorHandle() function in __anon6691a0bf0111::TensorHandle
|
| /external/tensorflow/tensorflow/c/eager/ |
| D | c_api_test.cc | 173 TEST(CAPI, TensorHandle) { in TEST() argument
|