Home
last modified time | relevance | path

Searched refs:wrap_kernel_call (Results 1 – 4 of 4) sorted by relevance

/external/tensorflow/tensorflow/core/kernels/
Dsparse_fill_empty_rows_op_gpu.cu.cc63 Status wrap_kernel_call(void (*func)(KernelArgs...), const GPUDevice& device, in wrap_kernel_call() function
71 using kernel_forward::wrap_kernel_call;
271 TF_RETURN_IF_ERROR(wrap_kernel_call( in operator ()()
300 TF_RETURN_IF_ERROR(wrap_kernel_call(ComputeEmptyRowIndicatorKernel<Tindex>, in operator ()()
400 wrap_kernel_call(ScatterInputElementsKernel<T, Tindex>, in operator ()()
410 wrap_kernel_call(ScatterNewElementsKernel<T, Tindex>, in operator ()()
464 TF_RETURN_IF_ERROR(wrap_kernel_call(CopyRowIndicesKernel<Tindex>, in ArgSortByRows()
531 TF_RETURN_IF_ERROR(wrap_kernel_call( in operator ()()
Dtraining_ops_gpu.cu.cc373 void wrap_kernel_call(void (*func)(KernelArgs...), const GPUDevice& d, T var, in wrap_kernel_call() function
383 using kernel_forward::wrap_kernel_call;
392 wrap_kernel_call(ApplyAdagradKernel<T>, d, var, accum, lr, grad, in operator ()()
414 wrap_kernel_call(ApplyAdagradV2Kernel<T>, d, var, accum, lr, epsilon, grad, in operator ()()
463 wrap_kernel_call(ApplyProximalAdagradKernel<T>, d, var, accum, lr, l1, l2, in operator ()()
525 wrap_kernel_call(ApplyAdadeltaKernel<T>, d, var, accum, accum_update, lr, in operator ()()
963 wrap_kernel_call(ApplyRMSPropKernel<T>, d, var, ms, mom, lr, rho, momentum, in operator ()()
993 wrap_kernel_call(ApplyCenteredRMSPropKernel<T>, d, var, mg, ms, mom, lr, in operator ()()
/external/pytorch/torch/_inductor/codegen/
Dcpp_wrapper_cpu.py128 self.writeline(self.wrap_kernel_call(kernel_name, new_args))
130 self.writeline(self.wrap_kernel_call(kernel_name, call_args))
1331 self.writeline(self.wrap_kernel_call(kernel, args))
1409 self.writeline(self.wrap_kernel_call(kernel, args))
Dwrapper.py838 self.writeline(self.wrap_kernel_call(kernel, args))
1499 def wrap_kernel_call(self, name, call_args): member in WrapperCodeGen
1746 self.writeline(self.wrap_kernel_call(kernel_name, call_args))