Home
last modified time | relevance | path

Searched refs:operand_count (Results 1 – 25 of 90) sorted by relevance

1234

/external/tensorflow/tensorflow/compiler/xla/service/
Dstable_sort_expander.cc86 sort->operand_count() * 2, scalar_shape, in ExpandInstruction()
87 absl::StrCat("p.", sort->operand_count(), ".lhs"))); in ExpandInstruction()
90 sort->operand_count() * 2 + 1, scalar_shape, in ExpandInstruction()
91 absl::StrCat("p.", sort->operand_count(), ".rhs"))); in ExpandInstruction()
101 std::vector<Shape> new_shapes = sort->operand_count() == 1 in ExpandInstruction()
114 tuple_elements.reserve(sort->operand_count()); in ExpandInstruction()
115 for (int64 i = 0; i < sort->operand_count(); ++i) { in ExpandInstruction()
126 iota_index = sort->operand_count() - 1; in ExpandInstruction()
164 if (inst->operand_count() == 0) { in ExpandInstruction()
168 new_operands.reserve(inst->operand_count()); in ExpandInstruction()
Ddfs_hlo_visitor_with_default_test.cc43 TF_RET_CHECK(!(hlo->IsElementwise() && hlo->operand_count() == 2)) in TEST_F()
45 TF_RET_CHECK(!(hlo->IsElementwise() && hlo->operand_count() == 1)) in TEST_F()
52 TF_RET_CHECK(hlo->IsElementwise() && hlo->operand_count() == 2) in TEST_F()
58 TF_RET_CHECK(hlo->IsElementwise() && hlo->operand_count() == 1) in TEST_F()
Dsort_simplifier.cc56 for (int64 i = 0; i < sort->operand_count() * 2; ++i) { in RemoveUnusedOperandFromSort()
64 if (used_indices.size() == sort->operand_count()) { in RemoveUnusedOperandFromSort()
71 for (int64 i = 0; i < sort->operand_count(); ++i) { in RemoveUnusedOperandFromSort()
86 for (int64 i = 0; i < sort->operand_count(); ++i) { in RemoveUnusedOperandFromSort()
115 for (int64 i = 0; i < sort->operand_count(); ++i) { in RemoveUnusedOperandFromSort()
Dbfloat16_normalization.cc263 std::vector<PrimitiveType> operand_types(hlo->operand_count()); in HandleMultipleOutputs()
264 std::vector<PrimitiveType> output_types(hlo->operand_count()); in HandleMultipleOutputs()
269 for (int64 i = 0; i < hlo->operand_count(); ++i) { in HandleMultipleOutputs()
310 for (int64 i = 0; i < hlo->operand_count(); ++i) { in HandleMultipleOutputs()
348 std::vector<HloInstruction*> output_elements(hlo->operand_count()); in HandleMultipleOutputs()
350 for (int64 i = 0; i < hlo->operand_count(); ++i) { in HandleMultipleOutputs()
387 for (int64 i = 0; i < hlo->operand_count(); ++i) { in HandleInstruction()
422 for (int i = 0; i < hlo->operand_count(); ++i) { in HandleInstruction()
458 for (int i = 0; i < hlo->operand_count(); ++i) { in HandleInstruction()
473 for (int i = 0; i < hlo->operand_count(); ++i) { in HandleInstruction()
[all …]
Dstable_sort_expander_test.cc38 if (a->opcode() != b->opcode() || a->operand_count() != b->operand_count()) { in IsSameComputationExceptParams()
46 if (a->operand_count() == 0) { in IsSameComputationExceptParams()
50 for (int64 i = 0; i < a->operand_count(); ++i) { in IsSameComputationExceptParams()
Dhlo_verifier.cc61 if (hlo->operand_count() != expected) { in CheckOperandCount()
308 ? hlo->operand_count() in HandleAllToAll()
320 TF_RET_CHECK(hlo->operand_count() == 1); in HandleAllToAll()
543 if (sort->operand_count() < 1) { in HandleSort()
561 CheckParameterCount(sort, compare, sort->operand_count() * 2)); in HandleSort()
583 for (int64 operand = 1; operand < sort->operand_count(); ++operand) { in HandleSort()
660 if (reduce->operand_count() % 2 != 0) { in HandleReduce()
722 dynamic_reshape->operand_count()); in HandleDynamicReshape()
723 for (int64 i = 1; i < dynamic_reshape->operand_count(); ++i) { in HandleDynamicReshape()
763 if (fused_parameters.size() != fusion->operand_count()) { in HandleFusion()
[all …]
Ddynamic_dimension_inference.cc302 if (hlo->operand_count() > 2) { in HandleCustomCall()
416 int64 operand_count = reduce->operand_count(); in HandleReduce() local
417 bool is_variadic_reduce = operand_count > 2; in HandleReduce()
418 CHECK_EQ(operand_count % 2, 0); in HandleReduce()
419 if (operand_index >= operand_count / 2) { in HandleReduce()
439 for (int64 i = 0; i < operand_count / 2; ++i) { in HandleReduce()
599 for (int64 i = 0; i < hlo->operand_count(); ++i) { in HandleConcatenate()
1276 int64 operand_count = in HandleConditional() local
1288 for (int64 i = 0; i < tuple_operand->operand_count(); ++i) { in HandleConditional()
1300 operand_count++; in HandleConditional()
[all …]
Dconditional_code_motion.cc277 index < root->operand_count()); in CopyInOrOutOfConditional()
313 for (int64 operand_num = 0; operand_num < old_root->operand_count(); in FindSpecialConverts()
438 std::vector<HloInstruction*> new_operands(old_root->operand_count()); in ConvertSpecialMove()
441 for (int64 operand_num = 0; operand_num < old_root->operand_count(); in ConvertSpecialMove()
446 for (int64 operand_num = 0; operand_num < old_root->operand_count(); in ConvertSpecialMove()
455 int64 new_tuple_count = old_root->operand_count(); in ConvertSpecialMove()
678 for (int i = 0; i < old_root->operand_count(); ++i) { in MoveInstructionIn()
945 : (instruction->operand_count() > 0) in WorthHoisting()
974 CHECK(index < tuple->operand_count()); in ReusesBeforeBoundary()
1190 int64 operand_count = (b.IsInsideBranch()) in AddBoundaries() local
[all …]
Dslice_sinker.cc95 for (int64 i = 0; i < candidate->operand_count(); ++i) { in IsSimilarOperationOnSlices()
251 if (!instruction->IsElementwise() || instruction->operand_count() == 0 || in Run()
Dhlo_cost_analysis.cc59 for (int64 i = 0; i < hlo->operand_count(); ++i) { in Preprocess()
293 for (int i = 0; i < tuple->operand_count(); ++i) { in HandleTuple()
316 for (int i = 0; i < domain->operand_count(); ++i) { in HandleDomain()
355 for (int64 i = 0; i < outfeed->operand_count(); ++i) { in HandleOutfeed()
548 for (int i = 0; i < token->operand_count(); ++i) { in HandleAfterAll()
562 for (int i = 0; i < add_dependency->operand_count(); ++i) { in HandleAddDependency()
944 for (int i = 0; i < custom_call->operand_count(); ++i) { in HandleCustomCall()
1103 for (int operand_number = 0; operand_number < hlo.operand_count(); in GetBytesRead()
Dbfloat16_conversion_folding.cc217 crs->operand_count()); in HandleAllReduce()
225 for (int64 i = 0; i < crs->operand_count(); ++i) { in HandleAllReduce()
Dconvert_operand_folding.cc50 for (int i = 0; i < instruction->operand_count(); ++i) { in ExpandInstruction()
Dlayout_assignment.cc370 for (int64 i = 0; i < instruction->operand_count(); ++i) { in ToString()
486 for (int64 i = 0; i < custom_call->operand_count(); ++i) { in AddMandatoryConstraints()
558 TF_RET_CHECK(instruction->operand_count() == in AddMandatoryConstraints()
560 for (int64 i = 0; i < instruction->operand_count(); ++i) { in AddMandatoryConstraints()
577 CHECK_EQ(1, instruction->operand_count()); in AddMandatoryConstraints()
677 TF_RET_CHECK(computation->num_parameters() == call->operand_count()); in CheckCallLayout()
693 for (int64 i = 0; i < custom_call->operand_count(); ++i) { in CheckCustomCallLayout()
754 for (int64 i = 0; i < fusion->operand_count(); ++i) { in CheckFusionLayout()
1328 user->operand_count() == 1 in PropagateOperandConstraint()
1363 for (int64 operand_no = 0; operand_no < user->operand_count(); in PropagateOperandConstraint()
[all …]
Dtranspose_folding.cc73 for (int64 i = 0; i < dot.operand_count(); ++i) { in CanFoldOperandsIntoDot()
94 for (int64 i = 0; i < convolution.operand_count(); ++i) { in CanFoldOperandsIntoConvolution()
/external/google-breakpad/src/third_party/libdisasm/
Dx86_operand_list.c14 insn->operand_count = 1; in x86_oplist_append()
29 insn->operand_count = insn->operand_count + 1; in x86_oplist_append()
62 insn->operand_count = 0; in x86_oplist_free()
159 return insn->operand_count; in x86_operand_count()
/external/tensorflow/tensorflow/compiler/xla/service/gpu/
Dnccl_collective_thunk.h68 int64 operand_count; member
81 config.operand_count = op.operands().size(); in GetNcclCollectiveConfigForMlir()
82 config.operand_element_type.reserve(config.operand_count); in GetNcclCollectiveConfigForMlir()
83 for (int i = 0; i < config.operand_count; i++) { in GetNcclCollectiveConfigForMlir()
Dgpu_copy_insertion.cc51 for (int64 i = hlo->operand_count() - 2; i < hlo->operand_count(); ++i) { in Run()
Dgpu_fusible.cc218 ((instr.IsElementwise() && instr.operand_count() > 0) || in IsLoopFusible()
406 if (instr1.operand_count() + instr2.operand_count() - 1 + in FusionWouldBeTooLarge()
412 << "(" << instr1.ToString() << " ) = " << instr1.operand_count() in FusionWouldBeTooLarge()
414 << " ) = " << instr2.operand_count() in FusionWouldBeTooLarge()
Dcudnn_pad_for_convolutions.cc81 new_operands.reserve(conv->operand_count()); in PadConv()
82 for (int i = 0; i < conv->operand_count(); ++i) { in PadConv()
89 for (int i = 0; i < conv->operand_count(); ++i) { in PadConv()
369 for (int64 i = 0; i < conv->operand_count(); ++i) { in TryResolvePaddedShapesForIntegerConvolution()
376 for (int64 i = 0; i < conv->operand_count(); ++i) { in TryResolvePaddedShapesForIntegerConvolution()
Dgpu_hlo_schedule.cc111 CHECK_EQ(hlo->operand_count(), 0); in GpuHloOrdering()
164 if (hlo->operand_count() == 0) { in BFSLaunchOrder()
Dalias_passthrough_params.cc32 for (int64 i = 0; i < root->operand_count(); ++i) { in Run()
/external/tensorflow/tensorflow/compiler/xla/service/spmd/
Dschedule_aware_all_gather_cse_test.cc62 EXPECT_EQ(tuple->operand_count(), 2); in TEST_F()
83 EXPECT_EQ(tuple->operand_count(), 2); in TEST_F()
104 EXPECT_EQ(tuple->operand_count(), 2); in TEST_F()
125 EXPECT_EQ(tuple->operand_count(), 2); in TEST_F()
/external/llvm-project/lldb/source/Plugins/SymbolFile/DWARF/
DDWARFDebugMacro.cpp50 uint64_t operand_count = debug_macro_data.GetULEB128(offset); in SkipOperandTable() local
52 for (uint64_t j = 0; j < operand_count; j++) { in SkipOperandTable()
/external/tensorflow/tensorflow/compiler/xla/service/cpu/
Dcpu_layout_assignment.cc119 for (int i = 0; i < instruction->operand_count(); i++) { in AddBackendConstraints()
129 for (int64 operand_no = 0; operand_no < instruction->operand_count(); in AddBackendConstraints()
/external/tensorflow/tensorflow/compiler/xla/service/cpu/tests/
Dcpu_fusion_test.cc278 EXPECT_EQ(0, fusion1->operand_count()); in TEST_F()
279 EXPECT_EQ(0, fusion2->operand_count()); in TEST_F()
333 EXPECT_EQ(0, fusion_inst->operand_count()); in TEST_F()

1234