Home
last modified time | relevance | path

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

/aosp_15_r20/external/executorch/backends/example/example_operators/
H A Dconv2d.py12 _annotate_nodes,
31 _annotate_nodes(
34 _annotate_nodes(
37 _annotate_nodes([(conv_node,)], quant_config.output_quant_spec)
H A Dadd.py12 _annotate_nodes,
29 _annotate_nodes(
32 _annotate_nodes(
35 _annotate_nodes([(add_node,)], quant_config.output_quant_spec)
H A Dlinear.py12 _annotate_nodes,
47 _annotate_nodes(
50 _annotate_nodes(
53 _annotate_nodes([(linear_node,)], quant_config.output_quant_spec)
H A Dconv_relu.py12 _annotate_nodes,
34 _annotate_nodes(
37 _annotate_nodes(
40 _annotate_nodes([(relu_node,)], quant_config.output_quant_spec)
H A Dflatten.py12 _annotate_nodes,
28 _annotate_nodes(
31 _annotate_nodes([(flatten_node,)], quant_config.output_quant_spec)
H A Ddropout.py12 _annotate_nodes,
31 _annotate_nodes(
34 _annotate_nodes([(dropout_node,)], quant_config.output_quant_spec)
H A Dadaptive_avg_pool2d.py12 _annotate_nodes,
33 _annotate_nodes(
38 _annotate_nodes([(adaptive_avg_pool2d_node,)], quant_config.output_quant_spec)
H A Dutils.py22 def _annotate_nodes(node_tuples, quant_spec, input_node=False): function