Searched defs:expected_output_vjps (Results 1 – 1 of 1) sorted by relevance
220 std::vector<size_t> expected_output_vjps = {0, 1}; in TEST() local265 std::vector<size_t> expected_output_vjps = {0}; // only a requires grad in TEST() local