Home
last modified time | relevance | path

Searched refs:new_input_subtuple (Results 1 – 1 of 1) sorted by relevance

/external/tensorflow/tensorflow/compiler/xla/service/spmd/
Ddot_handler.cc2942 auto new_input_subtuple = in SinkInputNodesIntoWindowedDotGeneralLoopOnContractingDimensions() local
2945 non_windowed_operand_index, new_input_subtuple)); in SinkInputNodesIntoWindowedDotGeneralLoopOnContractingDimensions()
2956 new_input_subtuple->shape(); in SinkInputNodesIntoWindowedDotGeneralLoopOnContractingDimensions()
2961 new_input_subtuple->shape(), body_param, non_windowed_operand_index)); in SinkInputNodesIntoWindowedDotGeneralLoopOnContractingDimensions()
3174 auto new_input_subtuple = in MoveUsersIntoWindowedDotGeneralLoopOnNonContractingDimensions() local
3177 input_tuple->ReplaceOperandWithDifferentShape(2, new_input_subtuple)); in MoveUsersIntoWindowedDotGeneralLoopOnNonContractingDimensions()
3187 new_input_subtuple->shape(); in MoveUsersIntoWindowedDotGeneralLoopOnNonContractingDimensions()
3191 new_input_subtuple->shape(), body_param, 2)); in MoveUsersIntoWindowedDotGeneralLoopOnNonContractingDimensions()