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.cc3316 auto new_input_subtuple = in SinkInputNodesIntoWindowedDotGeneralLoopOnContractingDimensions() local
3319 non_windowed_operand_index, new_input_subtuple)); in SinkInputNodesIntoWindowedDotGeneralLoopOnContractingDimensions()
3330 new_input_subtuple->shape(); in SinkInputNodesIntoWindowedDotGeneralLoopOnContractingDimensions()
3335 new_input_subtuple->shape(), body_param, non_windowed_operand_index)); in SinkInputNodesIntoWindowedDotGeneralLoopOnContractingDimensions()
3547 auto new_input_subtuple = in MoveUsersIntoWindowedDotGeneralLoopOnNonContractingDimensions() local
3550 input_tuple->ReplaceOperandWithDifferentShape(2, new_input_subtuple)); in MoveUsersIntoWindowedDotGeneralLoopOnNonContractingDimensions()
3560 new_input_subtuple->shape(); in MoveUsersIntoWindowedDotGeneralLoopOnNonContractingDimensions()
3564 new_input_subtuple->shape(), body_param, 2)); in MoveUsersIntoWindowedDotGeneralLoopOnNonContractingDimensions()