Home
last modified time | relevance | path

Searched refs:IrEmitterNested (Results 1 – 3 of 3) sorted by relevance

/external/tensorflow/tensorflow/compiler/xla/service/gpu/
Dir_emitter_nested.h40 class IrEmitterNested : public IrEmitter {
45 IrEmitterNested(const HloModuleConfig& hlo_module_config,
48 IrEmitterNested(const IrEmitterNested&) = delete;
49 IrEmitterNested& operator=(const IrEmitterNested&) = delete;
Dir_emitter_nested.cc38 IrEmitterNested::IrEmitterNested(const HloModuleConfig& hlo_module_config, in IrEmitterNested() function in xla::gpu::IrEmitterNested
47 llvm::Function* IrEmitterNested::EmitBasePointersForNestedComputation( in EmitBasePointersForNestedComputation()
112 Status IrEmitterNested::HandleParameter(HloInstruction* parameter) { in HandleParameter()
116 Status IrEmitterNested::EmitTargetElementLoop( in EmitTargetElementLoop()
Dir_emitter.cc158 IrEmitterNested ir_emitter_nested(hlo_module_config_, nested_computation, in EmitCallToNestedComputation()