Home
last modified time | relevance | path

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

/aosp_15_r20/external/executorch/exir/
H A Dmemory_planning.py212 if not (specs := get_node_tensor_specs(nd)):
320 for spec in get_node_tensor_specs(node):
330 for spec in get_node_tensor_specs(node):
375 if not (specs := get_node_tensor_specs(node)):
521 def get_node_tensor_specs( function
/aosp_15_r20/external/executorch/exir/passes/
H A Dmemory_planning_pass.py17 get_node_tensor_specs,
66 specs = get_node_tensor_specs(node)
H A D__init__.py221 specs = memory_planning.get_node_tensor_specs(node)
/aosp_15_r20/external/executorch/util/
H A Dactivation_memory_profiler.py17 from executorch.exir.memory_planning import get_node_tensor_specs
61 tensor_specs = get_node_tensor_specs(node)
/aosp_15_r20/external/executorch/exir/tests/
H A Dtest_memory_planning.py19 get_node_tensor_specs,
547 specs = get_node_tensor_specs(node)