Searched refs:operand_number (Results 1 – 11 of 11) sorted by relevance
/external/tensorflow/tensorflow/compiler/xla/service/ |
D | hlo_value.cc | 59 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()
|
D | tuple_simplifier.cc | 80 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()
|
D | hlo_value.h | 69 int64 operand_number; member 78 operand_number == other.operand_number &&
|
D | bfloat16_propagation.cc | 249 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()
|
D | hlo_dataflow_analysis.cc | 693 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()
|
D | hlo_verifier.h | 150 int64 operand_number,
|
D | hlo_ordering.cc | 185 use.instruction->mutable_operand(use.operand_number), in UseIsBeforeValueDefinition()
|
D | layout_assignment.h | 482 Status AddCopyForOperand(HloInstruction* instruction, int64 operand_number);
|
D | layout_assignment.cc | 2128 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()
|
D | hlo_verifier.cc | 233 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/ |
D | test_utils.cc | 373 const int64 op_num = use.operand_number; in NeedsInitValue() 400 const int64 op_num = use.operand_number; in FindConstrainedUses()
|