Searched refs:RemoteTensorHandleData (Results 1 – 4 of 4) sorted by relevance
| /aosp_15_r20/external/tensorflow/tensorflow/core/distributed_runtime/eager/ |
| H A D | remote_tensor_handle_data.cc | 88 RemoteTensorHandleData::RemoteTensorHandleData(int64_t op_id, int output_num, in RemoteTensorHandleData() function in tensorflow::RemoteTensorHandleData
101 RemoteTensorHandleData::RemoteTensorHandleData(int64_t op_id, int output_num, in RemoteTensorHandleData() function in tensorflow::RemoteTensorHandleData
117 RemoteTensorHandleData::~RemoteTensorHandleData() { in ~RemoteTensorHandleData()
125 Status RemoteTensorHandleData::Shape(TensorShape* shape) const { in Shape()
134 Status RemoteTensorHandleData::NumDims(int* num_dims) const { in NumDims()
143 Status RemoteTensorHandleData::Dim(int dim_index, int64_t* dim) const { in Dim()
152 Status RemoteTensorHandleData::NumElements(int64_t* num_elements) const { in NumElements()
161 bool RemoteTensorHandleData::IsReady() const { in IsReady()
166 void RemoteTensorHandleData::Poison(Status status) { in Poison()
172 Status RemoteTensorHandleData::IsPoisoned() const { in IsPoisoned()
[all …]
|
| H A D | remote_tensor_handle_data.h | 25 class RemoteTensorHandleData {
34 RemoteTensorHandleData(int64_t op_id, int output_num, uint64 context_view_id,
38 RemoteTensorHandleData(int64_t op_id, int output_num,
40 ~RemoteTensorHandleData();
|
| /aosp_15_r20/external/tensorflow/tensorflow/core/common_runtime/eager/ |
| H A D | tensor_handle.h | 319 std::unordered_map<string, RemoteTensorHandleData> resource_shape_mirrors_
323 std::unordered_map<string, RemoteTensorHandleData> remote_mirrors_
386 RemoteTensorHandleData>
|
| H A D | tensor_handle.cc | 389 data_(absl::in_place_type<RemoteTensorHandleData>, op_id, output_num, in TensorHandle()
412 data_(absl::in_place_type<RemoteTensorHandleData>, op_id, output_num, in TensorHandle()
730 auto& data = absl::get<RemoteTensorHandleData>(data_); in RemoteAddress()
854 auto& data = absl::get<RemoteTensorHandleData>(data_); in SetRemoteShapeAndDevice()
895 auto& data = absl::get<RemoteTensorHandleData>(data_); in PoisonRemote()
|