Searched refs:TypedKernel (Results 1 – 5 of 5) sorted by relevance
508 class TypedKernel : public KernelBase {513 explicit TypedKernel(StreamExecutor *parent) : KernelBase(parent) {}517 TypedKernel(StreamExecutor *parent, internal::KernelInterface *implementation)598 SE_DISALLOW_COPY_AND_ASSIGN(TypedKernel);718 struct KernelParamsOk<TypedKernel<Params...>, Args...> {
457 port::StatusOr<std::unique_ptr<TypedKernel<Args...>>> CreateTypedKernel(539 friend class TypedKernel; variable792 inline port::StatusOr<std::unique_ptr<TypedKernel<Args...>>>796 auto kernel_base = absl::make_unique<TypedKernel<Args...>>(this); in CreateTypedKernel()
186 const TypedKernel<Params...> &kernel, Args... args);2120 const TypedKernel<Params...> &kernel, in ThenLaunch()
554 se::TypedKernel<se::DeviceMemory<ElementT>, se::DeviceMemory<ElementT>,
176 TypedKernel<DeviceMemory<uint8>, uint8, uint64, DeviceMemory<uint64>>;