Searched refs:output_inside (Results 1 – 1 of 1) sorted by relevance
3343 auto output_inside = in MoveUsersIntoWindowedDotGeneralLoopOnNonContractingDimensions() local3350 output_inside->shape(), last_iter_result, reduce_dus_offsets, in MoveUsersIntoWindowedDotGeneralLoopOnNonContractingDimensions()3351 output_inside->shape().dimensions())); in MoveUsersIntoWindowedDotGeneralLoopOnNonContractingDimensions()3352 output_inside = body->AddInstruction(HloInstruction::CreateBinary( in MoveUsersIntoWindowedDotGeneralLoopOnNonContractingDimensions()3353 output_inside->shape(), in MoveUsersIntoWindowedDotGeneralLoopOnNonContractingDimensions()3355 output_inside, input_slice)); in MoveUsersIntoWindowedDotGeneralLoopOnNonContractingDimensions()3358 if (!ShapeUtil::Compatible(output_inside->shape(), in MoveUsersIntoWindowedDotGeneralLoopOnNonContractingDimensions()3360 output_inside = in MoveUsersIntoWindowedDotGeneralLoopOnNonContractingDimensions()3362 last_iter_result->shape(), last_iter_result, output_inside, in MoveUsersIntoWindowedDotGeneralLoopOnNonContractingDimensions()3365 new_outputs_inside[index_in_operand] = output_inside; in MoveUsersIntoWindowedDotGeneralLoopOnNonContractingDimensions()