Searched refs:CudaPtxInMemory (Results 1 – 2 of 2) sorted by relevance
/external/tensorflow/tensorflow/stream_executor/ |
D | kernel_spec.cc | 47 const std::tuple<int, int> CudaPtxInMemory::kMinimumCapability{1, 0}; 49 CudaPtxInMemory::CudaPtxInMemory(absl::string_view ptx, in CudaPtxInMemory() function in stream_executor::CudaPtxInMemory 62 CudaPtxInMemory::CudaPtxInMemory( in CudaPtxInMemory() function in stream_executor::CudaPtxInMemory 63 const std::initializer_list<CudaPtxInMemory::PtxSpec> &spec_list, in CudaPtxInMemory() 80 std::string CudaPtxInMemory::DecompressPtx(const char *ptx) { in DecompressPtx() 92 const char *CudaPtxInMemory::default_text() const { in default_text() 113 const char *CudaPtxInMemory::original_default_text() const { in original_default_text() 121 const char *CudaPtxInMemory::text(int compute_capability_major, in text() 146 const char *CudaPtxInMemory::original_text(int compute_capability_major, in original_text() 217 new CudaPtxInMemory{ptx, kernelname, false /* ptx_compressed */}); in AddCudaPtxInMemory() [all …]
|
D | kernel_spec.h | 142 class CudaPtxInMemory : public KernelLoaderSpec { 156 CudaPtxInMemory(absl::string_view ptx, absl::string_view kernelname, 162 CudaPtxInMemory(const std::initializer_list<PtxSpec> &spec_list, 164 ~CudaPtxInMemory() override {} in ~CudaPtxInMemory() 215 SE_DISALLOW_COPY_AND_ASSIGN(CudaPtxInMemory); 306 const CudaPtxInMemory &cuda_ptx_in_memory() const { in cuda_ptx_in_memory() 348 std::initializer_list<CudaPtxInMemory::PtxSpec> spec_list, 351 std::initializer_list<CudaPtxInMemory::PtxSpec> spec_list, 361 std::unique_ptr<CudaPtxInMemory>
|