Searched refs:user_gte (Results 1 – 1 of 1) sorted by relevance
3050 auto user_gte = loop->users().front(); in MoveUsersIntoWindowedDotGeneralLoopOnNonContractingDimensions() local3051 CHECK_EQ(user_gte->opcode(), HloOpcode::kGetTupleElement); in MoveUsersIntoWindowedDotGeneralLoopOnNonContractingDimensions()3052 CHECK_EQ(user_gte->tuple_index(), 2); in MoveUsersIntoWindowedDotGeneralLoopOnNonContractingDimensions()3062 Shape padded_shape = user_gte->shape(); in MoveUsersIntoWindowedDotGeneralLoopOnNonContractingDimensions()3063 Shape unpadded_shape = user_gte->shape(); in MoveUsersIntoWindowedDotGeneralLoopOnNonContractingDimensions()3064 auto original_output = user_gte; in MoveUsersIntoWindowedDotGeneralLoopOnNonContractingDimensions()3066 if (user_gte->user_count() == 1 && in MoveUsersIntoWindowedDotGeneralLoopOnNonContractingDimensions()3067 user_gte->users().back()->opcode() == HloOpcode::kSlice) { in MoveUsersIntoWindowedDotGeneralLoopOnNonContractingDimensions()3068 original_output = user_gte->users().back(); in MoveUsersIntoWindowedDotGeneralLoopOnNonContractingDimensions()