| /external/tensorflow/tensorflow/compiler/xla/service/llvm_ir/ |
| D | fused_ir_emitter.cc | 92 HloInstruction* get_tuple_element) { in HandleGetTupleElement() argument
94 auto operand = get_tuple_element->operand(0); in HandleGetTupleElement()
104 get_tuple_element->shape(), get_tuple_element->tuple_index(), in HandleGetTupleElement()
106 gte_values_.insert(std::make_pair(get_tuple_element, tuple_element_ptr)); in HandleGetTupleElement()
108 if (!ShapeUtil::IsTuple(get_tuple_element->shape())) { in HandleGetTupleElement()
109 generators_[get_tuple_element] = in HandleGetTupleElement()
112 return IrArray(tuple_element_ptr, get_tuple_element->shape()) in HandleGetTupleElement()
|
| D | fused_ir_emitter.h | 67 Status HandleGetTupleElement(HloInstruction* get_tuple_element) override;
|
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | tuple_points_to_analysis.cc | 223 HloInstruction* get_tuple_element) { in HandleGetTupleElement() argument
226 int64 element_index = get_tuple_element->tuple_index(); in HandleGetTupleElement()
228 PointsToSet& points_to_set = CreateEmptyPointsToSet(get_tuple_element); in HandleGetTupleElement()
230 *PerInst(get_tuple_element->operand(0))->points_to_set; in HandleGetTupleElement()
|
| D | dfs_hlo_visitor_with_default.h | 118 Status HandleGetTupleElement(HloInstructionPtr get_tuple_element) override { in HandleGetTupleElement() argument
119 return DefaultAction(get_tuple_element); in HandleGetTupleElement()
|
| D | logical_buffer_analysis.h | 60 Status HandleGetTupleElement(HloInstruction* get_tuple_element) override;
|
| D | hlo_verifier.cc | 150 Status ShapeVerifier::HandleGetTupleElement(HloInstruction* get_tuple_element) { in HandleGetTupleElement() argument
151 return CheckShape(get_tuple_element, in HandleGetTupleElement()
153 get_tuple_element->operand(0)->shape(), in HandleGetTupleElement()
154 get_tuple_element->tuple_index())); in HandleGetTupleElement()
|
| D | hlo_evaluator.h | 152 Status HandleGetTupleElement(HloInstruction* get_tuple_element) override;
|
| D | hlo_verifier.h | 53 Status HandleGetTupleElement(HloInstruction* get_tuple_element) override;
|
| D | tuple_points_to_analysis.h | 249 Status HandleGetTupleElement(HloInstruction* get_tuple_element) override;
|
| D | hlo_cost_analysis.h | 56 const HloInstruction* get_tuple_element) override;
|
| D | algebraic_simplifier.cc | 147 Status HandleGetTupleElement(HloInstruction* get_tuple_element) override;
1022 HloInstruction* get_tuple_element) { in HandleGetTupleElement() argument
1023 auto operand = get_tuple_element->mutable_operand(0); in HandleGetTupleElement()
1028 << get_tuple_element->ToString(); in HandleGetTupleElement()
1030 get_tuple_element, in HandleGetTupleElement()
1031 operand->mutable_operand(get_tuple_element->tuple_index()))) { in HandleGetTupleElement()
|
| D | hlo_evaluator.cc | 2405 Status HloEvaluator::HandleGetTupleElement(HloInstruction* get_tuple_element) { in HandleGetTupleElement() argument
2406 const auto result_shape = get_tuple_element->shape(); in HandleGetTupleElement()
2407 const int64 index = get_tuple_element->tuple_index(); in HandleGetTupleElement()
2409 auto operand = get_tuple_element->operand(0); in HandleGetTupleElement()
2420 evaluated_[get_tuple_element] = MakeUnique<Literal>( in HandleGetTupleElement()
2422 return evaluated_[get_tuple_element]->CopyFrom(operand_tuple_literal, in HandleGetTupleElement()
|
| D | tuple_points_to_analysis_test.cc | 252 auto get_tuple_element = builder.AddInstruction( in TEST_F() local
257 auto& points_to_set = points_to_analysis_->GetPointsToSet(get_tuple_element); in TEST_F()
|
| /external/tensorflow/tensorflow/compiler/xla/service/gpu/ |
| D | ir_emitter.cc | 112 Status IrEmitter::HandleGetTupleElement(HloInstruction* get_tuple_element) { in HandleGetTupleElement() argument
113 auto operand = get_tuple_element->operand(0); in HandleGetTupleElement()
116 *get_tuple_element, in HandleGetTupleElement()
118 get_tuple_element->shape(), get_tuple_element->tuple_index(), in HandleGetTupleElement()
|
| D | ir_emitter_unnested.h | 71 Status HandleGetTupleElement(HloInstruction* get_tuple_element) override;
|
| D | ir_emitter.h | 75 Status HandleGetTupleElement(HloInstruction* get_tuple_element) override;
|
| /external/tensorflow/tensorflow/compiler/xla/service/cpu/ |
| D | ir_emitter.h | 123 Status HandleGetTupleElement(HloInstruction* get_tuple_element) override;
|
| D | ir_emitter.cc | 300 Status IrEmitter::HandleGetTupleElement(HloInstruction* get_tuple_element) { in HandleGetTupleElement() argument
305 auto operand = get_tuple_element->operand(0); in HandleGetTupleElement()
306 const Shape& shape = get_tuple_element->shape(); in HandleGetTupleElement()
307 emitted_value_[get_tuple_element] = llvm_ir::EmitGetTupleElement( in HandleGetTupleElement()
308 shape, get_tuple_element->tuple_index(), MinimumAlignmentForShape(shape), in HandleGetTupleElement()
|