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 {
42 static StatusOr<std::unique_ptr<IrEmitterNested>> Create(
47 IrEmitterNested(const IrEmitterNested&) = delete;
48 IrEmitterNested& operator=(const IrEmitterNested&) = delete;
67 IrEmitterNested(const HloModuleConfig& hlo_module_config,
Dir_emitter_nested.cc38 IrEmitterNested::IrEmitterNested(const HloModuleConfig& hlo_module_config, in IrEmitterNested() function in xla::gpu::IrEmitterNested
44 StatusOr<std::unique_ptr<IrEmitterNested>> IrEmitterNested::Create( in Create()
48 std::unique_ptr<IrEmitterNested> emitter(new IrEmitterNested( in Create()
56 Status IrEmitterNested::CodegenNestedComputation() { in CodegenNestedComputation()
162 Status IrEmitterNested::HandleParameter(HloInstruction* parameter) { in HandleParameter()
166 Status IrEmitterNested::EmitTargetElementLoop( in EmitTargetElementLoop()
Dir_emitter.cc233 IrEmitterNested::Create(hlo_module_config_, nested_computation, in EmitCallToNestedComputation()