Home
last modified time | relevance | path

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

/external/pytorch/torch/profiler/
D_memory_profiler.py217 class SchemaMatcher: class
441 op.inputs, SchemaMatcher.inputs_are_mutable(op)
590 or bool(SchemaMatcher.match_schemas(e.typed[1]))
/external/pytorch/test/profiler/
Dtest_memory_profiler.py308 schemas = _memory_profiler.SchemaMatcher.match_schemas(e)
315 out.append((name, _memory_profiler.SchemaMatcher.inputs_are_mutable(e)))