Home
last modified time | relevance | path

Searched refs:GradientPair (Results 1 – 2 of 2) sorted by relevance

/external/pytorch/torch/csrc/jit/runtime/
Dsymbolic_script.h10 struct GradientPair { struct
15 TORCH_API std::optional<GradientPair> gradientInfoForSchema( argument
Dsymbolic_script.cpp1488 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()