Searched defs:output_grads (Results 1 – 5 of 5) sorted by relevance
54 auto UndefinedGradBackward::apply(variable_list&& output_grads) in apply()
264 output_grads, argument
2469 def post_hook_add_one(output_grads, input_grads): argument2473 def post_hook_add_two(output_grads, input_grads): argument
521 OutputList output_grads; in TEST_F() local
3323 const std::vector<int64_t>& unneeded_gradients) {