Searched defs:grad_values (Results 1 – 11 of 11) sorted by relevance
177 std::vector<Value*> gradient(ArrayRef<Value*> grad_values) { in gradient()197 const ArrayRef<Value*>& grad_values) { in buildSymbolicGradient()310 ArrayRef<Value*> grad_values) { in linearGradientForNode()
101 auto grad_values = grad._values(); in step() local
652 ShapeHandle grad_values = c->input(1); in __anone91b66371902() local
402 const auto grad_values = grad_values_t->vec<T>(); in Compute() local
490 GpuLaunchConfig cfg, const Tindex* reverse_index_map, const T* grad_values, in GatherOriginalGradValuesKernel()
414 auto grad_values = grad._values().contiguous(); in cpu_sparse_coo_softmax_backward() local
1817 Tensor grad_values = grad._values(); in _sparse_sum_backward_cpu() local
468 auto grad_values = grad._values().contiguous(); in cuda_sparse_coo_softmax_backward() local
596 Tensor grad_values = grad._values(); in _sparse_sum_backward_cuda() local
308 auto grad_values = grad_self.values(); in spmm_reduce_backward_input_arg_kernel_impl() local
1474 grad_values, argument