Home
last modified time | relevance | path

Searched refs:BindHloToIrValue (Results 1 – 4 of 4) sorted by relevance

/external/tensorflow/tensorflow/compiler/xla/service/gpu/
Dhlo_to_ir_bindings.cc54 BindHloToIrValue(*io_hlo, EmitGetTupleElement(io_hlo, &*arg_iter)); in EmitBasePointersForHlos()
56 BindHloToIrValue(*io_hlo, &*arg_iter); in EmitBasePointersForHlos()
86 BindHloToIrValue(*non_io_hlo, in EmitBasePointersForHlos()
113 BindHloToIrValue(*non_io_hlo, b_->CreateAlloca(pointee_type), in EmitBasePointersForHlos()
119 BindHloToIrValue(*non_io_hlo, global_for_constant); in EmitBasePointersForHlos()
123 BindHloToIrValue( in EmitBasePointersForHlos()
176 void HloToIrBindings::BindHloToIrValue(const HloInstruction& hlo, in BindHloToIrValue() function in xla::gpu::HloToIrBindings
Dhlo_to_ir_bindings.h53 void BindHloToIrValue(const HloInstruction& hlo, llvm::Value* ir_value,
Dir_emitter.cc92 bindings_.BindHloToIrValue(*bitcast, GetBasePointer(*operand)); in HandleBitcast()
104 bindings_.BindHloToIrValue(*add_dependency, GetBasePointer(*operand)); in HandleAddDependency()
112 bindings_.BindHloToIrValue( in HandleGetTupleElement()
Dir_emitter_unnested.cc1686 bindings_.BindHloToIrValue(*instr, loc, index); in BuildKernelThunk()