Searched refs:init_op_name (Results 1 – 5 of 5) sorted by relevance
| /external/tensorflow/tensorflow/cc/saved_model/ |
| D | loader_util.cc | 32 string* init_op_name) { in GetInitOp() argument
37 *init_op_name = init_op_sig_it->second.outputs() in GetInitOp()
58 *init_op_name = init_op_it->second.node_list().value(0); in GetInitOp()
|
| D | loader.cc | 188 Session* session, const string& init_op_name) { in RunInitOp() argument
189 if (!init_op_name.empty()) { in RunInitOp()
195 return RunOnce(run_options, inputs, {}, {init_op_name}, in RunInitOp()
419 string init_op_name; in RestoreSession() local
421 internal::GetInitOp(export_dir, meta_graph, &init_op_name)); in RestoreSession()
423 asset_file_defs, session->get(), init_op_name)); in RestoreSession()
|
| D | loader_util.h | 31 string* init_op_name);
|
| /external/tensorflow/tensorflow/python/grappler/ |
| D | memory_optimizer_test.py | 231 self, config, metagraph, init_op_name, train_op_name, loss_op_name): argument
235 init_op = graph.get_operation_by_name(init_op_name)
246 (original_metagraph, init_op_name, train_op_name, loss_op_name
251 init_op_name=init_op_name,
257 init_op_name=init_op_name,
|
| /external/tensorflow/tensorflow/compiler/mlir/tensorflow/translate/ |
| D | import_model.cc | 3428 std::string init_op_name; in CheckGraphNameValidity() local
3429 if (internal::GetInitOp("", meta_graph_def(), &init_op_name).ok()) { in CheckGraphNameValidity()
3430 if (init_op_name == name) return true; in CheckGraphNameValidity()
3775 std::string init_op_name; in ConvertSignatures() local
3777 internal::GetInitOp("", input_.meta_graph_def(), &init_op_name)); in ConvertSignatures()
3778 if (!init_op_name.empty()) { in ConvertSignatures()
3779 TF_RETURN_IF_ERROR(ConvertInitializer(init_op_name, assets)); in ConvertSignatures()
3780 init_sym_refs.push_back(builder.getSymbolRefAttr(init_op_name)); in ConvertSignatures()
|