Searched defs:with_grad (Results 1 – 3 of 3) sorted by relevance
132 ArgumentSpec ArgumentSpecCreator::create(bool with_grad, const Stack& input) in create()
453 bool with_grad = false) { in _propagate_shapes()465 bool with_grad = false, in _propagate_and_assign_input_shapes()
552 bool with_grad) { in initJITBindings()