Home
last modified time | relevance | path

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

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