Home
last modified time | relevance | path

Searched refs:TensorHandle (Results 1 – 21 of 21) sorted by relevance

/external/tensorflow/tensorflow/go/
Dtensor_handle.go41 type TensorHandle struct { struct
46 func NewTensorHandle(t *Tensor) (*TensorHandle, error) {
53 th := &TensorHandle{c: cHandle}
54 runtime.SetFinalizer(th, (*TensorHandle).finalizer)
58 func (th *TensorHandle) finalizer() { argument
63 func newTensorHandleFromC(c *C.TFE_TensorHandle) *TensorHandle {
64 th := &TensorHandle{c: c}
65 runtime.SetFinalizer(th, (*TensorHandle).finalizer)
70 func (th *TensorHandle) DataType() DataType { argument
75 func (th *TensorHandle) Shape() ([]int64, error) { argument
[all …]
/external/tensorflow/tensorflow/core/common_runtime/eager/
Dtensor_handle.cc48 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
116 bool TensorHandle::IsReady() { in IsReady()
122 bool TensorHandle::IsRemote() { in IsRemote()
126 Status TensorHandle::WaitForNode(uint64 node_id, bool return_if_is_ready) { in WaitForNode()
137 Status TensorHandle::WaitReady() { return WaitForNode(node_id_, true); } in WaitReady()
139 Status TensorHandle::Tensor(const tensorflow::Tensor** t) { in Tensor()
151 Status TensorHandle::TensorValue(tensorflow::TensorValue* t) { in TensorValue()
[all …]
Dcopy_to_device_node.h28 CopyToDeviceNode(TensorHandle* src, Device* dstd, EagerContext* ctx) in CopyToDeviceNode()
33 dst_(new TensorHandle(id, dstd_, dstd_, nullptr, src->dtype, ctx)) { in CopyToDeviceNode()
44 TensorHandle* temp = nullptr; in Run()
56 TensorHandle* dst() { return dst_; } in dst()
59 TensorHandle* src_;
62 TensorHandle* dst_;
Dexecute.h41 tensorflow::gtl::InlinedVector<tensorflow::TensorHandle*, 2>* retvals,
47 const gtl::InlinedVector<TensorHandle*, 4>& op_inputs,
51 TensorHandle** retvals, int num_retvals);
54 Status EagerCopyToDevice(TensorHandle* h, EagerContext* ctx,
55 const char* device_name, TensorHandle** result);
Deager_operation.h35 for (tensorflow::TensorHandle* h : inputs_) { in ~EagerOperation()
47 const tensorflow::gtl::InlinedVector<tensorflow::TensorHandle*, 4>& Inputs() in Inputs()
51 tensorflow::gtl::InlinedVector<tensorflow::TensorHandle*, 4>*
55 void AddInput(tensorflow::TensorHandle* h);
56 void ConsumeInput(tensorflow::TensorHandle* h);
74 tensorflow::gtl::InlinedVector<tensorflow::TensorHandle*, 4> inputs_;
Dtensor_handle.h58 class TensorHandle : public core::RefCounted {
60 TensorHandle(const Tensor& t, Device* d, Device* op_device,
62 TensorHandle(uint64 node_id, Device* d, Device* op_device,
66 TensorHandle(int64 op_id, int32 output_num, uint64 remote_shape_node_id,
71 TensorHandle(OutputGraphNode symbolic_tensor, DataType dtype);
73 ~TensorHandle() override { in ~TensorHandle()
106 TensorHandle** output);
Dexecute.cc95 TensorHandle** handle) { in MaybeCopyInputToExpectedDevice()
141 TensorHandle* result_handle = nullptr; in MaybeCopyInputToExpectedDevice()
194 tensorflow::TensorHandle* handle = op->Inputs()[i]; in ValidateInputTypeAndPlacement()
326 for (TensorHandle* tensor_handle : op->Inputs()) { in AddInputDevicesToCacheKey()
367 gtl::InlinedVector<TensorHandle*, 2>* retvals, in EagerLocalExecute() argument
506 (*retvals)[i] = new TensorHandle( in EagerLocalExecute()
575 Status EagerRemoteSendTensor(EagerContext* ctx, TensorHandle* h, in EagerRemoteSendTensor()
576 Device* recv_device, TensorHandle** result) { in EagerRemoteSendTensor()
599 TensorHandle* actual_handle; in EagerRemoteSendTensor()
628 *result = new TensorHandle(id, /*output_num=*/0, /*remote_shape_node_id=*/0, in EagerRemoteSendTensor()
[all …]
Dexecute_node.h35 const tensorflow::gtl::InlinedVector<TensorHandle*, 4>& inputs, in ExecuteNode() argument
39 const tensorflow::gtl::InlinedVector<TensorHandle*, 2>& retvals) in ExecuteNode()
81 tensorflow::gtl::InlinedVector<TensorHandle*, 4> inputs_;
86 tensorflow::gtl::InlinedVector<TensorHandle*, 2> retvals_;
Deager_operation.cc28 void EagerOperation::AddInput(tensorflow::TensorHandle* h) { in AddInput()
34 void EagerOperation::ConsumeInput(tensorflow::TensorHandle* h) { in ConsumeInput()
/external/tensorflow/tensorflow/python/ops/
Dsession_ops.py42 class TensorHandle(object): class
136 return feeder.op.name + ";" + TensorHandle._get_reader_key(handle)
215 handle_device = TensorHandle._get_device_name(handle)
240 handle_device = TensorHandle._get_device_name(handle)
257 graph_key = TensorHandle._get_reader_key(handle)
261 handle_device = TensorHandle._get_device_name(handle)
276 handle_device = TensorHandle._get_device_name(handle)
280 graph_key = TensorHandle._get_mover_key(feeder, handle)
297 handle_device = TensorHandle._get_device_name(handle)
/external/tensorflow/tensorflow/c/eager/
Dc_api_internal.h81 : handle(new tensorflow::TensorHandle(t, d, op_device, nullptr)) {} in TFE_TensorHandle()
83 TFE_TensorHandle(tensorflow::TensorHandle* handle) : handle(handle) {} in TFE_TensorHandle()
85 tensorflow::TensorHandle* handle;
89 : handle(new tensorflow::TensorHandle( in TFE_TensorHandle()
147 std::map<tensorflow::TensorHandle*, TF_Output> input_tensor_map;
148 std::vector<std::pair<tensorflow::TensorHandle*, TF_Output>>* input_tensors =
Dc_api.cc522 tensorflow::TensorHandle* h_cpu = nullptr; in TFE_TensorHandleResolve()
813 tensorflow::gtl::InlinedVector<tensorflow::TensorHandle*, 2> handle_retvals( in TFE_Execute()
829 tensorflow::TensorHandle* handle; in TFE_TensorHandleCopyToDevice()
Dc_api_test.cc441 TEST(CAPI, TensorHandle) { in TEST() argument
/external/tensorflow/tensorflow/core/distributed_runtime/eager/
Deager_service_impl.cc140 Status TensorHandleShape(TensorHandle* handle, TensorShapeProto* proto) { in TensorHandleShape()
174 tensorflow::TensorHandle* handle; in ExecuteOp()
190 tensorflow::gtl::InlinedVector<tensorflow::TensorHandle*, 2> retvals; in ExecuteOp()
281 tensorflow::gtl::InlinedVector<tensorflow::TensorHandle*, 2> tensors; in SendTensor()
288 TensorHandle* tensor_handle = in SendTensor()
289 new TensorHandle(tensor, nullptr, nullptr, nullptr); in SendTensor()
291 TensorHandle* copied_handle = nullptr; in SendTensor()
Deager_service_impl.h123 const gtl::ArraySlice<tensorflow::TensorHandle*>& handles, in AddOperationOutputs()
134 tensorflow::TensorHandle** handle) { in GetTensorHandle()
177 gtl::FlatMap<RemoteTensorHandleInternal, tensorflow::TensorHandle*,
Dremote_execute_node.h34 const gtl::InlinedVector<TensorHandle*, 4>& inputs, in RemoteExecuteNode() argument
84 gtl::InlinedVector<TensorHandle*, 4> inputs_;
Deager_service_impl_test.cc45 tensorflow::TensorHandle** handle) { in GetTensorHandle()
203 tensorflow::TensorHandle* tensor_handle; in TEST_F()
274 tensorflow::TensorHandle* tensor_handle; in TEST_F()
340 tensorflow::TensorHandle* tensor_handle; in TEST_F()
/external/tensorflow/tensorflow/lite/java/src/main/native/
Dtensor_jni.cc32 class TensorHandle { class
34 TensorHandle(tflite::Interpreter* interpreter, int tensor_index) in TensorHandle() function in __anone188269c0111::TensorHandle
51 return reinterpret_cast<TensorHandle*>(handle)->tensor(); in GetTensorFromHandle()
60 return reinterpret_cast<TensorHandle*>(handle)->index(); in GetTensorIndexFromHandle()
301 return reinterpret_cast<jlong>(new TensorHandle(interpreter, tensor_index)); in Java_org_tensorflow_lite_Tensor_create()
307 delete reinterpret_cast<TensorHandle*>(handle); in Java_org_tensorflow_lite_Tensor_delete()
/external/tensorflow/tensorflow/lite/delegates/flex/
Dkernel.cc137 tensorflow::TensorHandle* GetHandle(int i, bool remove) { in GetHandle()
163 tensorflow::gtl::InlinedVector<tensorflow::TensorHandle*, 2>*
171 tensorflow::gtl::InlinedVector<tensorflow::TensorHandle*, 2> vector_;
269 for (tensorflow::TensorHandle* h : *op_->MutableInputs()) { in ClearEagerInputs()
287 auto* handle = new tensorflow::TensorHandle( in BuildEagerInputs()
/external/tensorflow/tensorflow/python/client/
Dsession.py556 full_values.append(session_ops.TensorHandle(value, dtype, session))
1117 session_ops.TensorHandle)
/external/tensorflow/tensorflow/c/
Dc_api_experimental.cc9011 tensorflow::TensorHandle* handle, in getOrCreateSymbolicTensor()
9120 new std::vector<std::pair<tensorflow::TensorHandle*, TF_Output>>(); in TFE_FinalizeInputTensorsFromTraceContext()