Home
last modified time | relevance | path

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

/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/stream_executor/gpu/
H A Dasm_compiler.h113 executor->CreateTypedKernel<Args...>(kernel_name, ptx, cubin_data)); in LoadKernelOrGetPtr()
H A Dredzone_allocator.cc336 (executor->CreateTypedKernel<DeviceMemory<uint8>, uint8, uint64_t, in CheckRedzones()
/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/gpu/
H A Dprecompiled_kernels.cc130 auto kernel = stream_exec->CreateTypedKernel<KernelArgs...>( in Get()
H A Dbuffer_comparator.cc652 (executor->CreateTypedKernel<se::DeviceMemory<ElementT>, in DeviceCompare()
/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/stream_executor/
H A Dstream_executor_pimpl.h461 port::StatusOr<std::unique_ptr<TypedKernel<Args...>>> CreateTypedKernel(
811 StreamExecutor::CreateTypedKernel(absl::string_view kernel_name, in CreateTypedKernel() function