Searched refs:initializer_names (Results 1 – 2 of 2) sorted by relevance
| /external/tensorflow/tensorflow/contrib/meta_graph_transform/ |
| D | meta_graph_transform.py | 56 def _get_shared_init_op(initializer_names): argument
67 return_value = initializer_names.get(_saved_model_constants.MAIN_OP_KEY, None)
69 return_value = initializer_names.get(
74 def _gtt_transforms(graph_def, input_names, output_names, initializer_names, argument
97 [k for l in initializer_names.values() for k in l])
103 def _freeze_transform(graph_def, output_names, initializer_names, saver_def, argument
124 table_initializers = initializer_names.get(_ops.GraphKeys.TABLE_INITIALIZERS,
126 shared_init_op = _get_shared_init_op(initializer_names)
138 if _saved_model_constants.LEGACY_INIT_OP_KEY in initializer_names:
140 initializer_names[_saved_model_constants.LEGACY_INIT_OP_KEY])
[all …]
|
| D | meta_graph_transform_test.py | 109 initializer_names = {'init': init_nodes}
116 graph_def, input_names, output_names, initializer_names, transforms))
130 initializer_names = {
144 initializer_names, saver_def,
253 initializer_names = {
289 graph_def, input_names, output_names, initializer_names,
293 initializer_names, expected_sparsify_cmd)
311 initializer_names = {
319 sparsify_mock.return_value = (graph_def, initializer_names)
320 freeze_mock.return_value = (graph_def, initializer_names)
[all …]
|