Searched refs:NVPTXCompiler (Results 1 – 3 of 3) sorted by relevance
107 Status NVPTXCompiler::OptimizeHloConvolutionCanonicalization( in OptimizeHloConvolutionCanonicalization()160 Status NVPTXCompiler::OptimizeHloPostLayoutAssignment( in OptimizeHloPostLayoutAssignment()279 NVPTXCompiler::NVPTXCompiler() in NVPTXCompiler() function in xla::gpu::NVPTXCompiler283 HloDataflowAnalysis::CanShareBuffer NVPTXCompiler::GetCanShareBuffer() { in GetCanShareBuffer()287 GpuVersion NVPTXCompiler::GetGpuVersion(se::StreamExecutor* stream_exec) { in GetGpuVersion()301 NVPTXCompiler::CompileTargetBinary(const HloModuleConfig& module_config, in CompileTargetBinary()341 std::vector<uint8> NVPTXCompiler::CompileGpuAsmOrGetCachedResult( in CompileGpuAsmOrGetCachedResult()446 StatusOr<std::vector<uint8>> NVPTXCompiler::LinkModules( in LinkModules()
36 class NVPTXCompiler : public GpuCompiler {38 NVPTXCompiler();39 ~NVPTXCompiler() override {} in ~NVPTXCompiler()136 TF_DISALLOW_COPY_AND_ASSIGN(NVPTXCompiler);
21 []() { return absl::make_unique<xla::gpu::NVPTXCompiler>(); }); in InitModule()