Home
last modified time | relevance | path

Searched refs:operand_no (Results 1 – 7 of 7) sorted by relevance

/external/tensorflow/tensorflow/compiler/xla/service/cpu/
Dcpu_layout_assignment.cc130 for (int64_t operand_no = 0; operand_no < instruction->operand_count(); in AddBackendConstraints() local
131 ++operand_no) { in AddBackendConstraints()
133 if (constraints->OperandLayout(instruction, operand_no) != nullptr) { in AddBackendConstraints()
137 if (AnyOperandBufferForwarded(instruction, operand_no)) { in AddBackendConstraints()
141 if (!instruction->operand(operand_no)->shape().IsArray()) { in AddBackendConstraints()
145 RowMajorShape(instruction->operand(operand_no)->shape())); in AddBackendConstraints()
147 SetOperandLayout(operand_shape, instruction, operand_no)); in AddBackendConstraints()
/external/tensorflow/tensorflow/compiler/xla/service/
Dlayout_assignment.cc106 int64_t operand_no, bool mandatory, bool dfs, int64_t priority) in OperandLayoutConstraint() argument
110 operand_no_(operand_no) { in OperandLayoutConstraint()
113 instruction->operand(operand_no)->shape())) in OperandLayoutConstraint()
115 << instruction->operand(operand_no)->shape() << " (for operand " in OperandLayoutConstraint()
116 << operand_no << " of instruction " << instruction->ToString() << ")"; in OperandLayoutConstraint()
156 const HloInstruction* instruction, int64_t operand_no) const { in AnyOperandBufferForwarded()
161 GetBufferSet(instruction->operand(operand_no)); in AnyOperandBufferForwarded()
168 const HloInstruction* instruction, int64_t operand_no) const { in AllOperandBuffersForwarded()
173 GetBufferSet(instruction->operand(operand_no)); in AllOperandBuffersForwarded()
237 int64_t operand_no, bool mandatory, in SetOperandLayout() argument
[all …]
Dlayout_assignment.h111 const HloInstruction* instruction, int64_t operand_no,
116 const int64_t operand_no() const { return operand_no_; } in operand_no() function
277 int64_t operand_no) const;
279 const HloInstruction* instruction, int64_t operand_no) const;
282 const HloInstruction* instruction, int64_t operand_no,
335 int64_t operand_no, bool mandatory = true,
337 return SetArrayOperandLayout(layout, instruction, operand_no, mandatory,
342 int64_t operand_no, bool mandatory, bool dfs,
375 const HloInstruction* instruction, int64_t operand_no,
377 return SetOperandLayout(shape_with_layout, instruction, operand_no,
[all …]
Dlayout_assignment_test.cc491 for (int64_t operand_no = 0; operand_no < instruction->operand_count(); in PropagateBufferConstraint() local
492 ++operand_no) { in PropagateBufferConstraint()
493 const HloInstruction* operand = instruction->operand(operand_no); in PropagateBufferConstraint()
498 instruction, operand_no, in PropagateBufferConstraint()
Dhlo_verifier.h284 int64_t operand_no);
Dhlo_verifier.cc746 int64_t operand_no) { in CheckIsTokenOperand() argument
747 const HloInstruction* token = instruction->operand(operand_no); in CheckIsTokenOperand()
752 operand_no, StringifyShape(token->shape()), instruction->ToString()); in CheckIsTokenOperand()
Dhlo_dataflow_analysis.cc1958 int64_t operand_no = user->opcode() == HloOpcode::kTriangularSolve ? 1 : 0; in CanShareOperandBufferWithUser() local
1959 return operand_indices.size() == 1 && operand_indices[0] == operand_no; in CanShareOperandBufferWithUser()