Home
last modified time | relevance | path

Searched refs:allow_custom_ops (Results 1 – 24 of 24) 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.cc246 params.allow_custom_ops = true; in TEST_F()
260 params.allow_custom_ops = true; in TEST_F()
281 params.allow_custom_ops = true; in TEST_F()
300 params.allow_custom_ops = false; in TEST_F()
324 params.allow_custom_ops = false; in TEST_F()
354 params.allow_custom_ops = false; in TEST_F()
382 params.allow_custom_ops = false; in TEST_F()
433 params_.allow_custom_ops = false; in SetAllowedOpSets()
446 params_.allow_custom_ops = true; in SetAllowedOpSets()
Dexport.cc534 if (!params.allow_custom_ops) { in Export()
/external/tensorflow/tensorflow/lite/python/
Dconvert.py316 allow_custom_ops=False, argument
339 toco.allow_custom_ops = allow_custom_ops
374 allow_custom_ops=False, argument
485 allow_custom_ops, custom_opdefs,
Dtflite_convert.py185 if flags.allow_custom_ops:
186 converter.allow_custom_ops = flags.allow_custom_ops
335 if flags.custom_opdefs and not flags.allow_custom_ops:
Dlite.py435 self.allow_custom_ops = False
511 "allow_custom_ops": self.allow_custom_ops,
Dlite_v2_test.py875 converter.allow_custom_ops = True
Dlite_test.py1888 converter.allow_custom_ops = True
/external/tflite-support/tensorflow_lite_support/custom_ops/python/
Dsentencepiece_tokenizer_test.py127 converter.allow_custom_ops = True
170 converter.allow_custom_ops = True
208 converter.allow_custom_ops = True
Dragged_tensor_to_tensor_test.py39 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,
Dtoco_tooling.cc476 bool allow_custom_ops, in Export() argument
487 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/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.py95 self.allow_custom_ops = False
/external/tensorflow/tensorflow/compiler/mlir/lite/python/
Dtf_tfl_flatbuffer_helpers.cc297 bool emit_custom_ops = toco_flags.allow_custom_ops(); in ConvertMLIRToTFLiteFlatBuffer()
/external/tflite-support/tensorflow_lite_support/custom_ops/kernel/
Dngrams_test.py101 converter.allow_custom_ops = not flex
/external/tensorflow/tensorflow/lite/g3doc/guide/
Dops_custom.md101 attribute `allow_custom_ops` as shown below:
105 <b>converter.allow_custom_ops = True</b>
/external/tensorflow/tensorflow/lite/g3doc/convert/
Dindex.md183 converter.allow_custom_ops = True
/external/tensorflow/tensorflow/lite/g3doc/r1/convert/
Dcmdline_reference.md161 * `--allow_custom_ops`. Type: bool. Default: False. Indicates whether to allow