Home
last modified time | relevance | path

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

/aosp_15_r20/external/pytorch/torch/csrc/jit/runtime/
H A Dsymbolic_shape_registry.cpp72 cached_schema_to_graph; variable
285 cached_schema_to_graph[schema_string] = graph; in registerSchema()
374 cached_schema_to_graph.clear(); in loadFunctions()
383 if (cached_schema_to_graph.empty()) { in shapeComputeGraphForSchema()
388 auto cache_it = cached_schema_to_graph.find(&schema); in shapeComputeGraphForSchema()
389 if (cache_it != cached_schema_to_graph.end()) { in shapeComputeGraphForSchema()
416 if (cached_schema_to_graph.empty()) { in RegisterShapeComputeGraphForSchema()
422 cached_schema_to_graph[&schema] = g; in RegisterShapeComputeGraphForSchema()
427 if (cached_schema_to_graph.empty()) { in RegisteredShapeComputeSchemas()
432 schemas.reserve(cached_schema_to_graph.size()); in RegisteredShapeComputeSchemas()
[all …]