Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/compiler/xla/service/gpu/
Dnvptx_compiler.cc380 cache_value->cubin_data = std::move(maybe_cubin).ValueOrDie(); in CompileGpuAsmOrGetCachedResult()
382 << " CUBIN size: " << cache_value->cubin_data.size(); in CompileGpuAsmOrGetCachedResult()
443 return cache_value->cubin_data; in CompileGpuAsmOrGetCachedResult()
Dstream_executor_util.cc196 absl::Span<const uint8> cubin_data, se::StreamExecutor* stream_exec) { in CreateKernel() argument
200 if (!cubin_data.empty()) { in CreateKernel()
202 reinterpret_cast<const char*>(cubin_data.data()), kernel_name); in CreateKernel()
Dstream_executor_util.h71 absl::Span<const uint8> cubin_data, se::StreamExecutor* stream_exec);
Dnvptx_compiler.h124 std::vector<uint8> cubin_data; member
/external/tensorflow/tensorflow/stream_executor/
Dstream_executor_pimpl.h459 absl::Span<const uint8> cubin_data);
795 absl::Span<const uint8> cubin_data) { in CreateTypedKernel() argument
800 if (!cubin_data.empty()) { in CreateTypedKernel()
802 reinterpret_cast<const char *>(cubin_data.data()), kernel_name); in CreateTypedKernel()