Searched defs:layout_specs (Results 1 – 5 of 5) sorted by relevance
38 std::vector<ShardingSpec> layout_specs(2); in GetSuggestedPredictionsLayout() local49 std::vector<ShardingSpec> layout_specs(layout.rank()); in MatchBatchDim() local145 std::vector<std::string> layout_specs(1); in ComputeLayoutForward() local
58 std::vector<ShardingSpec> layout_specs; in ComputeLayoutForward() local91 std::vector<ShardingSpec> layout_specs; in ComputeLayoutBackward() local
32 std::vector<ShardingSpec> layout_specs(input_layout.rank()); in GetSuggestedLayout() local
641 std::vector<ShardingSpec> layout_specs(2); in ComputeLayoutForward() local690 std::vector<ShardingSpec> layout_specs(2); in ComputeLayoutBackward() local
639 std::vector<std::string> layout_specs; in MakeLayoutForReshape() local