Searched refs:jvp_index (Results 1 – 2 of 2) sorted by relevance
1034 for input_index, jvp_index in tape_indices:1036 if len(forward_wrapper_graph.inputs) != jvp_index:1040 .format(jvp_index, len(forward_wrapper_graph.inputs)))1044 external_jvp = input_tangents[jvp_index - num_inference_inputs]
2363 for (Py_ssize_t jvp_index = 0; jvp_index < num_jvps; ++jvp_index) { local2364 PyObject* tuple = jvp_index_seq_array[jvp_index];