- Sort Score
- Result 10 results
- Languages All
Results 1 - 5 of 5 for RegisterAllCustomOps (0.46 sec)
-
tensorflow/compiler/mlir/lite/python/tf_tfl_flatbuffer_helpers.h
#include "tensorflow/lite/toco/types.pb.h" #include "tsl/platform/statusor.h" namespace tensorflow { namespace internal { // Register all custom ops including user specified custom ops. Status RegisterAllCustomOps(const toco::TocoFlags& toco_flags); // Populate quantization specs (or not) given user specified ranges for each // input arrays. Status PopulateQuantizationSpecs(
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Sun May 12 12:39:37 UTC 2024 - 3.2K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/python/graphdef_to_tfl_flatbuffer.cc
specs.unconditionally_use_set_output_shapes = true; internal::WarningUnusedFlags(model_flags, toco_flags); // Register all custom ops, including user-specified custom ops. TF_RETURN_IF_ERROR(internal::RegisterAllCustomOps(toco_flags)); TF_ASSIGN_OR_RETURN( auto module, ConvertGraphdefToMlir(input, debug_info, specs, &context)); mlir::TFL::PassConfig pass_config(quant_specs);
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Mon Mar 11 19:29:56 UTC 2024 - 5.1K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/python/jax_to_tfl_flatbuffer.cc
&node_shapes, &node_mins, &node_maxs)); internal::WarningUnusedFlags(model_flags, toco_flags); // Register all custom ops, including user-specified custom ops. TF_RETURN_IF_ERROR(internal::RegisterAllCustomOps(toco_flags)); mlir::TFL::PassConfig pass_config(quant_specs); bool emit_builtin_tflite_ops = !toco_flags.force_select_tf_ops(); pass_config.emit_builtin_tflite_ops = emit_builtin_tflite_ops;
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Mon Mar 11 19:29:56 UTC 2024 - 8K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/python/saved_model_to_tfl_flatbuffer.cc
&node_shapes, &node_mins, &node_maxs)); internal::WarningUnusedFlags(model_flags, toco_flags); // Register all custom ops, including user-specified custom ops. TF_RETURN_IF_ERROR(internal::RegisterAllCustomOps(toco_flags)); auto& saved_model_tags = model_flags.saved_model_tags(); auto& saved_model_exported_names = model_flags.saved_model_exported_names(); std::unordered_set<std::string> tags(saved_model_tags.begin(),
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Sun May 12 12:39:37 UTC 2024 - 11K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/python/tf_tfl_flatbuffer_helpers.cc
*op_reg_data = OpRegistrationData(opdef); return absl::OkStatus(); }); } return absl::OkStatus(); } } // namespace absl::Status RegisterAllCustomOps(const toco::TocoFlags& toco_flags) { // Register any custom OpDefs. std::vector<std::string> extra_tf_opdefs(toco_flags.custom_opdefs().begin(),
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Sun May 12 12:39:37 UTC 2024 - 17.3K bytes - Viewed (0)