Searched refs:reduce_outside (Results 1 – 1 of 1) sorted by relevance
3290 auto reduce_outside = reduce_outputs[i]; in MoveUsersIntoWindowedDotGeneralLoopOnNonContractingDimensions() local3291 CHECK_EQ(reduce_outside->opcode(), HloOpcode::kReduce); in MoveUsersIntoWindowedDotGeneralLoopOnNonContractingDimensions()3294 auto operand0 = outside_to_inside[reduce_outside->operand(0)]; in MoveUsersIntoWindowedDotGeneralLoopOnNonContractingDimensions()3295 auto operand1 = outside_to_inside[reduce_outside->operand(1)]; in MoveUsersIntoWindowedDotGeneralLoopOnNonContractingDimensions()3299 reduce_outside->dimensions(), in MoveUsersIntoWindowedDotGeneralLoopOnNonContractingDimensions()3300 reduce_outside->to_apply()->ComputeProgramShape())); in MoveUsersIntoWindowedDotGeneralLoopOnNonContractingDimensions()3301 *reduce_shape.mutable_layout() = reduce_outside->shape().layout(); in MoveUsersIntoWindowedDotGeneralLoopOnNonContractingDimensions()3309 if (absl::c_linear_search(reduce_outside->dimensions(), i)) { in MoveUsersIntoWindowedDotGeneralLoopOnNonContractingDimensions()3310 if (reduce_outside->operand(0)->shape().dimensions(i) != in MoveUsersIntoWindowedDotGeneralLoopOnNonContractingDimensions()3333 reduce_outside->operand(0)->shape().dimensions(dim)))), in MoveUsersIntoWindowedDotGeneralLoopOnNonContractingDimensions()[all …]