Home
last modified time | relevance | path

Searched refs:branch_count (Results 1 – 24 of 24) sorted by relevance

/external/tensorflow/tensorflow/compiler/xla/service/gpu/
Dconditional_thunk.cc34 config.branch_count = instr->branch_count(); in GetConditionalThunkConfig()
96 if (branch_index < 0 || branch_index >= config_.branch_count) { in ExecuteOnStream()
97 branch_index = config_.branch_count - 1; in ExecuteOnStream()
Dconditional_thunk.h35 int64 branch_count; member
Dir_emitter_unnested.cc3819 for (int j = 0; j < conditional->branch_count(); ++j) { in CheckConditionalBuffersShareAllocation()
3904 int branch_count = hlo->branch_count(); in BuildConditionalThunk() local
3905 branch_thunks.reserve(branch_count); in BuildConditionalThunk()
3906 branch_profile_indices.reserve(branch_count); in BuildConditionalThunk()
3910 for (int j = 0; j < branch_count; ++j) { in BuildConditionalThunk()
/external/tensorflow/tensorflow/compiler/xla/service/
Dconditional_code_motion.cc309 int branch_count, in FindSpecialConverts() argument
322 for (int others = 1; others < branch_count; ++others) { in FindSpecialConverts()
392 int branch_count = conditional->branch_count(); in ConvertSpecialMove() local
393 if (branch_count <= 0) { in ConvertSpecialMove()
398 for (int branch_num = 0; branch_num < branch_count; ++branch_num) { in ConvertSpecialMove()
425 old_root, branch_count, conditional, is_layout_sensitive); in ConvertSpecialMove()
435 for (int branch = 0; branch < branch_count; branch++) { in ConvertSpecialMove()
593 int64 branch_count = conditional->branch_count(); in MoveInstructionOut() local
594 for (int i = 0; i < branch_count; i++) { in MoveInstructionOut()
646 int64 branch_count = conditional->branch_count(); in MoveInstructionIn() local
[all …]
Dconditional_simplifier.cc92 if (conditional->branch_count() == 1) { in TryRemoveConditional()
105 if (branch_index < 0 || branch_index >= conditional->branch_count()) { in TryRemoveConditional()
106 branch_index = conditional->branch_count() - 1; in TryRemoveConditional()
135 if (conditional->branch_count() != 2 || in TryRemoveConditional()
258 for (int64 branch = 0; branch < conditional->branch_count(); ++branch) { in TryRemoveUnusedConditionalOperands()
299 for (int64 branch_id = 0; branch_id < conditional_op->branch_count(); in ReplaceRootWithEmptyTupleIfNoUsers()
413 for (int branch_id = 0; branch_id < conditional_op->branch_count(); in RemoveUnusedTupleElements()
616 for (int64 branch = 0; branch < conditional->branch_count(); ++branch) { in Run()
Dflatten_call_graph.cc52 for (int b = 0; b < instruction->branch_count(); ++b) { in ReplaceCalledComputation()
53 if (b == instruction->branch_count() - 1) { in ReplaceCalledComputation()
Dhlo_ordering.cc97 for (int j = 0; j < a_ancestor->branch_count(); ++j) { in GetExecutionConstraint()
200 for (int j = 0; j < b.defining_instruction()->branch_count(); ++j) { in IsDefinedBefore()
332 for (int j = 0; j < conditional->branch_count(); ++j) { in UsesBeforeValueDefinition()
Dhlo_dataflow_analysis.cc516 std::vector<const InstructionValueSet*> inputs(conditional->branch_count()); in UpdateConditionalValueSet()
517 for (int j = 0; j < conditional->branch_count(); ++j) { in UpdateConditionalValueSet()
656 for (int j = 0; j < conditional->branch_count(); ++j) { in UpdateParameterValueSet()
881 for (int j = 0; j < user->branch_count(); ++j) { in Propagate()
Dlayout_assignment.cc614 for (int j = 1; j < instruction->branch_count(); ++j) { in AddMandatoryConstraints()
625 for (int k = 0; k < instruction->branch_count(); ++k) { in AddMandatoryConstraints()
627 int j = (k + largest_branch) % instruction->branch_count(); in AddMandatoryConstraints()
726 for (int j = 0; j < instruction->branch_count(); ++j) { in CheckConditionalLayout()
2131 for (int64 k = 0; k < caller->branch_count(); ++k) { in Run()
Dbfloat16_propagation.cc211 for (int64 i = 0; i < cond->branch_count(); ++i) { in DetermineConditionalComputationsPrecision()
510 for (int64 i = 0; i < hlo->branch_count(); ++i) { in AdjustCalledComputationParameters()
Dhlo_module_group_metadata.cc324 for (int b = 0; b < hlo->branch_count(); ++b) { in RecordInstructions()
Dhlo_module.cc196 for (int b = 0; b < instruction->branch_count(); ++b) { in ReplaceComputations()
Dhlo_verifier.cc950 const int num_branches = conditional->branch_count(); in HandleConditional()
1837 for (int b = 0; b < conditional->branch_count(); ++b) { in HandleConditional()
Dhlo_cost_analysis.cc989 for (int j = 1; j < conditional->branch_count(); ++j) { in HandleConditional()
Dhlo_evaluator.cc1981 if (branch_index < 0 || branch_index >= conditional->branch_count()) { in HandleConditional()
1982 branch_index = conditional->branch_count() - 1; in HandleConditional()
Dhlo_instruction.h1308 int branch_count() const;
Dhlo_instruction.cc1798 CHECK_EQ(new_operands.size(), branch_count() + 1); in CloneWithNewOperands()
2186 for (int j = 0; j < branch_count(); ++j) { in IdenticalSlowPath()
2517 int HloInstruction::branch_count() const { in branch_count() function in xla::HloInstruction
Ddynamic_dimension_inference.cc1266 for (int64 branch_index = 0; branch_index < hlo->branch_count(); in HandleConditional()
/external/python/jinja/src/jinja2/
Didtracking.py116 for name, branch_count in stores.items():
117 if branch_count == len(branch_symbols):
/external/libchrome/third_party/jinja2/
Didtracking.py117 for name, branch_count in iteritems(stores):
118 if branch_count == len(branch_symbols):
/external/vixl/test/aarch32/
Dtest-assembler-aarch32.cc4522 for (int branch_count = 1; branch_count < kMaxBranchCount; branch_count++) { in TEST_T32() local
4523 Label* targets = new Label[branch_count]; in TEST_T32()
4525 for (int i = 0; i < branch_count; i++) { in TEST_T32()
4529 for (int i = 0; i < branch_count; i++) { in TEST_T32()
4530 __ Bind(&targets[branch_count - i - 1]); in TEST_T32()
/external/tensorflow/tensorflow/compiler/mlir/xla/
Dhlo_function_importer.cc534 int num_branches = instruction->branch_count(); in ImportInstructionImpl()
/external/tensorflow/tensorflow/compiler/xla/service/spmd/
Dspmd_partitioner.cc2865 for (int64 i = 0; i < hlo->branch_count(); ++i) { in HandleConditional()
2877 for (int64 i = 0; i < hlo->branch_count(); ++i) { in HandleConditional()
/external/tensorflow/tensorflow/compiler/xla/service/cpu/
Dir_emitter.cc2635 int num_branches = conditional->branch_count(); in HandleConditional()