Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/stream_executor/
Dstream_executor_pimpl.h457 port::StatusOr<std::unique_ptr<TypedKernel<Args...>>> CreateTypedKernel(
793 StreamExecutor::CreateTypedKernel(absl::string_view kernel_name, in CreateTypedKernel() function
/external/tensorflow/tensorflow/compiler/xla/service/gpu/
Dbuffer_comparator.cc603 (executor->CreateTypedKernel<se::DeviceMemory<ElementT>, in DeviceCompare()
/external/tensorflow/tensorflow/stream_executor/gpu/
Dredzone_allocator.cc327 (executor->CreateTypedKernel<DeviceMemory<uint8>, uint8, uint64, in CheckRedzones()