Searched defs:SoftmaxLogits1DMaxKernel (Results 1 – 1 of 1) sorted by relevance
63 struct SoftmaxLogits1DMaxKernel struct67 SoftmaxLogits1DMaxKernelPtr ukernel; argument70 static const std::vector<SoftmaxLogits1DMaxKernel> &get_available_kernels(); argument