Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/compiler/xla/service/gpu/
Dnvptx_compiler.h49 HloDataflowAnalysis::CanShareBuffer GetCanShareBuffer() override;
Dgpu_compiler.h78 virtual HloDataflowAnalysis::CanShareBuffer GetCanShareBuffer() { in GetCanShareBuffer() function
Dgpu_compiler.cc383 pipeline.AddPass<LoopScheduleLinearizer>(GetCanShareBuffer()); in PrepareHloModuleForIrEmitting()
384 pipeline.AddPass<GpuCopyInsertion>(GetCanShareBuffer()); in PrepareHloModuleForIrEmitting()
907 GetCanShareBuffer(), pointer_size_, profile_index_map.get(), &llvm_module, in RunBackend()
Dnvptx_compiler.cc283 HloDataflowAnalysis::CanShareBuffer NVPTXCompiler::GetCanShareBuffer() { in GetCanShareBuffer() function in xla::gpu::NVPTXCompiler