Searched refs:get_registered_kernels_for_op (Results 1 – 4 of 4) sorted by relevance
35 kernel_list = kernels.get_registered_kernels_for_op("KernelLabel")
36 def get_registered_kernels_for_op(name): function
171 if kernels.get_registered_kernels_for_op('NcclAllReduce'):
1419 if kernels.get_registered_kernels_for_op("NcclAllReduce"):