Home
last modified time | relevance | path

Searched defs:schema_str (Results 1 – 6 of 6) sorted by relevance

/aosp_15_r20/external/pytorch/torch/csrc/jit/runtime/
H A Doperator.h302 const char* schema_str, in OperatorGenerator()
311 torch::detail::SelectiveStr<true> schema_str, in OperatorGenerator()
322 torch::detail::SelectiveStr<false> schema_str, in OperatorGenerator()
H A Dsymbolic_script.cpp1627 auto schema_str = canonicalSchemaString(schema); in gradientInfoForSchema() local
H A Dregister_ops_utils.h393 const char* schema_str; member
/aosp_15_r20/external/pytorch/torch/csrc/lazy/core/
H A Dshape.cpp90 const char* schema_str, in applySymbolicShapesOnLT()
/aosp_15_r20/external/pytorch/torch/nested/_internal/
H A Dops.py177 def register_func(tables, aten_ops, schema_str): argument
/aosp_15_r20/external/pytorch/torch/csrc/jit/tensorexpr/
H A Dlowerings.cpp19 for (const auto& schema_str : schemas) { in RegisterNNCLoweringsFunction() local
1992 NNCLoweringFunction getStandardLoweringFor(const std::string& schema_str) { in getStandardLoweringFor()