Searched defs:operand_no (Results 1 – 5 of 5) sorted by relevance
79 int64 operand_no, bool mandatory, bool dfs) in OperandLayoutConstraint()200 int64 operand_no, bool mandatory, in SetOperandLayout()252 const Layout& layout, const HloInstruction* instruction, int64 operand_no, in SetArrayOperandLayout()833 int64 operand_no) { in CopyOperandIfLayoutsDiffer()989 int64 operand_no) { in ChooseOperandLayoutFromOutputLayout()1055 int64 operand_no) { in ChooseOutputLayoutFromOperandLayout()1183 for (int64 operand_no : in GetArrayUsesOfBuffer() local1279 for (int64 operand_no = 0; operand_no < user->operand_count(); in PropagateOperandConstraint() local1390 for (int64 operand_no = 0; operand_no < instruction->operand_count(); in PropagateBufferConstraintToOperands() local1457 int64 operand_no = user_operand_no.second; in PropagateBufferConstraintToUses() local[all …]
100 const int64 operand_no() const { return operand_no_; } in operand_no() function
221 int64 operand_no) { in CheckIsTokenOperand()
528 for (int64 operand_no = 0; operand_no < instruction->operand_count(); in PropagateBufferConstraint() local
129 for (int64 operand_no = 0; operand_no < instruction->operand_count(); in AddBackendConstraints() local