Searched defs:input_grads (Results 1 – 5 of 5) sorted by relevance
56 tensor_list input_grads; in apply() local
45 variable_list input_grads; in _process_forward_mode_AD() local
5363 auto input_grads = std::vector<Tensor>(k < input.sym_size(-1) ? k + 1 : k); in householder_product_backward() local
2469 def post_hook_add_one(output_grads, input_grads): argument2473 def post_hook_add_two(output_grads, input_grads): argument
110 auto input_grads = grad({res}, {x}, {grad_output}, {}, true); in TEST() local