- Sort Score
- Result 10 results
- Languages All
Results 1 - 10 of 13 for function_aliases_ (0.22 sec)
-
tensorflow/compiler/mlir/quantization/stablehlo/cc/calibration/component.cc
py_function_lib_(ABSL_DIE_IF_NULL(py_function_lib)), // Crash OK src_saved_model_path_(src_saved_model_path), function_aliases_(std::move(function_aliases)), tags_(std::move(tags)), signature_def_map_(std::move(signature_def_map)), signature_keys_(std::move(signature_keys)) {} absl::Status CalibrationComponent::ExportToSavedModel( ModuleOp module_op, absl::string_view calibration_data_dir,
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue May 14 06:31:57 UTC 2024 - 9.2K bytes - Viewed (0) -
tensorflow/compiler/mlir/quantization/stablehlo/cc/calibration/component.h
std::string src_saved_model_path_; // Function alias mapping for pre-calibrated SavedModel. Used to preserve // aliased functions. absl::flat_hash_map<FunctionName, FunctionAlias> function_aliases_; // Tags to identify the MetaGraphDef to load from a SavedModel. const std::unordered_set<std::string> tags_; const absl::flat_hash_map<std::string, tensorflow::SignatureDef> signature_def_map_;
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue May 14 06:31:57 UTC 2024 - 5.4K bytes - Viewed (0) -
tensorflow/compiler/mlir/quantization/stablehlo/cc/saved_model_import_test.cc
return %arg : tensor<1x2xf32> } )mlir"); ASSERT_TRUE(module_op); absl::flat_hash_map<FunctionName, FunctionAlias> function_aliases; UpdateFunctionAliases(function_aliases, *module_op); EXPECT_THAT(function_aliases, IsEmpty()); } TEST_F(UpdateFunctionAliasesTest, AliasUpdatedByMlirFunctionName) { // MLIR @main function corresponds to the TF function "main_original".
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Mar 07 03:47:17 UTC 2024 - 4.6K bytes - Viewed (0) -
tensorflow/compiler/mlir/quantization/stablehlo/cc/static_range_ptq.cc
absl::StatusOr<absl::flat_hash_map<FunctionName, FunctionAlias>> function_aliases = GetFunctionAliases(src_saved_model_path, tags); if (!function_aliases.ok()) { return absl::InternalError(absl::StrCat( "Failed to get function alias: ", function_aliases.status().message())); } TF_ASSIGN_OR_RETURN( OwningOpRef<ModuleOp> module,
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed Apr 24 12:49:45 UTC 2024 - 6K bytes - Viewed (0) -
tensorflow/compiler/mlir/quantization/stablehlo/cc/saved_model_import.cc
absl::flat_hash_map<FunctionName, FunctionAlias> function_aliases( meta_graph.meta_info_def().function_aliases().begin(), meta_graph.meta_info_def().function_aliases().end()); return function_aliases; } void UpdateFunctionAliases( absl::flat_hash_map<FunctionName, FunctionAlias>& function_aliases, ModuleOp module_op) { absl::flat_hash_set<FunctionName> existing_func_names;
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed Apr 24 12:49:45 UTC 2024 - 6.4K bytes - Viewed (0) -
tensorflow/compiler/mlir/quantization/stablehlo/cc/weight_only_ptq.cc
absl::StatusOr<absl::flat_hash_map<FunctionName, FunctionAlias>> function_aliases = GetFunctionAliases(src_saved_model_path, tags); if (!function_aliases.ok()) { return absl::InternalError(absl::StrCat( "Failed to get function alias: ", function_aliases.status().message())); } TF_ASSIGN_OR_RETURN( auto module,
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Fri May 03 02:59:01 UTC 2024 - 5.1K bytes - Viewed (0) -
tensorflow/compiler/mlir/quantization/stablehlo/cc/saved_model_import.h
// Updates the function aliases. `module_op` may have different // function names from the original model, so it re-associates the aliases // with the new function names. Both the input `function_aliases` and the // returned value are function name -> alias mappings. `function_aliases` is // the function alias mapping of the original function. The original function's // name is retrieved by looking at the "tf._original_func_name" string attribute
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed Apr 24 12:49:45 UTC 2024 - 4.2K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/quantization/stablehlo/quantization.cc
const SavedModelBundle& saved_model_bundle) { const protobuf::Map<std::string, std::string>& function_aliases = saved_model_bundle.meta_graph_def.meta_info_def().function_aliases(); return absl::flat_hash_map<std::string, std::string>(function_aliases.begin(), function_aliases.end()); } } // namespace absl::StatusOr<mlir::ModuleOp> RunQuantization(
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Sun Apr 14 10:49:12 UTC 2024 - 7.3K bytes - Viewed (0) -
tensorflow/compiler/mlir/quantization/stablehlo/cc/saved_model_export.h
const std::unordered_set<std::string>& tags, const ::stablehlo::quantization::QuantizationConfig& quantization_config, absl::string_view debug_name_prefix, const absl::flat_hash_map<FunctionName, FunctionAlias>& function_aliases, MLIRContext& ctx ABSL_ATTRIBUTE_LIFETIME_BOUND, ModuleOp module_op); // Factory function for `ExportedModel`. [[nodiscard]] tensorflow::quantization::ExportedModel CreateExportedModelFromGraphDef(
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed Mar 20 11:11:25 UTC 2024 - 6.9K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow_to_stablehlo/tf_to_stablehlo.cc
std::unordered_set<std::string> tag_set(tag_names.begin(), tag_names.end()); TF_ASSIGN_OR_RETURN( auto function_aliases, quant::stablehlo::GetFunctionAliases(input_path, tag_set)); quant::stablehlo::UpdateFunctionAliases(function_aliases, *module_op); absl::c_for_each(function_aliases, [&](const auto& aliases) { return aliased_function_names.insert(aliases.first); }); }
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue May 21 22:58:42 UTC 2024 - 5.5K bytes - Viewed (0)