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.cc129 for (int64 operand_no = 0; operand_no < instruction->operand_count(); in AddBackendConstraints() local
130 ++operand_no) { in AddBackendConstraints()
132 if (constraints->OperandLayout(instruction, operand_no) != nullptr) { in AddBackendConstraints()
136 if (constraints->OperandBufferForwarded(instruction, operand_no)) { in AddBackendConstraints()
140 if (!instruction->operand(operand_no)->shape().IsArray()) { in AddBackendConstraints()
144 RowMajorShape(instruction->operand(operand_no)->shape())); in AddBackendConstraints()
146 operand_shape, instruction, operand_no)); in AddBackendConstraints()
/external/tensorflow/tensorflow/compiler/xla/service/
Dlayout_assignment.cc79 int64 operand_no, bool mandatory, bool dfs) in OperandLayoutConstraint() argument
83 operand_no_(operand_no) { in OperandLayoutConstraint()
86 instruction->operand(operand_no)->shape())) in OperandLayoutConstraint()
88 << instruction->operand(operand_no)->shape() << " (for operand " in OperandLayoutConstraint()
89 << operand_no << " of instruction " << instruction->ToString() << ")"; in OperandLayoutConstraint()
143 const HloInstruction* instruction, int64 operand_no) const { in OperandBufferForwarded()
148 GetBufferSet(instruction->operand(operand_no)); in OperandBufferForwarded()
200 int64 operand_no, bool mandatory, in SetOperandLayout() argument
203 << operand_no << " : " in SetOperandLayout()
207 GetOperandLayoutConstraint(instruction, operand_no); in SetOperandLayout()
[all …]
Dlayout_assignment.h95 const HloInstruction* instruction, int64 operand_no,
100 const int64 operand_no() const { return operand_no_; } in operand_no() function
160 int64 operand_no) const;
162 const HloInstruction* instruction, int64 operand_no) const;
171 const HloInstruction* instruction, int64 operand_no,
179 int64 operand_no, bool mandatory = true,
193 int64 operand_no) const;
364 int64 operand_no);
371 int64 operand_no);
471 int64 operand_no);
Dlayout_assignment_test.cc528 for (int64 operand_no = 0; operand_no < instruction->operand_count(); in PropagateBufferConstraint() local
529 ++operand_no) { in PropagateBufferConstraint()
530 const HloInstruction* operand = instruction->operand(operand_no); in PropagateBufferConstraint()
535 buffer_constraint.layout(), instruction, operand_no, in PropagateBufferConstraint()
Dhlo_verifier.h145 int64 operand_no);
Dhlo_verifier.cc221 int64 operand_no) { in CheckIsTokenOperand() argument
222 const HloInstruction* token = instruction->operand(operand_no); in CheckIsTokenOperand()
227 operand_no, StringifyShape(token->shape()), instruction->ToString()); in CheckIsTokenOperand()
/external/tensorflow/tensorflow/compiler/xla/service/gpu/
Dgpu_layout_assignment.cc284 layout_constraint.operand_no() == 0) { in PropagateOperandConstraint()
296 layout_constraint.operand_no() == 0) { in PropagateOperandConstraint()
310 (layout_constraint.operand_no() == 0 || in PropagateOperandConstraint()
311 layout_constraint.operand_no() == 4)) { in PropagateOperandConstraint()
318 int64 operand_to_set = layout_constraint.operand_no() == 0 ? 4 : 0; in PropagateOperandConstraint()