Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/stream_executor/gpu/
Dasm_compiler.h58 struct CubinOrPTXImage { struct
66 std::vector<CubinOrPTXImage> images, const std::string preferred_cuda_dir); argument
81 gpu::GpuContext* context, std::vector<CubinOrPTXImage> images);
Dasm_compiler.cc286 std::vector<CubinOrPTXImage> images, const std::string preferred_cuda_dir) { in BundleGpuAsm()
293 for (const CubinOrPTXImage& img : images) { in BundleGpuAsm()
/external/tensorflow/tensorflow/stream_executor/cuda/
Dcuda_asm_compiler.cc36 gpu::GpuContext* context, std::vector<CubinOrPTXImage> images) { in LinkGpuAsm()
/external/tensorflow/tensorflow/compiler/mlir/tools/kernel_gen/transforms/
Dgpu_kernel_to_blob_pass.cc163 std::vector<tensorflow::se::CubinOrPTXImage> images; in GetGpuBinaryBlob()
/external/tensorflow/tensorflow/compiler/xla/service/gpu/
Dnvptx_compiler.cc448 std::vector<stream_executor::CubinOrPTXImage> images; in LinkModules()