| /external/tensorflow/tensorflow/cc/experimental/base/public/ |
| D | tensorhandle.h | 37 class TensorHandle {
45 static TensorHandle FromTensor(const Tensor& tensor, const Runtime& runtime,
49 TensorHandle(TensorHandle&&) = default;
50 TensorHandle& operator=(TensorHandle&&) = default;
54 explicit TensorHandle(TFE_TensorHandle* handle) : handle_(handle) {} in TensorHandle() function
57 TensorHandle(const TensorHandle&) = delete;
58 TensorHandle& operator=(const TensorHandle&) = delete;
74 inline Tensor TensorHandle::Resolve(Status* status) { in Resolve()
83 inline TensorHandle TensorHandle::FromTensor(const Tensor& tensor, in FromTensor()
89 return TensorHandle(nullptr); in FromTensor()
[all …]
|
| /external/tensorflow/tensorflow/core/common_runtime/eager/ |
| D | tensor_handle.cc | 67 TensorHandle::PackedTensorHandleData::PackedTensorHandleData( in PackedTensorHandleData()
68 std::vector<TensorHandle*>&& handles, const TensorShape& shape) in PackedTensorHandleData()
75 TensorHandle::PackedTensorHandleData::~PackedTensorHandleData() { in ~PackedTensorHandleData()
81 Status TensorHandle::PackedTensorHandleData::Shape(TensorShape* shape) const { in Shape()
86 Status TensorHandle::PackedTensorHandleData::NumDims(int* num_dims) const { in NumDims()
91 Status TensorHandle::PackedTensorHandleData::Dim(int dim_index, in Dim()
97 Status TensorHandle::PackedTensorHandleData::NumElements( in NumElements()
103 Status TensorHandle::PackedTensorHandleData::Unprotect() { in Unprotect()
111 bool TensorHandle::PackedTensorHandleData::IsReady() const { in IsReady()
126 Status TensorHandle::PackedTensorHandleData::WaitReady( in WaitReady()
[all …]
|
| D | tensor_handle.h | 56 class TensorHandle : public ImmediateExecutionTensorHandle {
58 TensorHandle(tensorflow::Tensor&& t, Device* d, Device* op_device,
61 TensorHandle(tensorflow::Tensor&& t, Device* d, Device* op_device,
63 TensorHandle(Device* d, Device* op_device, Device* resource_device,
67 TensorHandle(int64 op_id, int32 output_num, const string& remote_task,
70 TensorHandle(int64 op_id, int32 output_num, tensorflow::DataType dtype,
76 static TensorHandle* CreateLocalHandle(const tensorflow::Tensor& t);
77 static TensorHandle* CreateLocalHandle(tensorflow::Tensor&& t, Device* d,
79 static TensorHandle* CreateLocalHandle(tensorflow::Tensor&& t, Device* d,
83 static TensorHandle* CreateEmptyLocalHandle(Device* d, Device* op_device,
[all …]
|
| D | tensor_handle_test.cc | 43 TensorHandle* sync_th = in TEST()
44 TensorHandle::CreateLocalHandle(std::move(t), nullptr, nullptr, ctx); in TEST()
45 TensorHandle* async_th = TensorHandle::CreateEmptyLocalHandle( in TEST()
124 bool IsReady(TensorHandle* handle) const { return handle->IsReady(); } in IsReady()
145 std::vector<TensorHandle*> handles; in TEST_F()
148 TensorHandle* h0 = in TEST_F()
149 TensorHandle::CreateLocalHandle(std::move(t0), d0, d0, d0, context()); in TEST_F()
154 TensorHandle* h1 = in TEST_F()
155 TensorHandle::CreateLocalHandle(std::move(t1), d1, d1, d1, context()); in TEST_F()
162 TensorHandle* h2 = TensorHandle::CreateUnshapedRemoteHandle( in TEST_F()
[all …]
|
| D | execute_node.cc | 23 TensorHandle* handle) { in IsRemote()
25 if (handle->Type() == TensorHandle::REMOTE || in IsRemote()
38 TensorHandle* packed_handle) { in InitPackedHandle()
43 TensorHandle* h = nullptr; in InitPackedHandle()
54 if (h->Type() == TensorHandle::PACKED) { in InitPackedHandle()
65 EagerContext* ctx, const gtl::InlinedVector<TensorHandle*, 4>& op_inputs, in Init() argument
74 TensorHandle* const* op_inputs_flat = &op_inputs[0]; in Init()
77 TensorHandle* in = op_inputs_flat[i]; in Init()
86 if (in->Type() != TensorHandle::PACKED) { in Init()
104 TensorHandle* h = op_inputs[index.index]; in Init()
[all …]
|
| D | execute_node_test.cc | 88 TensorHandle* h0 = in TEST()
89 TensorHandle::CreateLocalHandle(std::move(t0), device0, device0, ctx); in TEST()
92 TensorHandle* h1 = in TEST()
93 TensorHandle::CreateLocalHandle(std::move(t1), device0, device0, ctx); in TEST()
95 TensorHandle* h2 = TensorHandle::CreateLazyRemoteHandle( in TEST()
97 TensorHandle* h3 = TensorHandle::CreateLazyRemoteHandle( in TEST()
100 TensorHandle* packed_h = nullptr; in TEST()
101 TF_ASSERT_OK(TensorHandle::CreatePackedHandle({h1, h2}, ctx, &packed_h)); in TEST()
104 absl::InlinedVector<TensorHandle*, 4> inputs = {h0, packed_h, h3}; in TEST()
|
| D | execute_node.h | 55 const absl::InlinedVector<TensorHandle*, 4>& op_inputs,
75 bool IsRemote(EagerContext* ctx, Device* input_device, TensorHandle* handle);
80 Device* input_device, TensorHandle* packed_handle);
95 EagerContext* ctx, const absl::InlinedVector<TensorHandle*, 4>& inputs, in ExecuteNode() argument
100 absl::Span<TensorHandle*> retvals, in ExecuteNode()
114 for (TensorHandle* h : inputs_) { in Run()
139 const absl::InlinedVector<TensorHandle*, 4>& inputs_;
144 absl::Span<TensorHandle*> retvals_;
151 EagerContext* ctx, const absl::InlinedVector<TensorHandle*, 4>& inputs, in AsyncExecuteNode() argument
156 absl::Span<TensorHandle*> retvals, in AsyncExecuteNode()
[all …]
|
| D | execute.h | 45 Status EagerExecute(EagerOperation* op, TensorHandle** retvals,
51 EagerContext* ctx, const absl::InlinedVector<TensorHandle*, 4>& op_inputs,
55 absl::Span<TensorHandle*> retvals,
63 Status EagerCopyToDevice(TensorHandle* h, EagerContext* ctx,
65 TensorHandle** result);
85 void EagerLocalExecuteAsync(EagerOperation* op, TensorHandle** retvals,
|
| D | core.cc | 37 AbstractTensorInterface* TensorHandle::Resolve(Status* status) { in Resolve()
44 TensorHandle* h_cpu = nullptr; in Resolve()
141 TensorHandle* input = TensorHandleFromInterface(handle); in CopyTensorHandleToDevice()
144 reinterpret_cast<tensorflow::TensorHandle**>(&result)); in CopyTensorHandleToDevice()
158 return TensorHandle::CreateLocalHandle(std::move(tensor), /*d=*/HostCPU(), in CreateLocalHandle()
166 return TensorHandle::CreateLocalHandle(std::move(t), /*d=*/HostCPU(), in CreateLocalHandleFromTFTensor()
171 return TensorHandle::CreateLocalHandle(std::move(t), /*d=*/d, in CreateLocalHandleFromTFTensor()
201 if (TensorHandle::classof(handle)) { in Execute()
202 TF_RETURN_IF_ERROR(down_cast<TensorHandle*>(handle)->WaitUnknownDevice()); in Execute()
225 tensorflow::TensorHandle** retval_array = in Execute()
[all …]
|
| D | execute.cc | 106 TensorHandle* handle, // op->Inputs()[i] in CopyInputToExpectedDevice()
109 TensorHandle** result) { in CopyInputToExpectedDevice()
155 TensorHandle* result_handle = nullptr; in CopyInputToExpectedDevice()
195 const absl::InlinedVector<TensorHandle*, 4>* handles; in ValidateInputTypeAndPlacement()
198 TensorHandle* handle = (*handles)[i]; in ValidateInputTypeAndPlacement()
200 if (!kernel->IsFunction() && handle->Type() == TensorHandle::PACKED) { in ValidateInputTypeAndPlacement()
206 TensorHandle* h = nullptr; in ValidateInputTypeAndPlacement()
218 !is_function || handle->Type() != TensorHandle::REMOTE; in ValidateInputTypeAndPlacement()
268 Status GetDeviceForInput(const EagerContext& ctx, TensorHandle* tensor_handle, in GetDeviceForInput()
272 if (tensor_handle->Type() != TensorHandle::LOCAL) { in GetDeviceForInput()
[all …]
|
| D | copy_to_device_node.h | 29 CopyToDeviceNode(TensorHandle* src, TensorHandle* dst, Device* dstd, in CopyToDeviceNode()
79 TensorHandle* dst() { return dst_; } in dst()
82 TensorHandle* src_;
83 TensorHandle* dst_;
|
| D | shape_inference.h | 30 const gtl::InlinedVector<TensorHandle*, 4>& inputs,
31 const gtl::InlinedVector<TensorHandle*, 2>& retvals);
|
| D | custom_device_test.cc | 128 core::RefCountPtr<TensorHandle> physical_resource_tensor( in TEST()
129 TensorHandle::CreateLocalHandle(std::move(resource_tensor), in TEST()
133 core::RefCountPtr<TensorHandle> physical_float_tensor( in TEST()
134 TensorHandle::CreateLocalHandle(std::move(float_tensor), physical_device, in TEST()
|
| D | shape_inference.cc | 30 const gtl::InlinedVector<TensorHandle*, 4>& inputs, in RunShapeInference() argument
31 const gtl::InlinedVector<TensorHandle*, 2>& retvals) { in RunShapeInference()
|
| /external/tensorflow/tensorflow/go/ |
| D | tensor_handle.go | 41 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/distributed_runtime/eager/ |
| D | remote_mgr.h | 47 const gtl::ArraySlice<tensorflow::TensorHandle*> handles,
50 void AddOperationOutput(tensorflow::TensorHandle* handles, int64 operation_id,
54 tensorflow::TensorHandle** handle);
70 TensorHandle* in, const bool wait_until_ready, RemoteTensorHandle* out,
77 TensorHandle** out);
90 Status GetRemoteTensorHandle(const tensorflow::TensorHandle* handle,
96 tensorflow::TensorHandle** handle)
106 gtl::FlatMap<RemoteTensorHandleInternal, tensorflow::TensorHandle*,
|
| D | remote_mgr_test.cc | 72 TensorHandle* handle = TensorHandle::CreateLocalHandle( in TEST_F()
95 TensorHandle* handle = TensorHandle::CreateLazyRemoteHandle( in TEST_F()
111 TensorHandle* handle = TensorHandle::CreateLocalHandle( in TEST_F()
136 TensorHandle* handle = TensorHandle::CreateUnshapedRemoteHandle( in TEST_F()
145 handle = TensorHandle::CreateUnshapedRemoteHandle( in TEST_F()
|
| D | remote_copy_node.h | 63 RemoteCopyNode(EagerContext* ctx, EagerExecutor* executor, TensorHandle* src,
64 TensorHandle* dst, Device* recv_device, uint64 recv_op_id);
132 explicit CapturedSharedState(TensorHandle* d) : dst_(d) { dst_->Ref(); } in CapturedSharedState()
150 TensorHandle* dst() { return dst_; } in dst()
154 TensorHandle* const dst_;
163 TensorHandle* const src_;
|
| D | remote_mgr.cc | 28 const gtl::ArraySlice<tensorflow::TensorHandle*> handles, in AddOperationOutputs()
38 void RemoteMgr::AddOperationOutput(tensorflow::TensorHandle* handle, in AddOperationOutput()
47 tensorflow::TensorHandle** handle) { in GetTensorHandleImpl()
62 tensorflow::TensorHandle** handle) { in GetTensorHandle()
83 Status RemoteMgr::GetRemoteTensorHandle(const tensorflow::TensorHandle* handle, in GetRemoteTensorHandle()
88 tensorflow::TensorHandle* h; in GetRemoteTensorHandle()
124 TensorHandle* in, const bool wait_until_ready, RemoteTensorHandle* out, in SerializeRemoteTensorHandle()
155 TensorHandle** out) { in DeserializeRemoteTensorHandle()
168 *out = TensorHandle::CreateLazyRemoteHandle(in.op_id(), in.output_num(), in DeserializeRemoteTensorHandle()
|
| D | remote_execute_node.h | 45 const gtl::InlinedVector<TensorHandle*, 4>& inputs, in RemoteExecuteNode() argument
46 absl::Span<TensorHandle*> retvals) in RemoteExecuteNode()
72 for (const TensorHandle* input : inputs_) { in RemoteExecuteNode()
134 gtl::InlinedVector<TensorHandle*, 4> inputs_;
135 gtl::InlinedVector<TensorHandle*, 2> retvals_;
|
| D | eager_service_impl.cc | 117 tensorflow::TensorHandle* handle; in GetEagerOperationAndNumRetvals()
129 handle = TensorHandle::CreateLocalHandle(std::move(tensor), nullptr, in GetEagerOperationAndNumRetvals()
148 Status TensorHandleProto(TensorHandle* handle, TensorProto* proto) { in TensorHandleProto()
155 Status TensorHandleShape(TensorHandle* handle, TensorShapeProto* proto) { in TensorHandleShape()
159 if (handle->Type() == TensorHandle::LOCAL) { in TensorHandleShape()
174 const std::vector<int32>& output_nums, TensorHandle** retvals, in AddOpRetvalsToResponse()
192 if (retvals[i]->Type() == TensorHandle::REMOTE) { in AddOpRetvalsToResponse()
471 auto* retvals = new absl::FixedArray<TensorHandle*>(*num_retvals); in RunComponentFunction()
526 absl::FixedArray<tensorflow::TensorHandle*> retvals(num_retvals); in ExecuteOp()
677 tensorflow::gtl::InlinedVector<tensorflow::TensorHandle*, 2> tensors; in SendTensor()
[all …]
|
| /external/tensorflow/tensorflow/c/experimental/saved_model/core/revived_types/ |
| D | variable.cc | 89 std::vector<TensorHandle*> handles; in CreateUninitialized()
95 if (!tensorflow::isa<TensorHandle>(handlePtr.get())) { in CreateUninitialized()
98 handles.push_back(reinterpret_cast<TensorHandle*>(handlePtr.release())); in CreateUninitialized()
100 TensorHandle* packed_handle; in CreateUninitialized()
101 TF_RETURN_IF_ERROR(TensorHandle::CreatePackedHandle( in CreateUninitialized()
108 TensorHandle* component; in CreateUninitialized()
|
| /external/tensorflow/tensorflow/python/ops/ |
| D | session_ops.py | 42 class TensorHandle(object): class
135 return feeder.op.name + ";" + TensorHandle._get_reader_key(handle)
214 handle_device = TensorHandle._get_device_name(handle)
239 handle_device = TensorHandle._get_device_name(handle)
256 graph_key = TensorHandle._get_reader_key(handle)
260 handle_device = TensorHandle._get_device_name(handle)
275 handle_device = TensorHandle._get_device_name(handle)
279 graph_key = TensorHandle._get_mover_key(feeder, handle)
296 handle_device = TensorHandle._get_device_name(handle)
|
| /external/tensorflow/tensorflow/cc/experimental/base/tests/ |
| D | tensorhandle_test.cc | 38 using tensorflow::experimental::cc::TensorHandle;
68 TensorHandle handle = in TYPED_TEST()
69 TensorHandle::FromTensor(original_tensor, *runtime, &status); in TYPED_TEST()
110 TensorHandle handle = in TYPED_TEST()
111 TensorHandle::FromTensor(original_tensor, *runtime, &status); in TYPED_TEST()
160 TensorHandle handle = in TYPED_TEST()
161 TensorHandle::FromTensor(original_tensor, *runtime, &status); in TYPED_TEST()
|
| /external/tensorflow/tensorflow/compiler/jit/ |
| D | get_compiler_ir.h | 27 class TensorHandle; variable
43 absl::Span<const TensorHandle* const> inputs);
|