Searched defs:operand_no (Results 1 – 6 of 6) sorted by relevance
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | layout_assignment.cc | 106 int64_t operand_no, bool mandatory, bool dfs, int64_t priority) in OperandLayoutConstraint()
237 int64_t operand_no, bool mandatory, in SetOperandLayout()
300 const HloInstruction* instruction, int64_t operand_no, in InsertOperandLayoutConstraint()
331 const Layout& layout, const HloInstruction* instruction, int64_t operand_no, in SetArrayOperandLayout()
1051 int64_t operand_no) { in CopyOperandIfLayoutsDiffer()
1259 int64_t operand_no) { in ChooseOperandLayoutFromOutputLayout()
1335 int64_t operand_no) { in ChooseOutputLayoutFromOperandLayout()
1470 for (int64_t operand_no : in GetArrayUsesOfBuffer() local
1595 for (int64_t operand_no = 0; operand_no < user->operand_count(); in PropagateOperandConstraint() local
1608 for (int64_t operand_no = 0; operand_no < user->operand_count(); in PropagateOperandConstraint() local
[all …]
|
| D | layout_assignment.h | 116 const int64_t operand_no() const { return operand_no_; } in operand_no() function
|
| D | layout_assignment_test.cc | 491 for (int64_t operand_no = 0; operand_no < instruction->operand_count(); in PropagateBufferConstraint() local
|
| D | hlo_dataflow_analysis.cc | 1958 int64_t operand_no = user->opcode() == HloOpcode::kTriangularSolve ? 1 : 0; in CanShareOperandBufferWithUser() local
|
| D | hlo_verifier.cc | 746 int64_t operand_no) { in CheckIsTokenOperand()
|
| /external/tensorflow/tensorflow/compiler/xla/service/cpu/ |
| D | cpu_layout_assignment.cc | 130 for (int64_t operand_no = 0; operand_no < instruction->operand_count(); in AddBackendConstraints() local
|