|
185 | 185 | "source": [ |
186 | 186 | "from tensorflow.python.framework.convert_to_constants import convert_variables_to_constants_v2\n", |
187 | 187 | "\n", |
188 | | - "def save_frezon_pb(model, mod_path):\n", |
| 188 | + "def save_frozen_pb(model, mod_path):\n", |
189 | 189 | " # Convert Keras model to ConcreteFunction\n", |
190 | 190 | " full_model = tf.function(lambda x: model(x))\n", |
191 | 191 | " concrete_function = full_model.get_concrete_function(\n", |
|
199 | 199 | " logdir=\".\",\n", |
200 | 200 | " name=mod_path,\n", |
201 | 201 | " as_text=False)\n", |
202 | | - "fp32_frezon_pb_file = \"fp32_frozen.pb\"\n", |
203 | | - "save_frezon_pb(model, fp32_frezon_pb_file)" |
| 202 | + "fp32_frozen_pb_file = \"fp32_frozen.pb\"\n", |
| 203 | + "save_frozen_pb(model, fp32_frozen_pb_file)" |
204 | 204 | ] |
205 | 205 | }, |
206 | 206 | { |
|
209 | 209 | "metadata": {}, |
210 | 210 | "outputs": [], |
211 | 211 | "source": [ |
212 | | - "!ls -la fp32_frezon.pb" |
| 212 | + "!ls -la fp32_frozen.pb" |
213 | 213 | ] |
214 | 214 | }, |
215 | 215 | { |
|
328 | 328 | " return q_model\n", |
329 | 329 | "\n", |
330 | 330 | "\n", |
331 | | - "def save_int8_frezon_pb(q_model, path):\n", |
| 331 | + "def save_int8_frozen_pb(q_model, path):\n", |
332 | 332 | " from tensorflow.python.platform import gfile\n", |
333 | 333 | " f = gfile.GFile(path, 'wb')\n", |
334 | 334 | " f.write(q_model.as_graph_def().SerializeToString())\n", |
|
421 | 421 | }, |
422 | 422 | "outputs": [], |
423 | 423 | "source": [ |
424 | | - "!python profiling_inc.py --input-graph=./fp32_frezon.pb --omp-num-threads=4 --num-inter-threads=1 --num-intra-threads=4 --index=32" |
| 424 | + "!python profiling_inc.py --input-graph=./fp32_frozen.pb --omp-num-threads=4 --num-inter-threads=1 --num-intra-threads=4 --index=32" |
425 | 425 | ] |
426 | 426 | }, |
427 | 427 | { |
|
0 commit comments