Searched refs:first_reshape_operand (Results 1 – 1 of 1) sorted by relevance
137 HloInstruction* UpdateOperand(const HloInstruction* first_reshape_operand, in UpdateOperand() argument147 if (first_reshape_operand->opcode() == HloOpcode::kReshape) { in UpdateOperand()152 CHECK(first_reshape_operand->opcode() == HloOpcode::kTranspose); in UpdateOperand()155 InversePermutation(first_reshape_operand->dimensions()); in UpdateOperand()188 HloInstruction* instruction, const HloInstruction* first_reshape_operand) { in PerformSinkReshapeOrTranspose() argument191 const Shape& new_operand_shape = first_reshape_operand->operand(0)->shape(); in PerformSinkReshapeOrTranspose()195 << first_reshape_operand->ToString(print_no_metadata) in PerformSinkReshapeOrTranspose()210 UpdateOperand(first_reshape_operand, new_operand_shape, operands[i]); in PerformSinkReshapeOrTranspose()239 switch (first_reshape_operand->opcode()) { in PerformSinkReshapeOrTranspose()249 first_reshape_operand->dimensions()); in PerformSinkReshapeOrTranspose()[all …]