Searched refs:sharded_subshape (Results 1 – 1 of 1) sorted by relevance
49 Shape sharded_subshape, in GetShardedShape()51 sharded_subshapes.emplace_back(std::move(sharded_subshape)); in GetShardedShape()176 const Shape& sharded_subshape = in DetermineArgumentLayoutsFromCompileOptions() local181 choose_compact_layout_for_shape_function(sharded_subshape)); in DetermineArgumentLayoutsFromCompileOptions()