Searched refs:prev_layer (Results 1 – 2 of 2) sorted by relevance
| /frameworks/ml/nn/tools/systrace_parser/parser/ |
| D | tree.py | 147 prev_layer = (node.parent and node.parent.layer)
150 prev_phase, prev_layer,
164 assert ((prev_layer is None) or
166 (layer == prev_layer) or
167 (layer in layer_order.get(prev_layer, [])) or
|
| D | tracker.py | 144 def recurse(node, prev_layer, prev_phase, indent, in_pe_layers): argument
169 recurse(c, layer or prev_layer, phase or prev_phase,
|