Searched defs:operand_no (Results 1 – 6 of 6) sorted by relevance
87 int64_t operand_no, bool mandatory, bool dfs, int64_t priority) in OperandLayoutConstraint()225 int64_t operand_no, bool mandatory, in SetOperandLayout()300 const Layout& layout, const HloInstruction* instruction, int64_t operand_no, in SetArrayOperandLayout()948 int64_t operand_no) { in CopyOperandIfLayoutsDiffer()1131 int64_t operand_no) { in ChooseOperandLayoutFromOutputLayout()1198 int64_t operand_no) { in ChooseOutputLayoutFromOperandLayout()1329 for (int64_t operand_no : in GetArrayUsesOfBuffer() local1451 for (int64_t operand_no = 0; operand_no < user->operand_count(); in PropagateOperandConstraint() local1464 for (int64_t operand_no = 0; operand_no < user->operand_count(); in PropagateOperandConstraint() local1590 for (int64_t operand_no = 0; operand_no < instruction->operand_count(); in PropagateBufferConstraintToOperands() local[all …]
113 const int64 operand_no() const { return operand_no_; } in operand_no() function
528 for (int64_t operand_no = 0; operand_no < instruction->operand_count(); in PropagateBufferConstraint() local
1744 int64_t operand_no = user->opcode() == HloOpcode::kTriangularSolve ? 1 : 0; in CanShareOperandBufferWithUser() local
730 int64_t operand_no) { in CheckIsTokenOperand()
129 for (int64_t operand_no = 0; operand_no < instruction->operand_count(); in AddBackendConstraints() local