Home
last modified time | relevance | path

Searched refs:model_flags (Results 1 – 17 of 17) sorted by relevance

/external/tensorflow/tensorflow/lite/toco/
Dtoco_convert_test.cc26 ParsedModelFlags model_flags; in TEST() local
27 EXPECT_DEATH(Convert(toco_flags, model_flags).ok(), in TEST()
33 ModelFlags model_flags; in TEST() local
38 EXPECT_DEATH(Convert(input, toco_flags, model_flags, &output).ok(), in TEST()
44 ModelFlags model_flags; in TEST() local
50 EXPECT_DEATH(Convert(input, toco_flags, model_flags, &output).ok(), in TEST()
57 ModelFlags model_flags; in TEST() local
60 model_flags.add_output_arrays("output1"); in TEST()
64 EXPECT_DEATH(Convert(input, toco_flags, model_flags, &output).ok(), in TEST()
72 ModelFlags model_flags; in TEST() local
[all …]
Dmodel_cmdline_flags.cc182 const ParsedModelFlags& parsed_model_flags, ModelFlags* model_flags) { in ReadModelFlagsFromCommandLineFlags() argument
197 model_flags)) in ReadModelFlagsFromCommandLineFlags()
213 model_flags->add_output_arrays(parsed_model_flags.output_array.value()); in ReadModelFlagsFromCommandLineFlags()
220 model_flags->add_output_arrays(output_array); in ReadModelFlagsFromCommandLineFlags()
244 model_flags->add_input_arrays()->set_name( in ReadModelFlagsFromCommandLineFlags()
251 model_flags->add_input_arrays()->set_name(string(input_array)); in ReadModelFlagsFromCommandLineFlags()
256 model_flags->mutable_input_arrays(0)->set_mean_value( in ReadModelFlagsFromCommandLineFlags()
263 QCHECK(mean_values.size() == model_flags->input_arrays_size()); in ReadModelFlagsFromCommandLineFlags()
266 model_flags->mutable_input_arrays(i)->set_mean_value( in ReadModelFlagsFromCommandLineFlags()
273 model_flags->mutable_input_arrays(0)->set_std_value( in ReadModelFlagsFromCommandLineFlags()
[all …]
Dtoco_convert.cc57 TocoFlags* toco_flags, ModelFlags* model_flags, in ReadInputData() argument
76 const ModelFlags& model_flags, in Convert() argument
79 Import(toco_flags, model_flags, graph_def_contents); in Convert()
87 ModelFlags model_flags; in Convert() local
88 ReadModelFlagsFromCommandLineFlags(parsed_model_flags, &model_flags); in Convert()
95 &model_flags, &graph_def_contents); in Convert()
99 TF_RETURN_IF_ERROR(Convert(graph_def_contents, toco_flags, model_flags, in Convert()
Dimport_tensorflow.h38 const ModelFlags& model_flags, const TensorFlowImportFlags& tf_import_flags,
42 const ModelFlags& model_flags, const TensorFlowImportFlags& tf_import_flags,
Dtooling_util.cc841 void CheckInputArraysAreNotOutputArrays(const ModelFlags& model_flags) { in CheckInputArraysAreNotOutputArrays() argument
842 for (const auto& input_array : model_flags.input_arrays()) { in CheckInputArraysAreNotOutputArrays()
843 for (const string& output_array : model_flags.output_arrays()) { in CheckInputArraysAreNotOutputArrays()
874 void CheckNonAsciiIOArrays(const ModelFlags& model_flags) { in CheckNonAsciiIOArrays() argument
875 if (model_flags.allow_nonascii_arrays()) { in CheckNonAsciiIOArrays()
878 for (const auto& input_array : model_flags.input_arrays()) { in CheckNonAsciiIOArrays()
886 for (const string& output_array : model_flags.output_arrays()) { in CheckNonAsciiIOArrays()
1496 void ResolveModelFlags(const ModelFlags& model_flags, Model* model) { in ResolveModelFlags() argument
1498 for (const auto& specified_input_array : model_flags.input_arrays()) { in ResolveModelFlags()
1515 model_flags.input_arrays_size() == 1 && in ResolveModelFlags()
[all …]
Dtoco_tooling.cc206 const ModelFlags& model_flags, in Import() argument
220 model = ImportTensorFlowGraphDef(model_flags, tf_import_flags, in Import()
225 model = toco::tflite::Import(model_flags, input_file_contents); in Import()
226 ResolveModelFlags(model_flags, model.get()); in Import()
Dmodel_cmdline_flags.h35 const ParsedModelFlags& parsed_model_flags, ModelFlags* model_flags);
Dtoco_convert.h27 const ModelFlags& model_flags,
Dtoco_tooling.h29 const ModelFlags& model_flags,
DBUILD32 srcs = ["model_flags.proto"],
57 "model_flags.proto",
Dtooling_util.h204 void ResolveModelFlags(const ModelFlags& model_flags, Model* model);
Dimport_tensorflow.cc2559 const ModelFlags& model_flags, const TensorFlowImportFlags& tf_import_flags, in ImportTensorFlowGraphDef() argument
2569 for (const auto& specified_input_array : model_flags.input_arrays()) { in ImportTensorFlowGraphDef()
2574 for (const string& specified_output_array : model_flags.output_arrays()) { in ImportTensorFlowGraphDef()
2598 ResolveModelFlags(model_flags, model); in ImportTensorFlowGraphDef()
2616 const ModelFlags& model_flags, const TensorFlowImportFlags& tf_import_flags, in ImportTensorFlowGraphDef() argument
2626 return ImportTensorFlowGraphDef(model_flags, tf_import_flags, *tf_graph); in ImportTensorFlowGraphDef()
/external/tensorflow/tensorflow/lite/toco/python/
Dtoco_from_protos_test.py54 model_flags = model_flags_pb2.ModelFlags()
55 input_array = model_flags.input_arrays.add()
58 model_flags.output_arrays.append(TensorName(out_tensor))
64 fp_model.write(model_flags.SerializeToString())
Dtoco_python_api.cc60 toco::ModelFlags model_flags; in TocoConvert() local
61 if (!model_flags.ParseFromString(model_flags_proto_txt)) { in TocoConvert()
79 toco::Import(toco_flags, model_flags, input_contents_txt); in TocoConvert()
/external/tensorflow/tensorflow/lite/python/
Dconvert.py401 model_flags, toco_flags = build_toco_convert_protos(
405 input_array = model_flags.input_arrays.add()
417 model_flags.output_arrays.append(name)
419 data = toco_convert_protos(model_flags.SerializeToString(),
448 model_flags, toco_flags = build_toco_convert_protos(
450 data = toco_convert_protos(model_flags.SerializeToString(),
/external/tensorflow/tensorflow/lite/toco/tflite/
Dimport.cc168 void ImportIOTensors(const ModelFlags& model_flags, in ImportIOTensors() argument
172 if (model_flags.input_arrays().empty()) { in ImportIOTensors()
183 if (model_flags.output_arrays().empty()) { in ImportIOTensors()
201 std::unique_ptr<Model> Import(const ModelFlags& model_flags, in Import() argument
230 ImportIOTensors(model_flags, *input_model, tensors_table, model.get()); in Import()
Dimport.h26 std::unique_ptr<Model> Import(const ModelFlags &model_flags,