Searched refs:sparse_specs (Results 1 – 1 of 1) sorted by relevance
444 std::vector<ShardingSpec> sparse_specs = {internal_layout[0]}; in MaybeRelayoutInputs() local446 GetBroadcastedLayout(labels_global_shape, sparse_specs, in MaybeRelayoutInputs()