| 65 | } |
| 66 | |
| 67 | bool ShardableTensorMap::HasPartitionKey() const { |
| 68 | return !partition_key_.empty(); |
| 69 | } |
| 70 | |
| 71 | ShardsPtr<ShardableTensorMap> ShardableTensorMap::Partition() const { |
| 72 | // FIXME(@goldenleaves): Be careful, `n` should be initialized at runtime. |