node_output_sizes (reference) in projects: aosp_15_r20

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

/aosp_15_r20/external/pytorch/test/jit/
H A D	test_symbolic_shape_analysis.py	`564 self, shape_compute_graph, nodes, node_output_sizes, shape_inputs argument 578 for node, output_shape in zip(nodes, node_output_sizes):`