| /external/tensorflow/tensorflow/python/saved_model/ |
| D | builder_impl.py | 439 def _add_collections(self, assets_collection, main_op, train_op): argument
444 self._maybe_add_main_op(main_op)
467 def _maybe_add_main_op(self, main_op): argument
478 if main_op is None:
481 if not isinstance(main_op, ops.Operation):
482 raise TypeError("main_op needs to be an Operation: %r" % main_op)
492 ops.add_to_collection(constants.MAIN_OP_KEY, main_op)
522 main_op=None, argument
537 main_op = main_op if main_op is not None else legacy_init_op
540 self._add_collections(assets_collection, main_op, None)
[all …]
|
| D | main_op_impl.py | 34 def main_op(): function
70 with ops.control_dependencies([main_op()]):
|
| D | main_op.py | 26 from tensorflow.python.saved_model.main_op_impl import main_op
|
| D | saved_model.py | 27 from tensorflow.python.saved_model import main_op
|
| D | simple_save.py | 89 main_op=legacy_init_op,
|
| D | BUILD | 29 ":main_op",
146 name = "main_op",
148 "main_op.py",
172 ":main_op",
|
| D | loader_test.py | 97 sess, ["foo_graph"], sig_def_map, main_op=assign_op)
|
| D | saved_model_test.py | 46 from tensorflow.python.saved_model import main_op
896 with ops.control_dependencies([main_op.main_op()]):
1474 builder.add_meta_graph_and_variables(sess, ["foo"], main_op=init_op)
|
| /external/tensorflow/tensorflow/tools/api/golden/v1/ |
| D | tensorflow.saved_model.main_op.pbtxt | 1 path: "tensorflow.saved_model.main_op"
4 name: "main_op"
|
| D | tensorflow.saved_model.builder.-saved-model-builder.pbtxt | 12 …f_map\', \'assets_collection\', \'legacy_init_op\', \'clear_devices\', \'main_op\', \'strip_defaul…
16 …f_map\', \'assets_collection\', \'legacy_init_op\', \'clear_devices\', \'main_op\', \'strip_defaul…
|
| D | tensorflow.saved_model.-builder.pbtxt | 12 …f_map\', \'assets_collection\', \'legacy_init_op\', \'clear_devices\', \'main_op\', \'strip_defaul…
16 …f_map\', \'assets_collection\', \'legacy_init_op\', \'clear_devices\', \'main_op\', \'strip_defaul…
|
| D | tensorflow.saved_model.pbtxt | 136 name: "main_op"
|
| /external/tensorflow/tensorflow/compiler/mlir/tensorflow/tests/tf_saved_model/ |
| D | common_v1.py | 97 main_op=init_op,
|
| /external/tensorflow/tensorflow/compiler/mlir/lite/ |
| D | flatbuffer_export.cc | 1575 FuncOp main_op, const std::string& saved_model_tag) { in BuildSignaturedef() argument
1581 main_op.getAllArgAttrs(arg_attrs); in BuildSignaturedef()
1582 main_op.getAllResultAttrs(res_attrs); in BuildSignaturedef()
1594 main_op->getAttrOfType<mlir::DictionaryAttr>(kEntryFunctionAttributes); in BuildSignaturedef()
1604 if (input_names.size() != main_op.getNumArguments()) { in BuildSignaturedef()
1605 main_op.emitWarning() << "invalid entry function specification"; in BuildSignaturedef()
1609 auto term = main_op.back().getTerminator(); in BuildSignaturedef()
1611 main_op.emitWarning() << "output names (" << output_names.size() in BuildSignaturedef()
1620 main_op.emitWarning( in BuildSignaturedef()
1627 main_op->getAttrOfType<mlir::ArrayAttr>("tf_saved_model.exported_names"); in BuildSignaturedef()
[all …]
|
| /external/tensorflow/tensorflow/python/tools/api/generator/ |
| D | api_init_files_v1.bzl | 77 "saved_model/main_op/__init__.py",
|
| /external/tensorflow/tensorflow/python/distribute/integration_test/ |
| D | saved_model_test.py | 472 main_op=tf1.tables_initializer(),
|
| /external/tensorflow/tensorflow/lite/python/ |
| D | lite_v2_test.py | 858 main_op=init_op,
|
| /external/tensorflow/ |
| D | RELEASE.md | 2612 `saved_model.load` and `saved_model.main_op`, which will be replaced by
2613 `saved_model.main_op` in V2.
2945 MetaGraphs has been deprecated. Please use the equivalent main_op
2947 main_op or legacy_init_op at the time of SavedModel building, whereas
2948 the check on main_op was previously only done at load time.
|