Home
last modified time | relevance | path

Searched refs:AddCudaPtxInMemory (Results 1 – 5 of 5) sorted by relevance

/external/tensorflow/tensorflow/stream_executor/
Dmodule_spec.h50 void AddCudaPtxInMemory(const char* ptx) { in AddCudaPtxInMemory() function
Dkernel_spec.h343 MultiKernelLoaderSpec *AddCudaPtxInMemory(absl::string_view ptx,
347 MultiKernelLoaderSpec *AddCudaPtxInMemory(
Dkernel_spec.cc213 MultiKernelLoaderSpec *MultiKernelLoaderSpec::AddCudaPtxInMemory( in AddCudaPtxInMemory() function in stream_executor::MultiKernelLoaderSpec
229 MultiKernelLoaderSpec *MultiKernelLoaderSpec::AddCudaPtxInMemory( in AddCudaPtxInMemory() function in stream_executor::MultiKernelLoaderSpec
/external/tensorflow/tensorflow/compiler/xla/service/gpu/
Dkernel_thunk.cc44 loader_spec_->AddCudaPtxInMemory(executable.ptx(), kernel_name_); in Initialize()
Dgpu_executable.cc199 module_spec.AddCudaPtxInMemory(ptx().c_str()); in ResolveConstantGlobals()