| 1449 | } |
| 1450 | |
| 1451 | const TensorMetaData* RdmaMemoryMgr::GetTensorMetaData( |
| 1452 | const std::string& tensor_name) { |
| 1453 | mutex_lock l(tensor_meta_data_mu_); |
| 1454 | auto it = tensors_meta_data_.find(tensor_name); |
| 1455 | if (it == tensors_meta_data_.end()) { |
| 1456 | return nullptr; |
| 1457 | } |
| 1458 | return &it->second; |
| 1459 | } |
| 1460 | |
| 1461 | const TensorMetaData* RdmaMemoryMgr::SetTensorMetaData( |
| 1462 | const std::string& tensor_name, DataType dtype, const TensorShape& shape, |
no test coverage detected