Home
last modified time | relevance | path

Searched refs:cubin_data (Results 1 – 6 of 6) sorted by relevance

/external/tensorflow/tensorflow/compiler/xla/stream_executor/gpu/
Dasm_compiler.h97 absl::string_view ptx, absl::Span<const uint8> cubin_data) { in LoadKernelOrGetPtr() argument
113 executor->CreateTypedKernel<Args...>(kernel_name, ptx, cubin_data)); in LoadKernelOrGetPtr()
/external/tensorflow/tensorflow/compiler/xla/service/gpu/
Dnvptx_compiler.cc427 cache_value->cubin_data = std::move(maybe_cubin).ValueOrDie(); in CompileGpuAsmOrGetCachedResult()
429 << " CUBIN size: " << cache_value->cubin_data.size(); in CompileGpuAsmOrGetCachedResult()
484 return cache_value->cubin_data; in CompileGpuAsmOrGetCachedResult()
Dstream_executor_util.h83 absl::Span<const uint8_t> cubin_data, se::StreamExecutor* stream_exec);
Dstream_executor_util.cc333 absl::Span<const uint8_t> cubin_data, se::StreamExecutor* stream_exec) { in CreateKernel() argument
337 if (!cubin_data.empty()) { in CreateKernel()
339 reinterpret_cast<const char*>(cubin_data.data()), kernel_name); in CreateKernel()
Dnvptx_compiler.h117 std::vector<uint8_t> cubin_data; member
/external/tensorflow/tensorflow/compiler/xla/stream_executor/
Dstream_executor_pimpl.h463 absl::Span<const uint8> cubin_data);
813 absl::Span<const uint8> cubin_data) { in CreateTypedKernel() argument
818 if (!cubin_data.empty()) { in CreateTypedKernel()
820 reinterpret_cast<const char*>(cubin_data.data()), kernel_name); in CreateTypedKernel()