Searched defs:operand_no (Results 1 – 6 of 6) sorted by relevance
84 int64 operand_no, bool mandatory, bool dfs) in OperandLayoutConstraint()210 int64 operand_no, bool mandatory, in SetOperandLayout()262 const Layout& layout, const HloInstruction* instruction, int64 operand_no, in SetArrayOperandLayout()861 int64 operand_no) { in CopyOperandIfLayoutsDiffer()1046 int64 operand_no) { in ChooseOperandLayoutFromOutputLayout()1113 int64 operand_no) { in ChooseOutputLayoutFromOperandLayout()1241 for (int64 operand_no : in GetArrayUsesOfBuffer() local1363 for (int64 operand_no = 0; operand_no < user->operand_count(); in PropagateOperandConstraint() local1376 for (int64 operand_no = 0; operand_no < user->operand_count(); in PropagateOperandConstraint() local1502 for (int64 operand_no = 0; operand_no < instruction->operand_count(); in PropagateBufferConstraintToOperands() local[all …]
102 const int64 operand_no() const { return operand_no_; } in operand_no() function
1347 int64 operand_no = user->opcode() == HloOpcode::kTriangularSolve ? 1 : 0; in CanShareOperandBufferWithUser() local
528 for (int64 operand_no = 0; operand_no < instruction->operand_count(); in PropagateBufferConstraint() local
396 int64 operand_no) { in CheckIsTokenOperand()
129 for (int64 operand_no = 0; operand_no < instruction->operand_count(); in AddBackendConstraints() local