- Sort Score
- Result 10 results
- Languages All
Results 1 - 9 of 9 for FlatbufferExportOptions (0.24 sec)
-
tensorflow/compiler/mlir/lite/flatbuffer_export.h
#include "tensorflow/compiler/mlir/op_or_arg_name_mapper.h" #include "tensorflow/lite/toco/toco_flags.pb.h" namespace tflite { // Options for exporting to Flatbuffer. struct FlatbufferExportOptions { // TocoFlags proto. The following fields are migrated. // bool emit_builtin_tflite_ops -> !toco_flags.force_select_tf_ops() // bool emit_select_tf_ops -> toco_flags.enable_select_tf_ops()
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Nov 09 02:51:43 UTC 2023 - 2.8K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/sparsity/sparsify_model.cc
return absl::InternalError(absl::StrCat( "Failed to sparsify: ", statusHandler.ConsumeStatus().message())); } // Export the results to the builder std::string result; tflite::FlatbufferExportOptions options; options.toco_flags.set_force_select_tf_ops(false); options.toco_flags.set_enable_select_tf_ops(true); options.toco_flags.set_allow_custom_ops(true);
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Mon Jun 10 20:16:40 UTC 2024 - 4.3K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/experimental/tac/utils/utils.cc
} std::string result; if (output_mlir) { llvm::raw_string_ostream os(result); module.print(os); os.flush(); } else { tflite::FlatbufferExportOptions options; options.toco_flags.set_force_select_tf_ops(false); options.toco_flags.set_allow_custom_ops(true); if (enable_select_tf_ops) { options.toco_flags.set_enable_select_tf_ops(true);
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Mon Jun 03 03:47:03 UTC 2024 - 4.5K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/quantization/lite/quantize_weights.cc
LOG(ERROR) << "Failed to quantize: " << err; return kTfLiteError; } // Export the results to the builder std::string result; tflite::FlatbufferExportOptions options; options.toco_flags.set_force_select_tf_ops(false); options.toco_flags.set_enable_select_tf_ops(true); options.toco_flags.set_allow_custom_ops(true);
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed Jun 12 23:15:24 UTC 2024 - 9.5K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/quantization/lite/quantize_model.cc
const std::string err(statusHandler.ConsumeStatus().message()); LOG(ERROR) << "Failed to quantize: " << err; return kTfLiteError; } // Export the results. tflite::FlatbufferExportOptions options; options.toco_flags.set_force_select_tf_ops(false); options.toco_flags.set_enable_select_tf_ops(true); options.toco_flags.set_allow_custom_ops(true);
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed Jun 12 23:15:24 UTC 2024 - 6.3K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/mlir_tflite_runner.cc
if (main.getFunctionType().getNumInputs() != 0) LOG(QFATAL) << "NYI: Only nullary functions supported."; // Convert to flatbuffer. std::string serialized_flatbuffer; tflite::FlatbufferExportOptions options; options.toco_flags.set_force_select_tf_ops(!emit_builtin_tflite_ops); options.toco_flags.set_enable_select_tf_ops(emit_select_tf_ops); options.toco_flags.set_allow_custom_ops(emit_custom_ops);
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Sat Jun 03 00:14:05 UTC 2023 - 6.3K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/flatbuffer_translate.cc
op_or_arg_name_mapper = std::make_unique<tensorflow::OpOrArgStripNameMapper>(); } else { op_or_arg_name_mapper = std::make_unique<tensorflow::OpOrArgLocNameMapper>(); } tflite::FlatbufferExportOptions options; options.toco_flags.set_force_select_tf_ops(!emit_builtin_tflite_ops); options.toco_flags.set_enable_select_tf_ops(emit_select_tf_ops); options.toco_flags.set_allow_custom_ops(emit_custom_ops);
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Mar 14 19:15:40 UTC 2024 - 9.4K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/tf_to_tfl_flatbuffer.cc
return status_handler.Combine( absl::InvalidArgumentError("VHLO lowering failed")); } // Write MLIR Stablehlo dialect into FlatBuffer OpOrArgLocNameMapper op_or_arg_name_mapper; tflite::FlatbufferExportOptions options; options.toco_flags = toco_flags; options.saved_model_tags = saved_model_tags; options.op_or_arg_name_mapper = &op_or_arg_name_mapper; options.metadata[tflite::kModelUseStablehloTensorKey] = "true";
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Fri May 03 18:01:23 UTC 2024 - 23.8K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/flatbuffer_export.cc
} return control_edges; } } // namespace namespace tflite { bool MlirToFlatBufferTranslateFunction(mlir::ModuleOp module, const FlatbufferExportOptions& options, std::string* serialized_flatbuffer, bool serialize_stablehlo_ops) { auto maybe_translated = Translator::Translate(
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed Jun 12 21:41:49 UTC 2024 - 164.5K bytes - Viewed (2)