Searched refs:opt_shard_slice_shape (Results 1 – 3 of 3) sorted by relevance
347 ASSERT_EQ(tensor_layout.opt_shard_slice_shape(), slice_shape_expect); in TEST_F()368 ASSERT_EQ(tensor_layout.opt_shard_slice_shape(), slice_shape_expect); in TEST_F()
101 Shape opt_shard_slice_shape() { return opt_shard_slice_shape_; } in opt_shard_slice_shape() function
1689 slice_shape = tensor_layout.opt_shard_slice_shape(); in SetParallelShape()