| /aosp_15_r20/external/tensorflow/tensorflow/go/ |
| H A D | tensor_handle.go | 42 type TensorHandle struct { struct
43 c *C.TFE_TensorHandle
59 func (th *TensorHandle) finalizer() {
71 func (th *TensorHandle) DataType() DataType {
76 func (th *TensorHandle) Shape() ([]int64, error) {
92 func (th *TensorHandle) numDims() (int, error) {
100 func (th *TensorHandle) dim(index int) (int64, error) {
115 func (th *TensorHandle) DeviceName() (string, error) {
127 func (th *TensorHandle) BackingDeviceName() (string, error) {
138 func (th *TensorHandle) ToTensor() (*Tensor, error) {
[all …]
|
| /aosp_15_r20/external/tensorflow/tensorflow/core/common_runtime/eager/ |
| H A 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
|
| H A D | shape_inference.cc | 30 const gtl::InlinedVector<TensorHandle*, 4>& inputs, in RunShapeInference()
|
| H A D | execute_node.h | 95 const absl::InlinedVector<TensorHandle*, 4>& inputs, in ExecuteNode()
151 const absl::InlinedVector<TensorHandle*, 4>& inputs, in AsyncExecuteNode()
|
| H A D | eager_operation.cc | 430 const absl::InlinedVector<TensorHandle*, 4>** inputs) const { in TensorHandleInputs() argument
441 absl::InlinedVector<TensorHandle*, 4>** inputs) { in MutableTensorHandleInputs()
|
| H A D | custom_device.h | 28 class TensorHandle; variable
|
| H A D | execute_node.cc | 65 EagerContext* ctx, const gtl::InlinedVector<TensorHandle*, 4>& op_inputs, in Init()
|
| H A D | context.h | 84 class TensorHandle; variable
|
| H A D | execute.cc | 1828 EagerContext* ctx, const absl::InlinedVector<TensorHandle*, 4>& op_inputs, in EagerKernelExecute()
2042 EagerContext* ctx, const absl::InlinedVector<TensorHandle*, 4>& op_inputs, in EagerKernelExecuteAsync()
|
| /aosp_15_r20/external/tensorflow/tensorflow/lite/ |
| H A D | signature_runner.h | 30 class TensorHandle; // Class for friend declarations. variable
147 friend class TensorHandle; variable
|
| /aosp_15_r20/external/tensorflow/tensorflow/lite/java/src/main/native/ |
| H A D | tensor_jni.cc | 83 class TensorHandle { class
85 TensorHandle(Interpreter* interpreter, int tensor_index) { in TensorHandle() function in tflite::TensorHandle
91 TensorHandle(SignatureRunner* runner, const char* name, bool is_input) { in TensorHandle() function in tflite::TensorHandle
|
| /aosp_15_r20/external/tensorflow/tensorflow/cc/experimental/base/public/ |
| H A D | runtime.h | 44 friend class TensorHandle; variable
|
| H A D | status.h | 61 friend class TensorHandle; variable
|
| H A D | tensorhandle.h | 54 explicit TensorHandle(TFE_TensorHandle* handle) : handle_(handle) {} in TensorHandle() function
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/jit/ |
| H A D | get_compiler_ir.h | 27 class TensorHandle; variable
|
| /aosp_15_r20/external/armnn/src/backends/backendsCommon/ |
| H A D | TensorHandle.cpp | 43 TensorHandle::TensorHandle(const TensorInfo& tensorInfo) in TensorHandle() function in armnn::TensorHandle
|
| /aosp_15_r20/external/tensorflow/tensorflow/core/distributed_runtime/eager/ |
| H A D | remote_execute_node.h | 46 const gtl::InlinedVector<TensorHandle*, 4>& inputs, in RemoteExecuteNode()
|
| /aosp_15_r20/external/armnn/include/armnn/backends/ |
| H A D | TensorHandle.hpp | 79 class TensorHandle : public ConstTensorHandle class
|
| /aosp_15_r20/external/tensorflow/tensorflow/python/ops/ |
| H A D | session_ops.py | 38 class TensorHandle: class
|
| /aosp_15_r20/external/tensorflow/tensorflow/core/tfrt/eager/ |
| H A D | c_api_tfrt.h | 61 class TensorHandle; variable
|
| /aosp_15_r20/external/tensorflow/tensorflow/c/eager/ |
| H A D | c_api_test.cc | 174 TEST(CAPI, TensorHandle) { in TEST() argument
|