Searched defs:SoftmaxLogits1DKernel (Results 1 – 1 of 1) sorted by relevance
112 struct SoftmaxLogits1DKernel struct116 SoftmaxLogits1DKernelPtr ukernel; argument119 static const std::vector<SoftmaxLogits1DKernel> &get_available_kernels(); argument