Home
last modified time | relevance | path

Searched refs:model_flags (Results 1 – 24 of 24) 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(std::string(input_array)); in ReadModelFlagsFromCommandLineFlags()
259 model_flags->mutable_input_arrays(0)->set_mean_value( in ReadModelFlagsFromCommandLineFlags()
267 model_flags->input_arrays_size()); in ReadModelFlagsFromCommandLineFlags()
270 model_flags->mutable_input_arrays(i)->set_mean_value( in ReadModelFlagsFromCommandLineFlags()
277 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.cc687 const ModelFlags& model_flags, Model* model) { in ConvertUnsupportedOperator() argument
775 const ModelFlags& model_flags, Model* model) { in ConvertConstOperator() argument
831 const ModelFlags& model_flags, Model* model) { in ConvertConvOperator() argument
912 const ModelFlags& model_flags, Model* model) { in ConvertDepthwiseConvOperator() argument
990 const ModelFlags& model_flags, Model* model) { in ConvertDepthToSpaceOperator() argument
1013 const ModelFlags& model_flags, Model* model) { in ConvertSpaceToDepthOperator() argument
1036 const ModelFlags& model_flags, Model* model) { in ConvertBiasAddOperator() argument
1053 const ModelFlags& model_flags, Model* model) { in ConvertRandomUniform() argument
1071 const ModelFlags& model_flags, Model* model) { in ConvertIdentityOperator() argument
1094 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.cc869 void CheckInputArraysAreNotOutputArrays(const ModelFlags& model_flags) { in CheckInputArraysAreNotOutputArrays() argument
870 for (const auto& input_array : model_flags.input_arrays()) { in CheckInputArraysAreNotOutputArrays()
871 for (const std::string& output_array : model_flags.output_arrays()) { in CheckInputArraysAreNotOutputArrays()
902 void CheckNonAsciiIOArrays(const ModelFlags& model_flags) { in CheckNonAsciiIOArrays() argument
903 if (model_flags.allow_nonascii_arrays()) { in CheckNonAsciiIOArrays()
906 for (const auto& input_array : model_flags.input_arrays()) { in CheckNonAsciiIOArrays()
914 for (const std::string& output_array : model_flags.output_arrays()) { in CheckNonAsciiIOArrays()
1515 void ResolveModelFlags(const ModelFlags& model_flags, Model* model) { in ResolveModelFlags() argument
1517 for (const auto& specified_input_array : model_flags.input_arrays()) { in ResolveModelFlags()
1534 model_flags.input_arrays_size() == 1 && in ResolveModelFlags()
[all …]
Dmodel_cmdline_flags.h35 const ParsedModelFlags& parsed_model_flags, ModelFlags* model_flags);
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()
Dtoco_convert.h27 const ModelFlags& model_flags,
Dtoco_tooling.h29 const ModelFlags& model_flags,
Dimport_tensorflow_test.cc48 const ModelFlags& model_flags, Model* model);
54 const ModelFlags& model_flags, Model*,
Dtooling_util.h201 void ResolveModelFlags(const ModelFlags& model_flags, Model* model);
DBUILD34 srcs = ["model_flags.proto"],
/external/tensorflow/tensorflow/compiler/mlir/lite/python/
Dsaved_model_to_tfl_flatbuffer.cc49 Status HandleInputOutputArraysWithModule(const toco::ModelFlags& model_flags, in HandleInputOutputArraysWithModule() argument
72 function_input_names.reserve(model_flags.input_arrays().size()); in HandleInputOutputArraysWithModule()
80 if (function_input_names_size != model_flags.input_arrays().size()) { in HandleInputOutputArraysWithModule()
83 ", expected: ", model_flags.input_arrays().size()); in HandleInputOutputArraysWithModule()
88 for (const auto& input_array : model_flags.input_arrays()) { in HandleInputOutputArraysWithModule()
97 function_output_names.reserve(model_flags.output_arrays().size()); in HandleInputOutputArraysWithModule()
105 if (function_output_names_size != model_flags.output_arrays().size()) { in HandleInputOutputArraysWithModule()
108 ", expected: ", model_flags.output_arrays().size()); in HandleInputOutputArraysWithModule()
113 for (const auto& output_array : model_flags.output_arrays()) { in HandleInputOutputArraysWithModule()
122 Status ConvertSavedModelToTFLiteFlatBuffer(const toco::ModelFlags& model_flags, in ConvertSavedModelToTFLiteFlatBuffer() argument
[all …]
Dgraphdef_to_tfl_flatbuffer.cc46 Status ConvertGraphDefToTFLiteFlatBuffer(const toco::ModelFlags& model_flags, in ConvertGraphDefToTFLiteFlatBuffer() argument
64 model_flags, toco_flags, &quant_specs, &node_names, &node_dtypes, in ConvertGraphDefToTFLiteFlatBuffer()
71 std::vector<string> output_arrays(model_flags.output_arrays().begin(), in ConvertGraphDefToTFLiteFlatBuffer()
72 model_flags.output_arrays().end()); in ConvertGraphDefToTFLiteFlatBuffer()
80 internal::WarningUnusedFlags(model_flags, toco_flags); in ConvertGraphDefToTFLiteFlatBuffer()
Dtf_tfl_flatbuffer_helpers.cc195 const toco::ModelFlags& model_flags, const toco::TocoFlags& toco_flags, in PopulateQuantizationSpecs() argument
211 for (auto& flag : model_flags.input_arrays()) { in PopulateQuantizationSpecs()
335 void WarningUnusedFlags(const toco::ModelFlags& model_flags, in WarningUnusedFlags() argument
346 if (model_flags.change_concat_input_ranges()) { in WarningUnusedFlags()
352 if (model_flags.allow_nonexistent_arrays()) { in WarningUnusedFlags()
Dtf_tfl_flatbuffer_helpers.h41 const toco::ModelFlags& model_flags, const toco::TocoFlags& toco_flags,
57 void WarningUnusedFlags(const toco::ModelFlags& model_flags,
Dsaved_model_to_tfl_flatbuffer.h30 const toco::ModelFlags& model_flags, const toco::TocoFlags& toco_flags,
Dgraphdef_to_tfl_flatbuffer.h29 Status ConvertGraphDefToTFLiteFlatBuffer(const toco::ModelFlags& model_flags,
/external/tensorflow/tensorflow/lite/toco/python/
Dtoco_python_api.cc51 void PopulateConversionLogHelper(const toco::ModelFlags& model_flags, in PopulateConversionLogHelper() argument
63 toco::Import(*toco_flags, model_flags, input_contents_txt); in PopulateConversionLogHelper()
77 toco::Import(*toco_flags, model_flags, output_file_contents_txt); in PopulateConversionLogHelper()
126 toco::ModelFlags model_flags; in TocoConvert() local
127 if (!model_flags.ParseFromString(model_flags_proto_txt)) { in TocoConvert()
155 if (model_flags.saved_model_dir().empty()) { in TocoConvert()
182 if (!model_flags.saved_model_dir().empty()) { in TocoConvert()
184 model_flags, toco_flags, &output_file_contents_txt); in TocoConvert()
194 model_flags, toco_flags, debug_info, graph_def, in TocoConvert()
198 model_flags, &toco_flags, input_contents_txt, in TocoConvert()
[all …]
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/lite/python/
Dconvert.py571 model_flags, toco_flags, _ = build_toco_convert_protos(
575 input_array = model_flags.input_arrays.add()
589 model_flags.output_arrays.append(name)
592 model_flags.SerializeToString(),
624 model_flags, toco_flags, debug_info = build_toco_convert_protos(
628 model_flags.SerializeToString(),
642 model_flags = _model_flags_pb2.ModelFlags()
644 model_flags.saved_model_dir = saved_model_dir
645 model_flags.saved_model_version = saved_model_version
647 model_flags.saved_model_tags.extend(saved_model_tags)
[all …]
/external/tensorflow/tensorflow/lite/toco/tflite/
Dimport.cc169 void ImportIOTensors(const ModelFlags& model_flags, in ImportIOTensors() argument
173 if (model_flags.input_arrays().empty()) { in ImportIOTensors()
184 if (model_flags.output_arrays().empty()) { in ImportIOTensors()
202 std::unique_ptr<Model> Import(const ModelFlags& model_flags, in Import() argument
231 ImportIOTensors(model_flags, *input_model, tensors_table, model.get()); in Import()
Dimport.h26 std::unique_ptr<Model> Import(const ModelFlags &model_flags,