Searched defs:grad_ptr (Results 1 – 6 of 6) sorted by relevance
20 scalar_t* grad_ptr, in adagrad_math()89 scalar_t* grad_ptr, in adagrad_math()167 scalar_t* grad_ptr = grad_data + begin; in adagrad_fused_step_impl() local
20 scalar_t* grad_ptr, in sgd_math()112 scalar_t* grad_ptr, in sgd_math()212 scalar_t* grad_ptr = grad_data + begin; in sgd_fused_step_impl() local
22 scalar_t* grad_ptr, in adam_math()165 scalar_t* grad_ptr, in adam_math()308 scalar_t* grad_ptr = grad_data + begin; in adam_fused_step_impl() local
323 scalar_t* grad_ptr = grad_data + m * K; in spmm_reduce_backward_input_arg_kernel_impl() local405 scalar_t* grad_ptr = grad_data + m * K; in spmm_reduce_backward_other_arg_kernel_impl() local
985 auto grad_ptr = grad.const_data_ptr<scalar_t>(); in host_softmax_backward() local1009 auto grad_ptr = grad.const_data_ptr<accscalar_t>(); in host_softmax_backward() local
83 auto grad_ptr = get_nested_tensor_impl(grad); in nested_softmax_backward() local