Searched defs:sharding_specs (Results 1 – 11 of 11) sorted by relevance
211 std::vector<ShardingSpec> sharding_specs = pair.first; in VerifyOrFixLayout() local230 std::vector<ShardingSpec> sharding_specs(layout_rank); in GetSpecsFromLabelsAndMap() local360 std::vector<ShardingSpec> sharding_specs = in ComputeLayoutBackward() local
50 std::vector<ShardingSpec> sharding_specs(global_output_shape.size()); in ExpandOp() local
43 std::vector<ShardingSpec> sharding_specs( in MergeLayoutsForSplitOutput() local
435 std::vector<std::string> sharding_specs = layout.sharding_spec_strs(); in ComputeLayoutBackward() local
133 std::vector<std::string> sharding_specs(mask_layout.sharding_spec_strs()); in MergeLayouts() local
418 std::vector<ShardingSpec> sharding_specs(output_layout.rank()); in ExpandOp() local
969 std::vector<ShardingSpec> sharding_specs(input_layout->rank()); in RelayoutOneHotInput() local
39 repeated ShardingSpec sharding_specs = 1; field
321 absl::Span<const ShardingSpec> sharding_specs() const { in sharding_specs() function
664 std::vector<ShardingSpec> sharding_specs; in GetLayout() local675 const std::vector<ShardingSpec>& sharding_specs, const Mesh& mesh) { in GetLayout()
456 const std::vector<std::string> sharding_specs = layout.sharding_spec_strs(); in EmitHaloExchange() local