| 1459 | } |
| 1460 | |
| 1461 | const TensorMetaData* RdmaMemoryMgr::SetTensorMetaData( |
| 1462 | const std::string& tensor_name, DataType dtype, const TensorShape& shape, |
| 1463 | bool is_dead, size_t proto_size) { |
| 1464 | mutex_lock l(tensor_meta_data_mu_); |
| 1465 | TensorMetaData& meta_data = tensors_meta_data_[tensor_name]; |
| 1466 | meta_data.data_type_ = dtype; |
| 1467 | meta_data.tensor_shape_ = shape; |
| 1468 | meta_data.proto_size_ = proto_size; |
| 1469 | meta_data.is_dead_ = is_dead; |
| 1470 | return &meta_data; |
| 1471 | } |
| 1472 | |
| 1473 | //***************************************************************************** |
| 1474 | // RdmaTensorRequest |