Searched refs:need_gradients_for_jvps (Results 1 – 1 of 1) sorted by relevance
881 need_gradients_for_jvps): argument896 self._need_gradients_for_jvps = need_gradients_for_jvps1326 need_gradients_for_jvps): argument1329 delayed_rewrite_functions, need_gradients_for_jvps)2112 need_gradients_for_jvps = tape.should_record_backprop(2116 cache_key = (need_gradients_for_jvps, input_tangents.indices)2133 need_gradients_for_jvps=need_gradients_for_jvps)2161 need_gradients_for_jvps=need_gradients_for_jvps)