Home
last modified time | relevance | path

Searched refs:model_flags (Results 1 – 20 of 20) 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()
71 ModelFlags model_flags; in TEST() local
[all …]
Dmodel_cmdline_flags.cc185 const ParsedModelFlags& parsed_model_flags, ModelFlags* model_flags) { in ReadModelFlagsFromCommandLineFlags() argument
200 model_flags)) in ReadModelFlagsFromCommandLineFlags()
216 model_flags->add_output_arrays(parsed_model_flags.output_array.value()); in ReadModelFlagsFromCommandLineFlags()
223 model_flags->add_output_arrays(output_array); in ReadModelFlagsFromCommandLineFlags()
247 model_flags->add_input_arrays()->set_name( in ReadModelFlagsFromCommandLineFlags()
254 model_flags->add_input_arrays()->set_name(string(input_array)); in ReadModelFlagsFromCommandLineFlags()
259 model_flags->mutable_input_arrays(0)->set_mean_value( in ReadModelFlagsFromCommandLineFlags()
266 QCHECK(mean_values.size() == model_flags->input_arrays_size()); in ReadModelFlagsFromCommandLineFlags()
269 model_flags->mutable_input_arrays(i)->set_mean_value( in ReadModelFlagsFromCommandLineFlags()
276 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
80 Import(toco_flags, model_flags, graph_def_contents); in Convert()
92 ModelFlags model_flags; in Convert() local
93 ReadModelFlagsFromCommandLineFlags(parsed_model_flags, &model_flags); in Convert()
100 &model_flags, &graph_def_contents); in Convert()
104 TF_RETURN_IF_ERROR(Convert(graph_def_contents, toco_flags, model_flags, in Convert()
Dimport_tensorflow.cc651 const ModelFlags& model_flags, Model* model) { in ConvertUnsupportedOperator() argument
739 const ModelFlags& model_flags, Model* model) { in ConvertConstOperator() argument
791 const ModelFlags& model_flags, Model* model) { in ConvertConvOperator() argument
872 const ModelFlags& model_flags, Model* model) { in ConvertDepthwiseConvOperator() argument
950 const ModelFlags& model_flags, Model* model) { in ConvertDepthToSpaceOperator() argument
973 const ModelFlags& model_flags, Model* model) { in ConvertSpaceToDepthOperator() argument
996 const ModelFlags& model_flags, Model* model) { in ConvertBiasAddOperator() argument
1013 const ModelFlags& model_flags, Model* model) { in ConvertRandomUniform() argument
1031 const ModelFlags& model_flags, Model* model) { in ConvertIdentityOperator() argument
1054 const ModelFlags& model_flags, Model* model) { in ConvertIdentityNOperator() argument
[all …]
Dimport_tensorflow.h39 const ModelFlags& model_flags, const TensorFlowImportFlags& tf_import_flags,
45 const ModelFlags& model_flags, const TensorFlowImportFlags& tf_import_flags,
Dtooling_util.cc866 void CheckInputArraysAreNotOutputArrays(const ModelFlags& model_flags) { in CheckInputArraysAreNotOutputArrays() argument
867 for (const auto& input_array : model_flags.input_arrays()) { in CheckInputArraysAreNotOutputArrays()
868 for (const string& output_array : model_flags.output_arrays()) { in CheckInputArraysAreNotOutputArrays()
899 void CheckNonAsciiIOArrays(const ModelFlags& model_flags) { in CheckNonAsciiIOArrays() argument
900 if (model_flags.allow_nonascii_arrays()) { in CheckNonAsciiIOArrays()
903 for (const auto& input_array : model_flags.input_arrays()) { in CheckNonAsciiIOArrays()
911 for (const string& output_array : model_flags.output_arrays()) { in CheckNonAsciiIOArrays()
1512 void ResolveModelFlags(const ModelFlags& model_flags, Model* model) { in ResolveModelFlags() argument
1514 for (const auto& specified_input_array : model_flags.input_arrays()) { in ResolveModelFlags()
1531 model_flags.input_arrays_size() == 1 && in ResolveModelFlags()
[all …]
Dtoco_tooling.cc211 const ModelFlags& model_flags, in Import() argument
225 model = ImportTensorFlowGraphDef(model_flags, tf_import_flags, in Import()
230 model = toco::tflite::Import(model_flags, input_file_contents); in Import()
231 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,
DBUILD35 srcs = ["model_flags.proto"],
61 "model_flags.proto",
Dimport_tensorflow_test.cc47 const ModelFlags& model_flags, Model* model);
53 const ModelFlags& model_flags, Model*,
Dtooling_util.h203 void ResolveModelFlags(const ModelFlags& model_flags, Model* model);
/external/tensorflow/tensorflow/lite/toco/python/
Dtoco_python_api.cc40 void PopulateConversionLogHelper(const toco::ModelFlags& model_flags, in PopulateConversionLogHelper() argument
52 toco::Import(*toco_flags, model_flags, input_contents_txt); in PopulateConversionLogHelper()
66 toco::Import(*toco_flags, model_flags, output_file_contents_txt); in PopulateConversionLogHelper()
119 toco::ModelFlags model_flags; in TocoConvert() local
120 if (!model_flags.ParseFromString(model_flags_proto_txt)) { in TocoConvert()
168 model_flags, toco_flags, debug_info, graph_def, in TocoConvert()
171 PopulateConversionLogHelper(model_flags, &toco_flags, input_contents_txt, in TocoConvert()
176 status = Convert(input_contents_txt, toco_flags, model_flags, in TocoConvert()
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())
/external/tensorflow/tensorflow/compiler/mlir/lite/python/
Dgraphdef_to_tfl_flatbuffer.cc105 void WarningUnusedFlags(const toco::ModelFlags& model_flags, in WarningUnusedFlags() argument
116 if (model_flags.change_concat_input_ranges()) { in WarningUnusedFlags()
122 if (model_flags.allow_nonexistent_arrays()) { in WarningUnusedFlags()
166 Status ConvertGraphDefToTFLiteFlatBuffer(const toco::ModelFlags& model_flags, in ConvertGraphDefToTFLiteFlatBuffer() argument
191 for (auto& flag : model_flags.input_arrays()) { in ConvertGraphDefToTFLiteFlatBuffer()
236 std::vector<string> output_arrays(model_flags.output_arrays().begin(), in ConvertGraphDefToTFLiteFlatBuffer()
237 model_flags.output_arrays().end()); in ConvertGraphDefToTFLiteFlatBuffer()
256 WarningUnusedFlags(model_flags, toco_flags); in ConvertGraphDefToTFLiteFlatBuffer()
Dgraphdef_to_tfl_flatbuffer.h29 Status ConvertGraphDefToTFLiteFlatBuffer(const toco::ModelFlags& model_flags,
/external/tensorflow/tensorflow/lite/python/
Dconvert.py437 model_flags, toco_flags, _ = build_toco_convert_protos(
441 input_array = model_flags.input_arrays.add()
453 model_flags.output_arrays.append(name)
456 model_flags.SerializeToString(),
488 model_flags, toco_flags, debug_info = build_toco_convert_protos(
492 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,