Searched refs:EmitGetTupleElement (Results 1 – 7 of 7) sorted by relevance
| /external/tensorflow/tensorflow/compiler/xla/service/gpu/ |
| D | hlo_to_ir_bindings.cc | 101 llvm::Value* HloToIrBindings::EmitGetTupleElement(const HloInstruction* gte, in EmitGetTupleElement() function in xla::gpu::HloToIrBindings
105 return llvm_ir::EmitGetTupleElement( in EmitGetTupleElement()
109 return llvm_ir::EmitGetTupleElement( in EmitGetTupleElement()
111 EmitGetTupleElement(gte->operand(0), base_ptr), b_); in EmitGetTupleElement()
|
| D | ir_emitter_nested.cc | 146 llvm_ir::EmitGetTupleElement(element_shape, in CodegenNestedComputation()
150 llvm_ir::EmitGetTupleElement(element_shape, in CodegenNestedComputation()
|
| D | hlo_to_ir_bindings.h | 93 llvm::Value* EmitGetTupleElement(const HloInstruction* gte,
|
| D | ir_emitter.cc | 187 llvm_ir::EmitGetTupleElement( in HandleGetTupleElement()
|
| /external/tensorflow/tensorflow/compiler/xla/service/llvm_ir/ |
| D | tuple_ops.h | 86 llvm::Value* EmitGetTupleElement(const Shape& target_shape, int64 index,
|
| D | tuple_ops.cc | 110 llvm::Value* EmitGetTupleElement(const Shape& target_shape, int64 index, in EmitGetTupleElement() function
|
| /external/tensorflow/tensorflow/compiler/xla/service/cpu/ |
| D | ir_emitter.cc | 137 llvm::Value* destination = llvm_ir::EmitGetTupleElement( in EmitThreadLocalFunctionEpilogue()
143 llvm::Value* source = llvm_ir::EmitGetTupleElement( in EmitThreadLocalFunctionEpilogue()
363 emitted_value_[get_tuple_element] = llvm_ir::EmitGetTupleElement( in HandleGetTupleElement()
535 llvm::Value* tuple_element = llvm_ir::EmitGetTupleElement( in HandleOutfeed()
|