Searched refs:shape_compute_graph (Results 1 – 1 of 1) sorted by relevance
| /aosp_15_r20/external/pytorch/test/jit/ |
| H A D | test_symbolic_shape_analysis.py | 528 shape_compute_graph = (
531 g = shape_compute_graph.partial_eval_shape_graph()
543 shape_compute_graph = (
553 g = shape_compute_graph.partial_eval_shape_graph()
564 self, shape_compute_graph, nodes, node_output_sizes, shape_inputs argument
566 g = shape_compute_graph.partial_eval_shape_graph()
568 output_sym_map = shape_compute_graph.graph_output_to_symbolic_shape_dim()
606 shape_compute_graph = (
618 shape_compute_graph, nodes, output_shapes, ([1, 3, 224, 224],)
669 shape_compute_graph = (
[all …]
|