Searched refs:input_dim_indices (Results 1 – 3 of 3) sorted by relevance
1074 absl::Span<const int64> input_dim_indices) { in ReshapeLeavesDimensionsUnmodified() argument1075 CHECK(std::is_sorted(input_dim_indices.begin(), input_dim_indices.end())); in ReshapeLeavesDimensionsUnmodified()1081 for (int64 input_dim_index : input_dim_indices) { in ReshapeLeavesDimensionsUnmodified()
606 absl::Span<const int64> input_dim_indices);
2400 const HloInstruction* hlo, absl::Span<const int64> input_dim_indices) { in ReshapeLeavesDimensionsUnmodified() argument2403 hlo->operand(0)->shape(), hlo->shape(), input_dim_indices); in ReshapeLeavesDimensionsUnmodified()