Searched refs:loader_spec (Results 1 – 2 of 2) sorted by relevance
197 se::MultiKernelLoaderSpec loader_spec(num_args); in CreateKernel() local198 loader_spec.AddCudaPtxInMemory(ptx, kernel_name); in CreateKernel()201 loader_spec.AddCudaCubinInMemory( in CreateKernel()206 TF_RETURN_IF_ERROR(stream_exec->GetKernel(loader_spec, kernel_base.get())); in CreateKernel()
797 MultiKernelLoaderSpec loader_spec(kernel_base->kNumberOfParameters); in CreateTypedKernel()798 loader_spec.AddCudaPtxInMemory(ptx, kernel_name); in CreateTypedKernel()801 loader_spec.AddCudaCubinInMemory( in CreateTypedKernel()805 TF_RETURN_IF_ERROR(GetKernel(loader_spec, kernel_base.get())); in CreateTypedKernel()