Searched refs:allow_custom_ops (Results 1 – 25 of 42) sorted by relevance
12
| /external/tensorflow/tensorflow/lite/toco/tflite/ |
| D | export.h | 32 bool allow_custom_ops = false; member
53 inline void Export(const Model& model, bool allow_custom_ops, in Export() argument
56 params.allow_custom_ops = allow_custom_ops; in Export()
66 const Model& model, bool allow_custom_ops, bool quantize_weights, in Export() argument
70 params.allow_custom_ops = allow_custom_ops; in Export()
81 params.allow_custom_ops = true; in Export()
|
| D | export_test.cc | 251 params.allow_custom_ops = true; in TEST_F()
265 params.allow_custom_ops = true; in TEST_F()
284 params.allow_custom_ops = true; in TEST_F()
301 params.allow_custom_ops = false; in TEST_F()
325 params.allow_custom_ops = false; in TEST_F()
355 params.allow_custom_ops = false; in TEST_F()
383 params.allow_custom_ops = false; in TEST_F()
434 params_.allow_custom_ops = false; in SetAllowedOpSets()
447 params_.allow_custom_ops = true; in SetAllowedOpSets()
|
| /external/tflite-support/tensorflow_lite_support/custom_ops/python/ |
| D | sentencepiece_tokenizer_test.py | 127 converter.allow_custom_ops = True
170 converter.allow_custom_ops = True
208 converter.allow_custom_ops = True
|
| D | ragged_tensor_to_tensor_test.py | 39 converter.allow_custom_ops = True
|
| /external/tensorflow/tensorflow/lite/toco/ |
| D | toco_cmdline_flags.cc | 125 Flag("allow_custom_ops", parsed_flags.allow_custom_ops.bind(), in ParseTocoFlagsFromCommandLineFlags()
126 parsed_flags.allow_custom_ops.default_value(), in ParseTocoFlagsFromCommandLineFlags()
288 READ_TOCO_FLAG(allow_custom_ops, FlagRequirement::kNone); in ReadTocoFlagsFromCommandLineFlags()
|
| D | toco_tooling.h | 45 bool allow_custom_ops,
|
| D | toco_tooling.cc | 476 bool allow_custom_ops, in Export() argument
487 params.allow_custom_ops = allow_custom_ops; in Export()
|
| D | toco_convert.cc | 81 TF_RETURN_IF_ERROR(Export(toco_flags, *model, toco_flags.allow_custom_ops(), in Convert()
|
| /external/tensorflow/tensorflow/lite/testing/ |
| D | mlir_convert.py | 56 converter.allow_custom_ops = extra_convert_options.allow_custom_ops
|
| /external/tensorflow/tensorflow/lite/python/ |
| D | tflite_convert.py | 210 if flags.allow_custom_ops:
211 converter.allow_custom_ops = flags.allow_custom_ops
357 if flags.custom_opdefs and not flags.allow_custom_ops:
|
| /external/tensorflow/tensorflow/lite/experimental/mlir/testing/ |
| D | mlir_convert.py | 60 converter.allow_custom_ops = extra_toco_options.allow_custom_ops
|
| /external/tensorflow/tensorflow/lite/testing/op_tests/ |
| D | max_pool_with_argmax.py | 60 extra_convert_options.allow_custom_ops = True
|
| D | irfft2d.py | 62 extra_convert_options.allow_custom_ops = True
|
| D | pool3d.py | 70 extra_convert_options.allow_custom_ops = True
|
| D | broadcast_gradient_args.py | 61 extra_convert_options.allow_custom_ops = True
|
| D | roll.py | 127 extra_convert_options.allow_custom_ops = True
|
| D | parse_example.py | 87 extra_convert_options.allow_custom_ops = True
|
| /external/tensorflow/tensorflow/lite/experimental/mlir/testing/op_tests/ |
| D | max_pool_with_argmax.py | 64 extra_toco_options.allow_custom_ops = True
|
| D | irfft2d.py | 66 extra_toco_options.allow_custom_ops = True
|
| D | dense_image_warp.py | 57 extra_toco_options.allow_custom_ops = True
|
| D | pool3d.py | 74 extra_toco_options.allow_custom_ops = True
|
| D | broadcast_gradient_args.py | 65 extra_toco_options.allow_custom_ops = True
|
| D | parse_example.py | 91 extra_toco_options.allow_custom_ops = True
|
| D | roll.py | 131 extra_toco_options.allow_custom_ops = True
|
| /external/tensorflow/tensorflow/lite/toco/python/ |
| D | toco_from_protos_test.py | 49 toco_flags.allow_custom_ops = True
|
12