Home
last modified time | relevance | path

Searched refs:custom_opdefs (Results 1 – 13 of 13) sorted by relevance

/external/tensorflow/tensorflow/compiler/mlir/lite/python/
Dsaved_model_to_tfl_flatbuffer.cc160 std::vector<std::string> custom_opdefs(toco_flags.custom_opdefs().begin(), in ConvertSavedModelToTFLiteFlatBuffer() local
161 toco_flags.custom_opdefs().end()); in ConvertSavedModelToTFLiteFlatBuffer()
165 absl::MakeSpan(custom_opdefs), in ConvertSavedModelToTFLiteFlatBuffer()
Dtf_tfl_flatbuffer_helpers.cc186 std::vector<string> extra_tf_opdefs(toco_flags.custom_opdefs().begin(), in RegisterAllCustomOps()
187 toco_flags.custom_opdefs().end()); in RegisterAllCustomOps()
/external/tensorflow/tensorflow/compiler/mlir/tfjs/translate/
Dtf_tfjs_translate.cc101 llvm::cl::list<std::string> custom_opdefs( variable
138 custom_opdefs, input_file_name, saved_model_tags, in main()
142 input_file_name, input_mlir, custom_opdefs, debug_info_file, in main()
/external/tensorflow/tensorflow/lite/python/
Dconvert.py317 custom_opdefs=None, argument
340 if custom_opdefs:
341 toco.custom_opdefs.extend(custom_opdefs)
375 custom_opdefs=None, argument
485 allow_custom_ops, custom_opdefs,
Dtflite_convert.py136 if flags.custom_opdefs:
137 register_custom_opdefs(_parse_array(flags.custom_opdefs))
332 if flags.custom_opdefs and not flags.experimental_new_converter:
335 if flags.custom_opdefs and not flags.allow_custom_ops:
/external/tensorflow/tensorflow/compiler/mlir/lite/
Dtf_tfl_translate.cc174 std::vector<std::string> extra_opdefs(custom_opdefs.begin(), in main()
175 custom_opdefs.end()); in main()
181 input_file_name, input_mlir, use_splatted_constant, custom_opdefs, in main()
Dtf_tfl_translate_cl.h38 extern llvm::cl::list<std::string> custom_opdefs;
Dtf_tfl_translate_cl.cc84 llvm::cl::list<std::string> custom_opdefs( variable
/external/tensorflow/tensorflow/lite/toco/
Dtoco_cmdline_flags.cc127 Flag("custom_opdefs", parsed_flags.custom_opdefs.bind(), in ParseTocoFlagsFromCommandLineFlags()
128 parsed_flags.custom_opdefs.default_value(), in ParseTocoFlagsFromCommandLineFlags()
Dargs.h176 Arg<std::string> custom_opdefs; member
Dtoco_flags.proto228 repeated string custom_opdefs = 32; field
/external/tensorflow/tensorflow/lite/g3doc/r1/convert/
Dcmdline_reference.md166 * `--custom_opdefs`. Type: string. String representing a list of custom ops
/external/tensorflow/tensorflow/compiler/mlir/lite/ir/
Dtfl_ops.td4481 custom_opdefs or linked which are not defined in TF dialect.