Searched defs:TensorHandle (Results 1 – 10 of 10) 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 | shape_inference.cc | 30 const gtl::InlinedVector<TensorHandle*, 4>& inputs, in RunShapeInference()
|
| D | execute_node.cc | 21 EagerContext* ctx, const gtl::InlinedVector<TensorHandle*, 4>& op_inputs) { in Init()
|
| D | tensor_handle.cc | 122 TensorHandle::TensorHandle(std::unique_ptr<LocalTensorHandleData> t, in TensorHandle() function in tensorflow::TensorHandle
141 TensorHandle::TensorHandle(std::unique_ptr<LocalTensorHandleData> t, in TensorHandle() function in tensorflow::TensorHandle
172 TensorHandle::TensorHandle(std::unique_ptr<EmptyLocalTensorHandleData> t, in TensorHandle() function in tensorflow::TensorHandle
215 TensorHandle::TensorHandle(std::unique_ptr<RemoteTensorHandleData> t, in TensorHandle() function in tensorflow::TensorHandle
250 TensorHandle::TensorHandle(std::unique_ptr<UnshapedRemoteTensorHandleData> t, in TensorHandle() function in tensorflow::TensorHandle
|
| D | execute_node.h | 75 EagerContext* ctx, const gtl::InlinedVector<TensorHandle*, 4>& inputs, in ExecuteNode()
|
| D | execute.cc | 945 EagerContext* ctx, const gtl::InlinedVector<TensorHandle*, 4>& op_inputs, in EagerKernelExecute()
|
| /external/tensorflow/tensorflow/core/distributed_runtime/eager/ |
| D | remote_execute_node.h | 42 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 | 37 class TensorHandle { class
39 TensorHandle(tflite_api_dispatcher::Interpreter* interpreter, in TensorHandle() function in __anon5574315e0111::TensorHandle
|
| /external/tensorflow/tensorflow/c/eager/ |
| D | c_api_test.cc | 152 TEST(CAPI, TensorHandle) { in TEST() argument
|