Searched refs:gpu_kernel_multiple_outputs (Results 1 – 5 of 5) sorted by relevance
39 gpu_kernel_multiple_outputs( in fake_quantize_tensor_cachemask_kernel_cuda()73 gpu_kernel_multiple_outputs( in fake_quantize_tensor_cachemask_tensor_qparams_kernel_cuda()102 gpu_kernel_multiple_outputs( in _fake_quantize_grad_learnable_tensor_kernel_cuda()191 gpu_kernel_multiple_outputs(iter, in _fake_quantize_grad_learnable_channel_kernel_cuda()
305 void gpu_kernel_multiple_outputs(TensorIteratorBase& iter, const func_t& f) { in gpu_kernel_multiple_outputs() function318 gpu_kernel_multiple_outputs(sub_iter, f); in gpu_kernel_multiple_outputs()
35 gpu_kernel_multiple_outputs(iter, in prelu_backward_kernel()
270 …gpu_kernel_multiple_outputs(iter, [=] GPU_LAMBDA (scalar_t a) -> thrust::tuple<scalar_t, int32_t> { in frexp_kernel_cuda()
368 gpu_kernel_multiple_outputs( in batch_norm_update_stats()404 gpu_kernel_multiple_outputs( in batch_norm_update_stats_and_invert()