| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | hlo_module_dce_test.cc | 49 const int64 tuple_index) { in WhileBodyHasPassThroughTupleElement() argument
59 auto* operand = while_body_root->operand(tuple_index); in WhileBodyHasPassThroughTupleElement()
61 operand->tuple_index() == tuple_index && in WhileBodyHasPassThroughTupleElement()
|
| D | while_loop_constant_sinking.cc | 30 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()
|
| D | ar_crs_combiner.cc | 137 GetAllTuples(tuple->mutable_operand(instruction->tuple_index())); in GetAllTuples()
221 return i1->tuple_index() == i2->tuple_index() || in InstructionsComputeSameValue()
222 TupleElementsComputeSameValue(operands1[0], i1->tuple_index(), in InstructionsComputeSameValue()
223 i2->tuple_index(), visited_pairs); in InstructionsComputeSameValue()
|
| D | tuple_simplifier.cc | 84 operand->tuple_index() != operand_number) { in Run()
119 instruction->tuple_index()); in Run()
|
| D | sort_simplifier.cc | 51 used_indices.insert(user->tuple_index()); in RemoveUnusedOperandFromSort()
131 user->ReplaceAllUsesWith(result_map.at(user->tuple_index()))); in RemoveUnusedOperandFromSort()
|
| D | hlo_matchers.h | 77 int64 tuple_index) in HloGetTupleElementMatcher() argument
79 tuple_index_(tuple_index) {}
305 ::testing::Matcher<const HloInstruction*> operand, int64 tuple_index) { in GetTupleElement() argument
307 new ::xla::testing::HloGetTupleElementMatcher(operand, tuple_index)); in GetTupleElement()
|
| D | hlo_matchers.cc | 114 if (instruction->tuple_index() != tuple_index_) { in MatchAndExplain()
115 *listener << "has wrong tuple index (got " << instruction->tuple_index() in MatchAndExplain()
|
| D | while_loop_simplifier.cc | 111 while_body_root->operand_index(user) == user->tuple_index() && in TryRemoveDeadWhileParams()
116 used_tuple_indices.insert(user->tuple_index()); in TryRemoveDeadWhileParams()
136 operand->tuple_index() != i) { in TryRemoveDeadWhileParams()
201 int64 old_idx = user->tuple_index(); in TryRemoveDeadWhileParams()
553 VLOG(3) << "tuple index " << instr->tuple_index() << " " in TryPropagateConstant()
555 auto iter = index_to_constant.find(instr->tuple_index()); in TryPropagateConstant()
|
| D | pattern_matcher.h | 1553 explicit constexpr HloInstructionPatternTupleIndexImpl(int64 tuple_index)
1554 : tuple_index_(tuple_index) {}
1576 if (inst->tuple_index() != tuple_index_) {
1966 constexpr auto WithTupleIndex(int64 tuple_index) const -> decltype(
1967 this->AppendImpl(HloInstructionPatternTupleIndexImpl(tuple_index))) {
1968 return AppendImpl(HloInstructionPatternTupleIndexImpl(tuple_index));
2364 inline auto GetTupleElement(Arg&& arg, int64 tuple_index)
2367 .WithTupleIndex(tuple_index)) {
2371 .WithTupleIndex(tuple_index);
2376 int64 tuple_index)
[all …]
|
| D | while_util.cc | 267 instr->tuple_index() == i && in GetInvariantGTEsForWhileBody()
282 result[user->tuple_index()].push_back(user); in GetGTEsMapForWhileConditional()
|
| D | while_loop_analysis.cc | 67 int64 tuple_idx = (*gte_it)->tuple_index(); in GetGTEOperandIndex()
456 int64 indvar_index = cond_gte->tuple_index(); in ComputeWhileLoopTripCountUpperBound()
|
| D | hlo_cse.cc | 101 ? instruction->tuple_index() in CseHash()
|
| D | tuple_points_to_analysis_test.cc | 736 HloInstruction* fusion_param, int64 tuple_index) { in GetFusionParameterUsersAt() argument
741 if (user->tuple_index() == tuple_index) { in GetFusionParameterUsersAt()
750 int64 tuple_index) { in GetUniqueFusionParameterUserAt() argument
752 GetFusionParameterUsersAt(fusion_param, tuple_index); in GetUniqueFusionParameterUserAt()
|
| D | bfloat16_conversion_folding.cc | 211 per_tuple_element_gtes[user->tuple_index()].push_back(user); in HandleAllReduce()
|
| D | hlo_liveness_analysis.cc | 159 operand_shape_index.push_front(instruction->tuple_index()); in PropagateLivenessThroughGTE()
|
| D | hlo_computation_test.cc | 291 EXPECT_EQ(0, tuple_copy->operand(0)->operand(0)->tuple_index()); in TEST_F()
292 EXPECT_EQ(1, tuple_copy->operand(1)->operand(0)->tuple_index()); in TEST_F()
|
| D | hlo_sharding_metadata.cc | 277 ? sharding_tree.find({user->tuple_index()}) in ApplyShardingFromUsers()
|
| D | hlo_instructions.cc | 1514 int64 old_tuple_index = old_gte->tuple_index(); in CloneAndFuseInternal()
1709 return {StrCat("index=", tuple_index())}; in ExtraAttributesToStringImpl()
1718 return tuple_index() == casted_other.tuple_index(); in IdenticalSlowPath()
1727 shape, new_operands[0], tuple_index()); in CloneWithNewOperandsImpl()
|
| D | dynamic_dimension_inference.cc | 114 if (hlo->tuple_index() == index[0]) { in HandleGetTupleElement()
|
| D | tuple_points_to_analysis.cc | 223 int64 element_index = get_tuple_element->tuple_index(); in HandleGetTupleElement()
|
| /external/python/cpython3/Objects/clinic/ |
| D | tupleobject.c.h | 14 {"index", (PyCFunction)tuple_index, METH_FASTCALL, tuple_index__doc__},
21 tuple_index(PyTupleObject *self, PyObject *const *args, Py_ssize_t nargs) in tuple_index() function
|
| /external/tensorflow/tensorflow/compiler/xla/service/gpu/ |
| D | while_transformer_test.cc | 38 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()
|
| D | hlo_to_ir_bindings.cc | 137 gte->shape(), gte->tuple_index(), /*alignment=*/1, in EmitGetTupleElement()
141 gte->shape(), gte->tuple_index(), /*alignment=*/1, in EmitGetTupleElement()
|
| D | ir_emitter.cc | 115 get_tuple_element->shape(), get_tuple_element->tuple_index(), in HandleGetTupleElement()
|
| /external/tensorflow/tensorflow/compiler/xla/service/llvm_ir/ |
| D | fused_ir_emitter.cc | 125 get_tuple_element->tuple_index(), in HandleGetTupleElement()
|