Searched refs:construct_autograd_kernel (Results 1 – 2 of 2) sorted by relevance
17 from .autograd import autograd_kernel_indirection, construct_autograd_kernel379 kernel = construct_autograd_kernel(
92 def construct_autograd_kernel( function