Home
last modified time | relevance | path

Searched defs:grad_i (Results 1 – 4 of 4) sorted by relevance

/aosp_15_r20/external/tensorflow/tensorflow/core/kernels/
H A Dtraining_ops_gpu.cu.cc55 T grad_i = grad[grad_index]; in SparseApplyAdagradKernel() local
94 T grad_i = grad[grad_index]; in SparseApplyProximalAdagradKernel() local
138 const T grad_i = grad[grad_index]; in SparseApplyFtrlKernel() local
241 T grad_i = grad[grad_index]; in SparseApplyKerasMomentumKernel() local
822 T grad_i = grad[grad_index]; in SparseApplyAdadeltaKernel() local
/aosp_15_r20/external/pytorch/aten/src/ATen/native/cuda/
H A DUpSampleNearest1d.cu76 scalar_t* grad_i, in upsample_nearest1d_backward_out_frame()
H A DUpSampleNearest3d.cu98 scalar_t* grad_i, in upsample_nearest3d_backward_out_frame()
H A DUpSampleNearest2d.cu122 scalar_t* grad_i, in upsample_nearest2d_backward_out_frame()