Searched refs:input_layout_specs (Results 1 – 1 of 1) sorted by relevance
540 std::vector<std::string> input_layout_specs; in ComputeLayoutBackward() local545 input_layout_specs.push_back(multiple == 1 in ComputeLayoutBackward()550 Layout::GetLayout(input_layout_specs, mesh)); in ComputeLayoutBackward()944 std::vector<std::string> input_layout_specs; in ComputeLayoutBackward() local948 input_layout_specs.push_back(output_layout.sharding_spec(dim_in_output)); in ComputeLayoutBackward()951 Layout::GetLayout(input_layout_specs, mesh)); in ComputeLayoutBackward()