Searched refs:MAIN_OP_KEY (Results 1 – 9 of 9) sorted by relevance
| /external/tensorflow/tensorflow/contrib/meta_graph_transform/ |
| D | meta_graph_transform.py | 67 return_value = initializer_names.get(_saved_model_constants.MAIN_OP_KEY, None)
141 if _saved_model_constants.MAIN_OP_KEY in initializer_names:
142 pruned_initializer_names[_saved_model_constants.MAIN_OP_KEY] = (
143 initializer_names[_saved_model_constants.MAIN_OP_KEY])
448 _saved_model_constants.MAIN_OP_KEY,
|
| D | meta_graph_transform_test.py | 93 saved_model_constants.MAIN_OP_KEY: [main_op],
133 saved_model_constants.MAIN_OP_KEY: [main_op],
149 saved_model_constants.MAIN_OP_KEY: [main_op],
|
| /external/tensorflow/tensorflow/tools/api/golden/v1/ |
| D | tensorflow.saved_model.constants.pbtxt | 16 name: "MAIN_OP_KEY"
|
| D | tensorflow.saved_model.pbtxt | 44 name: "MAIN_OP_KEY"
|
| /external/tensorflow/tensorflow/python/saved_model/ |
| D | constants.py | 52 MAIN_OP_KEY = "saved_model_main_op" variable
|
| D | loader_impl.py | 135 meta_graph_def_to_load, init_op_key=constants.MAIN_OP_KEY):
180 _get_op_from_collection(meta_graph_def, constants.MAIN_OP_KEY) or
|
| D | builder_impl.py | 476 for init_op_key in (constants.MAIN_OP_KEY, constants.LEGACY_INIT_OP_KEY):
482 ops.add_to_collection(constants.MAIN_OP_KEY, main_op)
|
| D | save.py | 514 meta_graph_def.collection_def[constants.MAIN_OP_KEY].node_list.value.append(
|
| D | saved_model_test.py | 1397 self._testInitOpsWithNonEmptyCollection(export_dir, constants.MAIN_OP_KEY)
|