Searched refs:non_windowed_operand_index (Results 1 – 1 of 1) sorted by relevance
2931 HloInstruction* loop, int64 non_windowed_operand_index) { in SinkInputNodesIntoWindowedDotGeneralLoopOnContractingDimensions() argument2933 auto old_operand = input_tuple->mutable_operand(non_windowed_operand_index); in SinkInputNodesIntoWindowedDotGeneralLoopOnContractingDimensions()2945 non_windowed_operand_index, new_input_subtuple)); in SinkInputNodesIntoWindowedDotGeneralLoopOnContractingDimensions()2955 {non_windowed_operand_index}) = in SinkInputNodesIntoWindowedDotGeneralLoopOnContractingDimensions()2961 new_input_subtuple->shape(), body_param, non_windowed_operand_index)); in SinkInputNodesIntoWindowedDotGeneralLoopOnContractingDimensions()2963 non_windowed_operand_index, new_operand_tuple_inside)); in SinkInputNodesIntoWindowedDotGeneralLoopOnContractingDimensions()3014 ou->tuple_index() == non_windowed_operand_index) { in SinkInputNodesIntoWindowedDotGeneralLoopOnContractingDimensions()