Function is_param_node

backends/xnnpack/utils/utils.py:99–105 · view source on GitHub ↗

(exp_prog: ExportedProgram, node: torch.fx.Node)

Source from the content-addressed store, hash-verified

97
98
99	def is_param_node(exp_prog: ExportedProgram, node: torch.fx.Node) -> bool:
100	return (
101	is_get_attr_node(node)
102	or is_param(exp_prog, node)
103	or is_buffer(exp_prog, node)
104	or is_lifted_tensor_constant(exp_prog, node)
105	)
106
107
108	def get_param_tensor(

generate_node_to_external_mapFunction · 0.90

define_nodeMethod · 0.90

is_graph_inputMethod · 0.90

gen_ids_and_flagsMethod · 0.90

get_serialized_buffer_indexMethod · 0.90

from_inputsMethod · 0.90

callMethod · 0.90

can_be_converted_to_nhwcMethod · 0.90

input_to_nhwcMethod · 0.90

input_to_nchwMethod · 0.90

can_fuseMethod · 0.90

can_decompose_batch_normMethod · 0.90

is_get_attr_nodeFunction · 0.70

no test coverage detected