Home
last modified time | relevance | path

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

/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/stream_executor/
H A Dmodule_spec.h35 return {cuda_cubin_in_memory_.data(), cuda_cubin_in_memory_.size()}; in cuda_cubin_in_memory()
48 cuda_cubin_in_memory_ = cubin_bytes; in AddCudaCubinInMemory()
58 port::ArraySlice<const uint8> cuda_cubin_in_memory_; // non-absl ok
H A Dkernel_spec.h285 return cuda_cubin_in_memory_ != nullptr; in has_cuda_cubin_in_memory()
304 return *cuda_cubin_in_memory_; in cuda_cubin_in_memory()
360 cuda_cubin_in_memory_; // Binary CUDA program in memory. variable
H A Dkernel_spec.cc202 CHECK(cuda_cubin_in_memory_ == nullptr); in AddCudaCubinInMemory()
203 cuda_cubin_in_memory_.reset(new CudaCubinInMemory{bytes, kernelname}); in AddCudaCubinInMemory()