Home
last modified time | relevance | path

Searched refs:allow_custom_ops (Results 1 – 20 of 20) sorted by relevance

/external/tensorflow/tensorflow/lite/toco/tflite/
Dexport.h30 bool allow_custom_ops = false; member
49 inline void Export(const Model& model, bool allow_custom_ops, in Export() argument
52 params.allow_custom_ops = allow_custom_ops; in Export()
62 const Model& model, bool allow_custom_ops, bool quantize_weights, in Export() argument
66 params.allow_custom_ops = allow_custom_ops; in Export()
77 params.allow_custom_ops = true; in Export()
Dexport_test.cc239 params.allow_custom_ops = true; in TEST_F()
253 params.allow_custom_ops = true; in TEST_F()
273 params.allow_custom_ops = true; in TEST_F()
291 params.allow_custom_ops = false; in TEST_F()
315 params.allow_custom_ops = false; in TEST_F()
345 params.allow_custom_ops = false; in TEST_F()
373 params.allow_custom_ops = false; in TEST_F()
424 params_.allow_custom_ops = false; in SetAllowedOpSets()
437 params_.allow_custom_ops = true; in SetAllowedOpSets()
Dexport.cc531 if (!params.allow_custom_ops) { in Export()
/external/tensorflow/tensorflow/lite/python/
Dtflite_convert.py175 if flags.allow_custom_ops:
176 converter.allow_custom_ops = flags.allow_custom_ops
307 if flags.custom_opdefs and not flags.allow_custom_ops:
Dconvert.py250 allow_custom_ops=False, argument
349 toco.allow_custom_ops = allow_custom_ops
Dlite.py171 self.allow_custom_ops = False
283 "allow_custom_ops": self.allow_custom_ops,
Dlite_test.py1500 converter.allow_custom_ops = True
/external/tensorflow/tensorflow/lite/toco/
Dtoco_cmdline_flags.cc123 Flag("allow_custom_ops", parsed_flags.allow_custom_ops.bind(), in ParseTocoFlagsFromCommandLineFlags()
124 parsed_flags.allow_custom_ops.default_value(), in ParseTocoFlagsFromCommandLineFlags()
276 READ_TOCO_FLAG(allow_custom_ops, FlagRequirement::kNone); in ReadTocoFlagsFromCommandLineFlags()
Dtoco_tooling.h45 bool allow_custom_ops, string* output_file_contents);
Dtoco_tooling.cc476 bool allow_custom_ops, string* output_file_contents) { in Export() argument
486 params.allow_custom_ops = allow_custom_ops; in Export()
Dtoco_convert.cc82 TF_RETURN_IF_ERROR(Export(toco_flags, *model, toco_flags.allow_custom_ops(), in Convert()
Dargs.h174 Arg<bool> allow_custom_ops = Arg<bool>(false); member
Dtoco_flags.proto138 optional bool allow_custom_ops = 10; field
198 // `enable_select_tf_ops` should always be used with `allow_custom_ops`.
/external/tensorflow/tensorflow/lite/testing/op_tests/
Drfft2d.py54 extra_toco_options.allow_custom_ops = True
/external/tensorflow/tensorflow/lite/toco/python/
Dtoco_from_protos_test.py53 toco_flags.allow_custom_ops = True
/external/tensorflow/tensorflow/lite/testing/
Dtoco_convert.py66 if extra_toco_options.allow_custom_ops:
Dzip_test_utils.py94 self.allow_custom_ops = False
/external/tensorflow/tensorflow/compiler/mlir/lite/python/
Dgraphdef_to_tfl_flatbuffer.cc251 bool emit_custom_ops = toco_flags.allow_custom_ops(); in ConvertGraphDefToTFLiteFlatBuffer()
/external/tensorflow/tensorflow/lite/g3doc/r1/convert/
Dcmdline_reference.md136 * `--allow_custom_ops`. Type: string. Default: False. Indicates whether to
/external/tensorflow/tensorflow/lite/g3doc/guide/
Dops_custom.md49 Optimizing Converter with `--allow_custom_ops` argument, and run it with the