Searched defs:grad_data (Results 1 – 10 of 10) sorted by relevance
170 XlaOp ScatterToGradData(XlaOpKernelContext* ctx, XlaOp grad_data, XlaOp indices, in ScatterToGradData()304 auto grad_data = xla::ConstantLiteral( in CalculateGradData() local656 auto grad_data = CalculateGradData(ctx, grad_output, ratio, gather_indices, in Compile() local
160 const scalar_t* grad_data, in segment_reduce_backward_kernel()324 auto* grad_data = grad_contig.const_data_ptr<scalar_t>(); in _segment_reduce_lengths_offsets_backward_cuda_kernel() local
318 scalar_t* grad_data = grad.mutable_data_ptr<scalar_t>(); in spmm_reduce_backward_input_arg_kernel_impl() local400 scalar_t* grad_data = grad.mutable_data_ptr<scalar_t>(); in spmm_reduce_backward_other_arg_kernel_impl() local
153 scalar_t* grad_data = grad.data_ptr<scalar_t>(); in adagrad_fused_step_impl() local
199 scalar_t* grad_data = grad.data_ptr<scalar_t>(); in sgd_fused_step_impl() local
287 scalar_t* grad_data = grad.data_ptr<scalar_t>(); in adam_fused_step_impl() local
228 const scalar_t* grad_data = grad_data_base + i * dim_size; in _vec_host_softmax_backward_lastdim() local
190 float grad_data[1]; in CalcNumericalGrad() local
215 auto* grad_data = grad_contig.const_data_ptr<scalar_t>(); in _segment_reduce_cpu_lengths_backward_kernel1() local
1690 auto* grad_data = grad.const_data_ptr<scalar_t>(); in _embedding_bag_per_sample_weights_backward_cpu_template() local