Home
last modified time | relevance | path

Searched refs:operand_number (Results 1 – 11 of 11) sorted by relevance

/external/tensorflow/tensorflow/compiler/xla/service/
Dhlo_value.cc59 string index_str = instruction->operand(operand_number)->shape().IsTuple() in ToString()
62 return StrCat(instruction->name(), ", operand ", operand_number, index_str); in ToString()
116 bool MayUseOperandValue(int64 operand_number, const ShapeIndex& index, in MayUseOperandValue() argument
124 CHECK_EQ(operand_number, 0); in MayUseOperandValue()
129 CHECK_GE(operand_number, 0); in MayUseOperandValue()
130 CHECK_LE(operand_number, 2); in MayUseOperandValue()
131 return operand_number == 0 || index.empty(); in MayUseOperandValue()
180 for (int64 operand_number : user->OperandIndices(position.instruction)) { in SetPositionsAndComputeUses() local
183 if (MayUseOperandValue(operand_number, position.index, user) || in SetPositionsAndComputeUses()
185 HloUse new_use{user, operand_number, position.index}; in SetPositionsAndComputeUses()
Dtuple_simplifier.cc80 for (int64 operand_number = 0; in Run() local
81 operand_number < instruction->operand_count(); ++operand_number) { in Run()
82 HloInstruction* operand = instruction->mutable_operand(operand_number); in Run()
84 operand->tuple_index() != operand_number) { in Run()
Dhlo_value.h69 int64 operand_number; member
78 operand_number == other.operand_number &&
Dbfloat16_propagation.cc249 use.instruction->fused_parameter(use.operand_number); in AllUsersConsumeBF16()
257 use.operand_number); in AllUsersConsumeBF16()
263 use.operand_number); in AllUsersConsumeBF16()
270 *use.instruction, use.operand_number)) { in AllUsersConsumeBF16()
277 *use.instruction, use.operand_number)) { in AllUsersConsumeBF16()
281 ShapeIndex use_output_index{use.operand_number}; in AllUsersConsumeBF16()
Dhlo_dataflow_analysis.cc693 for (int64 operand_number : user->OperandIndices(instruction)) { in Propagate() local
695 called_computation->parameter_instruction(operand_number)); in Propagate()
935 user->fused_parameter(use.operand_number); in DoesNotUseOperandBuffer()
1088 use.operand_number == other_add_operand_index; in CanShareOperandBufferWithUser()
1136 callee_root->IsElementwiseOnOperand(use.operand_number); in CanShareOperandBufferWithUser()
Dhlo_verifier.h150 int64 operand_number,
Dhlo_ordering.cc185 use.instruction->mutable_operand(use.operand_number), in UseIsBeforeValueDefinition()
Dlayout_assignment.h482 Status AddCopyForOperand(HloInstruction* instruction, int64 operand_number);
Dlayout_assignment.cc2128 int64 operand_number) { in AddCopyForOperand() argument
2129 HloInstruction* operand = instruction->mutable_operand(operand_number); in AddCopyForOperand()
2136 TF_RETURN_IF_ERROR(instruction->ReplaceOperandWith(operand_number, copy)); in AddCopyForOperand()
Dhlo_verifier.cc233 const HloInstruction* instruction, int64 operand_number, in CheckOperandAndParameter() argument
235 const HloInstruction* operand = instruction->operand(operand_number); in CheckOperandAndParameter()
/external/tensorflow/tensorflow/compiler/xla/tests/
Dtest_utils.cc373 const int64 op_num = use.operand_number; in NeedsInitValue()
400 const int64 op_num = use.operand_number; in FindConstrainedUses()