Searched refs:GradientPair (Results 1 – 2 of 2) sorted by relevance
10 struct GradientPair { struct15 TORCH_API std::optional<GradientPair> gradientInfoForSchema( argument
1488 std::unordered_map<std::string, GradientPair> schema_to_graphs;1493 std::unordered_map<const FunctionSchema*, GradientPair> cached_gradient_pairs;1552 GradientPair pair; in loadModule()1617 std::optional<GradientPair> gradientInfoForSchema( in gradientInfoForSchema()