Home
last modified time | relevance | path

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

/aosp_15_r20/external/pytorch/test/quantization/pt2e/
H A Dtest_xnnpack_quantizer.py685 capture_pre_autograd_graph_node_occurrence = None
687 capture_pre_autograd_graph_node_occurrence = {
721 capture_pre_autograd_graph_node_occurrence=capture_pre_autograd_graph_node_occurrence,
/aosp_15_r20/external/pytorch/torch/testing/_internal/
H A Dcommon_quantization.py1250 capture_pre_autograd_graph_node_occurrence=None, argument
1309 if capture_pre_autograd_graph_node_occurrence is not None:
1311 … ns.call_function(k): v for k, v in capture_pre_autograd_graph_node_occurrence.items()