diff --git a/graph_net/tools/single_operator_decompose.sh b/graph_net/tools/single_operator_decompose.sh new file mode 100644 index 000000000..dd860eb6a --- /dev/null +++ b/graph_net/tools/single_operator_decompose.sh @@ -0,0 +1,48 @@ +#!/bin/bash +set -x + +GRAPH_NET_ROOT=$(python3 -c "import graph_net; import os; print(os.path.dirname(os.path.dirname(graph_net.__file__)))") +DECOMPOSE_WORKSPACE=/tmp/workspace_single_operator_decompose + +mkdir -p "$DECOMPOSE_WORKSPACE" + +model_list="$GRAPH_NET_ROOT/graph_net/config/small10_torch_samples_list.txt" + +python3 -m graph_net.apply_sample_pass \ + --model-path-list $model_list \ + --sample-pass-file-path $GRAPH_NET_ROOT/graph_net/torch/sample_pass/op_names_extractor.py \ + --sample-pass-class-name OpNamesExtractor \ + --sample-pass-config=$(base64 -w 0 <