Home
last modified time | relevance | path

Searched defs:shard_vec (Results 1 – 2 of 2) sorted by relevance

/external/tensorflow/tensorflow/dtensor/tests/
Dtensor_layout_test.cc341 ShardVector shard_vec; in TEST_F() local
361 ShardVector shard_vec = layout.GetShardVector(); in TEST_F() local
/external/tensorflow/tensorflow/dtensor/cc/
Dtensor_layout.cc82 ShardVector ExpandShardVector(const ShardVector& shard_vec, in ExpandShardVector()
840 ShardVector shard_vec; in GetShardVector() local