Home
last modified time | relevance | path

Searched refs:tuple_index (Results 1 – 25 of 54) sorted by relevance

123

/external/tensorflow/tensorflow/compiler/xla/service/
Dwhile_loop_all_reduce_code_motion.cc90 std::vector<int> tuple_index; in IsAllReduceMovable() member
114 if (!result.tuple_index.empty()) { in IsAllReduceMovable()
118 result.tuple_index.push_back( in IsAllReduceMovable()
120 ->tuple_index()); in IsAllReduceMovable()
171 if (!result.tuple_index.empty()) { in IsAllReduceMovable()
175 result.tuple_index.push_back(user->operand_index(instruction)); in IsAllReduceMovable()
212 int tuple_index = accumulation.param_tuple_indices[0]; in IsAllReduceMovable() local
224 if (gte->tuple_index() == tuple_index) { in IsAllReduceMovable()
321 !origin_buffer_tuple_index.tuple_index.empty() && in IsAllReduceMovable()
322 ContainersEqual(origin_buffer_tuple_index.tuple_index, in IsAllReduceMovable()
[all …]
Dwhile_loop_constant_sinking.cc30 HloInstruction* while_body_root, int64 tuple_index) { in ReplaceUsesWhileKeepingLoopInvariance() argument
40 !(user == while_body_root && i == tuple_index)) { in ReplaceUsesWhileKeepingLoopInvariance()
68 int64 index = invariant_body_gte->tuple_index(); in TrySinkingConstantsIntoWhileLoop()
Dhlo_module_dce_test.cc48 const int64 tuple_index) { in WhileBodyHasPassThroughTupleElement() argument
58 auto* operand = while_body_root->operand(tuple_index); in WhileBodyHasPassThroughTupleElement()
60 operand->tuple_index() == tuple_index && in WhileBodyHasPassThroughTupleElement()
Dwhile_loop_simplifier.cc98 int64 old_idx = user->tuple_index(); in RemoveDeadTupleIndices()
279 while_body_root->operand_index(user) == user->tuple_index() && in TryRemoveDeadWhileParams()
284 used_tuple_indices.insert(user->tuple_index()); in TryRemoveDeadWhileParams()
304 operand->tuple_index() != i) { in TryRemoveDeadWhileParams()
335 HloInstruction* while_op, const int64 tuple_index, in TryRemoveRepeatedWhileTupleIndicesHelper() argument
350 comp->parameter_instruction(0)->shape().tuple_shapes(tuple_index), in TryRemoveRepeatedWhileTupleIndicesHelper()
351 comp->parameter_instruction(0), tuple_index)); in TryRemoveRepeatedWhileTupleIndicesHelper()
356 duplicates.contains(instr->tuple_index()) && in TryRemoveRepeatedWhileTupleIndicesHelper()
429 if (pivot_body_elem->tuple_index() != index_to_investigate) { in TryRemoveRepeatedWhileTupleIndices()
431 << pivot_body_elem->tuple_index() << " index_to_investigate " in TryRemoveRepeatedWhileTupleIndices()
[all …]
Dwhile_loop_analysis_test.cc198 EXPECT_EQ(aux_indices[0]->tuple_index(), 1); in TEST_F()
272 EXPECT_EQ(aux_indices[0]->tuple_index(), 1); in TEST_F()
Dconditional_simplifier.cc220 tuple_indices_to_keep.insert(user->tuple_index()); in TryRemoveUnusedConditionalOperands()
249 user->set_tuple_index(map[user->tuple_index()]); in TryRemoveUnusedConditionalOperands()
359 used_indices[user->tuple_index()] = true; in RemoveUnusedTupleElements()
441 const int old_index = user->tuple_index(); in RemoveUnusedTupleElements()
547 if (user->tuple_index() == j) { in MergeDuplicateTupleElements()
Dsort_simplifier.cc51 used_indices.insert(user->tuple_index()); in RemoveUnusedOperandFromSort()
131 user->ReplaceAllUsesWith(result_map.at(user->tuple_index()))); in RemoveUnusedOperandFromSort()
Dar_crs_combiner.cc280 tuple->mutable_operand(instruction->tuple_index()), visited); in GetAllTuples()
392 return i1->tuple_index() == i2->tuple_index() || in InstructionsComputeSameValue()
393 TupleElementsComputeSameValue(operands1[0], i1->tuple_index(), in InstructionsComputeSameValue()
394 i2->tuple_index(), visited_pairs); in InstructionsComputeSameValue()
Dconditional_code_motion.cc274 int64 index = gte->tuple_index(); in CopyInOrOutOfConditional()
416 if (instr->tuple_index() == index) { in ConvertSpecialMove()
578 int64 index = tuple_opd->tuple_index(); in MoveInstructionOut()
656 ? tuple_use->tuple_index() in MoveInstructionIn()
719 user->tuple_index() > use_index) { in MoveInstructionIn()
720 user->set_tuple_index(user->tuple_index() - 1); in MoveInstructionIn()
797 conditional->shape().tuple_shapes(user->tuple_index()); in MoveInstructionIn()
973 auto index = tuple_gte->tuple_index(); in ReusesBeforeBoundary()
1010 if (index == tuple_opd->tuple_index()) { in ReusesAfterBoundary()
1076 const auto* tuple_operand = root->operand(gte->tuple_index()); in BenefitForMovingBoundaries()
Dwhile_loop_analysis.cc69 int64 tuple_idx = (*gte_it)->tuple_index(); in GetGTEOperandIndex()
119 auto it = extractions.find(indx_instr->tuple_index()); in GetAuxiliaryLoopInductionVars()
126 extractions.insert(std::make_pair(indx_instr->tuple_index(), indx_instr)); in GetAuxiliaryLoopInductionVars()
598 int64 indvar_index = cond_gte->tuple_index(); in ComputeWhileLoopTripCountUpperBound()
Dhlo_matchers.h77 int64 tuple_index) in HloGetTupleElementMatcher() argument
79 tuple_index_(tuple_index) {}
336 ::testing::Matcher<const HloInstruction*> operand, int64 tuple_index) { in GetTupleElement() argument
338 new ::xla::testing::HloGetTupleElementMatcher(operand, tuple_index)); in GetTupleElement()
Dtopk_rewriter.cc178 if (gte->tuple_index() == 0) { in Run()
180 } else if (gte->tuple_index() == 1) { in Run()
Dhlo_matchers.cc118 if (instruction->tuple_index() != tuple_index_) { in MatchAndExplain()
119 *listener << " has wrong tuple index (got " << instruction->tuple_index() in MatchAndExplain()
Dtuple_simplifier.cc44 operand->tuple_index() != operand_number) { in RemoveWholeTuple()
Dpattern_matcher.h1539 explicit constexpr HloInstructionPatternTupleIndexImpl(int64 tuple_index)
1540 : tuple_index_(tuple_index) {}
1562 if (inst->tuple_index() != tuple_index_) {
1916 constexpr auto WithTupleIndex(int64 tuple_index) const {
1917 return AppendImpl(HloInstructionPatternTupleIndexImpl(tuple_index));
2239 inline auto GetTupleElement(Arg&& arg, int64 tuple_index) {
2243 .WithTupleIndex(tuple_index);
2248 int64 tuple_index) {
2252 .WithTupleIndex(tuple_index);
Dwhile_util.cc282 instr->tuple_index() == i && in GetInvariantGTEsForWhileBody()
297 result[user->tuple_index()].push_back(user); in GetGTEsMapForWhileConditional()
Dtuple_points_to_analysis_test.cc787 HloInstruction* fusion_param, int64 tuple_index) { in GetFusionParameterUsersAt() argument
792 if (user->tuple_index() == tuple_index) { in GetFusionParameterUsersAt()
801 int64 tuple_index) { in GetUniqueFusionParameterUserAt() argument
803 GetFusionParameterUsersAt(fusion_param, tuple_index); in GetUniqueFusionParameterUserAt()
Dconditional_simplifier_test.cc482 EXPECT_EQ(gte_0->tuple_index(), 0); in TEST_F()
483 EXPECT_EQ(gte_1->tuple_index(), 0); in TEST_F()
Dhlo_cse.cc109 ? instruction->tuple_index() in CseHash()
Dbfloat16_conversion_folding.cc222 per_tuple_element_gtes[user->tuple_index()].push_back(user); in HandleAllReduce()
Dhlo_liveness_analysis.cc159 operand_shape_index.push_front(instruction->tuple_index()); in PropagateLivenessThroughGTE()
/external/tensorflow/tensorflow/compiler/xla/service/gpu/
Dwhile_transformer_test.cc38 const int64 tuple_index, const int64 limit) { in BuildConditionComputation() argument
43 0, GetLoopStateShape(tuple_index), "loop_state")); in BuildConditionComputation()
46 limit_const->shape(), loop_state, tuple_index)); in BuildConditionComputation()
Dhlo_to_ir_bindings.cc106 gte->shape(), gte->tuple_index(), /*alignment=*/1, in EmitGetTupleElement()
110 gte->shape(), gte->tuple_index(), /*alignment=*/1, in EmitGetTupleElement()
/external/python/cpython3/Objects/clinic/
Dtupleobject.c.h14 {"index", (PyCFunction)(void(*)(void))tuple_index, METH_FASTCALL, tuple_index__doc__},
21 tuple_index(PyTupleObject *self, PyObject *const *args, Py_ssize_t nargs) in tuple_index() function
/external/python/cpython3/Objects/
Dgenericaliasobject.c179 tuple_index(PyObject *self, Py_ssize_t len, PyObject *item) in tuple_index() function
192 if (tuple_index(self, len, item) < 0) { in tuple_add()
277 Py_ssize_t iparam = tuple_index(params, nparams, arg); in subs_tvars()
341 Py_ssize_t iparam = tuple_index(alias->parameters, nparams, arg); in ga_getitem()

123