Home
last modified time | relevance | path

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

/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/spmd/
H A Ddot_handler.cc3878 auto new_input_subtuple = in SinkInputNodesIntoWindowedDotGeneralLoopOnContractingDimensions() local
3881 non_windowed_operand_index, new_input_subtuple)); in SinkInputNodesIntoWindowedDotGeneralLoopOnContractingDimensions()
3892 new_input_subtuple->shape(); in SinkInputNodesIntoWindowedDotGeneralLoopOnContractingDimensions()
3897 new_input_subtuple->shape(), body_param, non_windowed_operand_index)); in SinkInputNodesIntoWindowedDotGeneralLoopOnContractingDimensions()
4153 auto* new_input_subtuple = in MoveUsersIntoWindowedDotGeneralLoopOnNonContractingDimensions() local
4156 input_tuple->ReplaceOperandWithDifferentShape(2, new_input_subtuple)); in MoveUsersIntoWindowedDotGeneralLoopOnNonContractingDimensions()
4166 new_input_subtuple->shape(); in MoveUsersIntoWindowedDotGeneralLoopOnNonContractingDimensions()
4170 new_input_subtuple->shape(), body_param, 2)); in MoveUsersIntoWindowedDotGeneralLoopOnNonContractingDimensions()