Searched refs:RemoteTensorHandleInternal (Results 1 – 6 of 6) sorted by relevance
| /external/tensorflow/tensorflow/core/distributed_runtime/eager/ |
| D | remote_tensor_handle.h | 25 struct RemoteTensorHandleInternal { struct
26 explicit RemoteTensorHandleInternal(const RemoteTensorHandle& tensor_handle) in RemoteTensorHandleInternal() argument
28 RemoteTensorHandleInternal(int64 op_id, int32 output_num) in RemoteTensorHandleInternal() function
35 std::size_t operator()(const RemoteTensorHandleInternal& handle) const { in operator() argument
41 bool operator()(const RemoteTensorHandleInternal& first, in operator()
42 const RemoteTensorHandleInternal& second) const { in operator()
|
| D | remote_mgr.h | 53 Status GetTensorHandle(const RemoteTensorHandleInternal& remote_handle,
56 Status DeleteTensorHandle(const RemoteTensorHandleInternal& remote_handle);
95 Status GetTensorHandleImpl(const RemoteTensorHandleInternal& remote_handle,
100 const RemoteTensorHandleInternal& remote_handle,
106 gtl::FlatMap<RemoteTensorHandleInternal, tensorflow::TensorHandle*,
110 RemoteTensorHandleInternal, std::vector<DtypeAndPartialTensorShape>,
|
| D | remote_mgr.cc | 34 RemoteTensorHandleInternal(operation_id, i), handles[i]); in AddOperationOutputs()
42 RemoteTensorHandleInternal(operation_id, output_num), handle); in AddOperationOutput()
46 const RemoteTensorHandleInternal& remote_handle, in GetTensorHandleImpl()
61 const RemoteTensorHandleInternal& remote_handle, in GetTensorHandle()
68 const RemoteTensorHandleInternal& remote_handle, in GetMirroredResourceShape()
90 GetTensorHandleImpl(RemoteTensorHandleInternal(*op_id, *output_num), &h)); in GetRemoteTensorHandle()
100 const RemoteTensorHandleInternal& remote_handle) { in DeleteTensorHandle()
159 TF_RETURN_IF_ERROR(GetTensorHandle(RemoteTensorHandleInternal(in), out)); in DeserializeRemoteTensorHandle()
172 if (!GetMirroredResourceShape(RemoteTensorHandleInternal(in), in DeserializeRemoteTensorHandle()
183 RemoteTensorHandleInternal(in.op_id(), in.output_num()), in DeserializeRemoteTensorHandle()
|
| D | eager_service_impl.h | 175 std::unique_ptr<RemoteTensorHandleInternal> handle_to_delete) in ClientTensorHandleDeleteNode()
207 const std::unique_ptr<RemoteTensorHandleInternal> handle_to_delete_;
|
| D | eager_service_impl_test.cc | 58 const RemoteTensorHandleInternal& remote_handle, in GetTensorHandle()
441 context_id, RemoteTensorHandleInternal(2, 0), &tensor_handle)); in TEST_F()
549 context_id, RemoteTensorHandleInternal(2, 0), &tensor_handle)); in TestFunction()
642 context_id, RemoteTensorHandleInternal(2, output_num), in TestComponentFunction()
813 context_id_, RemoteTensorHandleInternal(2, 0), &tensor_handle)); in CheckOutputsAndClose()
920 context_id_, RemoteTensorHandleInternal(1, 0), &tensor_handle)); in TEST_F()
1085 context_id, RemoteTensorHandleInternal(2, 0), &tensor_handle)); in TEST_F()
1165 context_id, RemoteTensorHandleInternal(3, 0), &packed_handle)); in TEST_F()
|
| D | eager_service_impl.cc | 574 auto handle_to_decref = absl::make_unique<RemoteTensorHandleInternal>( in Enqueue()
|