Home
last modified time | relevance | path

Searched refs:shapeComputeGraphForSchema (Results 1 – 6 of 6) sorted by relevance

/aosp_15_r20/external/pytorch/torch/csrc/jit/runtime/
H A Dsymbolic_shape_registry.h57 TORCH_API std::optional<std::shared_ptr<Graph>> shapeComputeGraphForSchema(
H A Dsymbolic_shape_registry.cpp380 std::optional<std::shared_ptr<Graph>> shapeComputeGraphForSchema( in shapeComputeGraphForSchema() function
/aosp_15_r20/external/pytorch/torch/csrc/jit/passes/
H A Dsymbolic_shape_analysis.cpp592 auto maybe_graph = shapeComputeGraphForSchema(*schema_); in SymbolicShapeOpAnalyzer()
1134 auto has_shape_compute = shapeComputeGraphForSchema(*schema) != std::nullopt; in calculateSymbolicShapesOnOp()
H A Dtensorexpr_fuser.cpp1152 (node->maybeSchema() && shapeComputeGraphForSchema(node->schema()))); in canHandle()
/aosp_15_r20/external/pytorch/torch/csrc/jit/python/
H A Dinit.cpp224 return shapeComputeGraphForSchema(n->schema()); in initJITBindings()
/aosp_15_r20/external/pytorch/test/cpp/jit/
H A Dtest_misc.cpp3047 auto g = shapeComputeGraphForSchema(*schema); in TEST()