Searched refs:EmitGetTupleElement (Results 1 – 7 of 7) sorted by relevance
| /external/tensorflow/tensorflow/compiler/xla/service/gpu/ |
| D | hlo_to_ir_bindings.cc | 54 BindHloToIrValue(*io_hlo, EmitGetTupleElement(io_hlo, &*arg_iter)); in EmitBasePointersForHlos()
87 EmitGetTupleElement(non_io_hlo, base_ptr)); in EmitBasePointersForHlos()
132 llvm::Value* HloToIrBindings::EmitGetTupleElement(const HloInstruction* gte, in EmitGetTupleElement() function in xla::gpu::HloToIrBindings
136 return llvm_ir::EmitGetTupleElement( in EmitGetTupleElement()
140 return llvm_ir::EmitGetTupleElement( in EmitGetTupleElement()
142 EmitGetTupleElement(gte->operand(0), base_ptr), b_); in EmitGetTupleElement()
|
| D | hlo_to_ir_bindings.h | 95 llvm::Value* EmitGetTupleElement(const HloInstruction* gte,
|
| D | ir_emitter.cc | 114 llvm_ir::EmitGetTupleElement( in HandleGetTupleElement()
|
| /external/tensorflow/tensorflow/compiler/xla/service/llvm_ir/ |
| D | tuple_ops.h | 80 llvm::Value* EmitGetTupleElement(const Shape& target_shape, int64 index,
|
| D | tuple_ops.cc | 94 llvm::Value* EmitGetTupleElement(const Shape& target_shape, int64 index, in EmitGetTupleElement() function
|
| D | fused_ir_emitter.cc | 124 return llvm_ir::EmitGetTupleElement(get_tuple_element->shape(), in HandleGetTupleElement()
|
| /external/tensorflow/tensorflow/compiler/xla/service/cpu/ |
| D | ir_emitter.cc | 299 emitted_value_[get_tuple_element] = llvm_ir::EmitGetTupleElement( in HandleGetTupleElement()
494 llvm::Value* tuple_element = llvm_ir::EmitGetTupleElement( in HandleOutfeed()
|