Searched refs:kernels_cache (Results 1 – 2 of 2) sorted by relevance
409 JitRtKernelsCache* kernels_cache, int32_t grid_size_x,420 JitRtKernelsCache* kernels_cache, in operator ()() argument433 se::KernelBase* kernel = kernels_cache->Get(executor, ptx.data(), name); in operator ()()442 kernels_cache->Set(executor, ptx.data(), name, std::move(*created)); in operator ()()
159 JitRtKernelsCache& kernels_cache() { return kernels_cache_; } in kernels_cache() function in xla::gpu::GpuExecutable::JitRtExecutable651 &jitrt_executable->kernels_cache(), in ExecuteJitRt()