| /external/tensorflow/tensorflow/compiler/xla/service/llvm_ir/ |
| D | fused_ir_emitter.cc | 105 HloInstruction* get_tuple_element) { in HandleGetTupleElement() argument
107 const HloInstruction* tuple_operand = get_tuple_element->operand(0); in HandleGetTupleElement()
124 return llvm_ir::EmitGetTupleElement(get_tuple_element->shape(), in HandleGetTupleElement()
125 get_tuple_element->tuple_index(), in HandleGetTupleElement()
129 if (!get_tuple_element->shape().IsTuple()) { in HandleGetTupleElement()
130 indexed_generators_[get_tuple_element] = in HandleGetTupleElement()
135 return IrArray(tuple_element_ptr, get_tuple_element->shape()) in HandleGetTupleElement()
139 non_indexed_generators_[get_tuple_element] = emit_tuple_element_ptr; in HandleGetTupleElement()
|
| D | fused_ir_emitter.h | 75 Status HandleGetTupleElement(HloInstruction* get_tuple_element) override;
|
| /external/tensorflow/tensorflow/compiler/xla/service/gpu/ |
| D | ir_emitter.cc | 109 Status IrEmitter::HandleGetTupleElement(HloInstruction* get_tuple_element) { in HandleGetTupleElement() argument
110 auto operand = get_tuple_element->operand(0); in HandleGetTupleElement()
113 *get_tuple_element, in HandleGetTupleElement()
115 get_tuple_element->shape(), get_tuple_element->tuple_index(), in HandleGetTupleElement()
|
| D | ir_emitter.h | 80 Status HandleGetTupleElement(HloInstruction* get_tuple_element) override;
|
| D | ir_emitter_unnested.h | 168 Status HandleGetTupleElement(HloInstruction* get_tuple_element) override;
|
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | dfs_hlo_visitor_with_default.h | 133 Status HandleGetTupleElement(HloInstructionPtr get_tuple_element) override { in HandleGetTupleElement() argument
134 return DefaultAction(get_tuple_element); in HandleGetTupleElement()
|
| D | tuple_points_to_analysis.cc | 220 HloInstruction* get_tuple_element) { in HandleGetTupleElement() argument
223 int64 element_index = get_tuple_element->tuple_index(); in HandleGetTupleElement()
225 PointsToSet& points_to_set = CreateEmptyPointsToSet(get_tuple_element); in HandleGetTupleElement()
227 *PerInst(get_tuple_element->operand(0))->points_to_set; in HandleGetTupleElement()
|
| D | logical_buffer_analysis.h | 60 Status HandleGetTupleElement(HloInstruction* get_tuple_element) override;
|
| D | hlo_evaluator.cc | 1177 Status HloEvaluator::HandleGetTupleElement(HloInstruction* get_tuple_element) { in HandleGetTupleElement() argument
1178 const auto result_shape = get_tuple_element->shape(); in HandleGetTupleElement()
1179 const int64 index = get_tuple_element->tuple_index(); in HandleGetTupleElement()
1181 auto operand = get_tuple_element->operand(0); in HandleGetTupleElement()
1192 evaluated_[get_tuple_element] = in HandleGetTupleElement()
1194 return evaluated_[get_tuple_element].CopyFrom(operand_tuple_literal, in HandleGetTupleElement()
|
| D | hlo_verifier.cc | 416 Status ShapeVerifier::HandleGetTupleElement(HloInstruction* get_tuple_element) { in HandleGetTupleElement() argument
417 return CheckShape(get_tuple_element, in HandleGetTupleElement()
419 get_tuple_element->operand(0)->shape(), in HandleGetTupleElement()
420 get_tuple_element->tuple_index())); in HandleGetTupleElement()
|
| D | hlo_cost_analysis.h | 57 const HloInstruction* get_tuple_element) override;
|
| D | tuple_points_to_analysis.h | 248 Status HandleGetTupleElement(HloInstruction* get_tuple_element) override;
|
| D | hlo_evaluator.h | 209 Status HandleGetTupleElement(HloInstruction* get_tuple_element) override;
|
| D | hlo_verifier.h | 68 Status HandleGetTupleElement(HloInstruction* get_tuple_element) override;
|
| D | algebraic_simplifier.cc | 208 Status HandleGetTupleElement(HloInstruction* get_tuple_element) override;
1939 HloInstruction* get_tuple_element) { in HandleGetTupleElement() argument
1940 auto operand = get_tuple_element->mutable_operand(0); in HandleGetTupleElement()
1945 << get_tuple_element->ToString(); in HandleGetTupleElement()
1947 get_tuple_element, in HandleGetTupleElement()
1948 operand->mutable_operand(get_tuple_element->tuple_index()))) { in HandleGetTupleElement()
|
| D | tuple_points_to_analysis_test.cc | 251 auto get_tuple_element = builder.AddInstruction( in TEST_F() local
256 auto& points_to_set = points_to_analysis_->GetPointsToSet(get_tuple_element); in TEST_F()
|
| /external/tensorflow/tensorflow/compiler/xla/service/cpu/ |
| D | ir_emitter.h | 151 Status HandleGetTupleElement(HloInstruction* get_tuple_element) override;
|
| D | ir_emitter.cc | 292 Status IrEmitter::HandleGetTupleElement(HloInstruction* get_tuple_element) { in HandleGetTupleElement() argument
297 auto operand = get_tuple_element->operand(0); in HandleGetTupleElement()
298 const Shape& shape = get_tuple_element->shape(); in HandleGetTupleElement()
299 emitted_value_[get_tuple_element] = llvm_ir::EmitGetTupleElement( in HandleGetTupleElement()
300 shape, get_tuple_element->tuple_index(), MinimumAlignmentForShape(shape), in HandleGetTupleElement()
|