- Sort Score
- Result 10 results
- Languages All
Results 11 - 19 of 19 for serving_default (0.29 sec)
-
tensorflow/compiler/mlir/quantization/stablehlo/cc/calibration/representative_dataset.cc
// If the signature_key has not been explicitly specified, use the default // value of "serving_default". const std::string signature_key = dataset_config.has_signature_key() ? dataset_config.signature_key() : "serving_default"; if (repr_dataset_file_map.contains(signature_key)) { return absl::InvalidArgumentError(
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Mon Jan 29 04:53:21 UTC 2024 - 2.5K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow_to_stablehlo/python/pywrap_tensorflow_to_stablehlo.cc
// LINT.IfChange(savedmodel_to_stablehlo) m.def( "savedmodel_to_stablehlo", [](absl::string_view input_path, const std::vector<std::string>& exported_model_signatures = {"serving_default"}, const std::vector<std::string>& tag_names = {"serve"}, absl::string_view input_arg_shapes_str = "") -> py::bytes { auto module_bytecode =
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue May 21 22:58:42 UTC 2024 - 4.2K bytes - Viewed (0) -
tensorflow/compiler/mlir/quantization/common/func_test.cc
EXPECT_THAT(FindMainFuncOp(*module_op), IsNull()); } TEST_F(FindMainFuncOpTest, ReturnsServingDefaultFuncOp) { constexpr absl::string_view kModuleWithServingDefaultFunc = R"mlir( module { func.func @serving_default() -> () { return } } )mlir"; OwningOpRef<ModuleOp> module_op = ParseModuleOpString(kModuleWithServingDefaultFunc); EXPECT_THAT(*module_op, NotNull());
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue Mar 05 09:05:02 UTC 2024 - 3.3K bytes - Viewed (0) -
tensorflow/compiler/mlir/quantization/stablehlo/tests/components/pre_calibration_component.mlir
// CHECK: return %[[CUSTOM_AGGREGATOR_1]] : tensor<1x3xf32> // CHECK: } // CHECK: } // ----- // Tests that stablehlo op serialization also works for the "serving_default" // function. func.func @serving_default(%arg0: tensor<1x4xf32>) -> tensor<1x3xf32> { %0 = stablehlo.constant dense<1.0> : tensor<4x3xf32>
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Fri May 10 04:07:09 UTC 2024 - 5.1K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow_to_stablehlo/python/pywrap_tensorflow_to_stablehlo.pyi
# ============================================================================== # LINT.IfChange(savedmodel_to_stablehlo) def savedmodel_to_stablehlo( input_path: str, exported_model_signatures: list[str] = ["serving_default"], tag_names: list[str] = ["serve"], input_arg_shapes_str: str = "", ) -> bytes: ... # LINT.ThenChange() # LINT.IfChange(tensorflow_module_to_stablehlo) def tensorflow_module_to_stablehlo(
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue May 21 22:58:42 UTC 2024 - 1.1K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow_to_stablehlo/README.md
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue May 21 22:58:42 UTC 2024 - 4.4K bytes - Viewed (0) -
tensorflow/compiler/mlir/quantization/common/func.h
#include "mlir/IR/BuiltinOps.h" // from @llvm-project namespace mlir::quant { // Returns a public `func::FuncOp` in `module_op` whose name matches either // `main` or `serving_default`. If `func::FuncOps` with both names exist, the // function with name "main" takes precedence. Returns null if no such a // function exists. func::FuncOp FindMainFuncOp(ModuleOp module_op); } // namespace mlir::quant
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Mon Feb 19 06:55:11 UTC 2024 - 1.3K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow_to_stablehlo/tf_to_stablehlo_translate.cc
opt<std::string> exported_model_signatures( "exported-model-signatures", llvm::cl::desc( "Comma-separated list of exported model signatures to convert"), llvm::cl::Optional, llvm::cl::init("serving_default")); // NOLINTNEXTLINE opt<std::string> tag_names( "tags", llvm::cl::desc("Comma-separated list of tags for loading SavedModel. " "Ignored for MLIR input"),
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue May 21 22:58:42 UTC 2024 - 4.6K bytes - Viewed (0) -
tensorflow/compiler/mlir/quantization/tensorflow/quantization_options.proto
// within the SavedModel to analyze. If not specified, ["serve"] is used. repeated string tags = 5; // Sequence of keys identifying SignatureDef containing inputs and outputs. // If not specified, ["serving_default"] is used. repeated string signature_keys = 6; // A map from signature keys to the corresponding representative dataset. map<string, RepresentativeDatasetFile> representative_datasets = 7;
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue Mar 19 06:31:19 UTC 2024 - 9.2K bytes - Viewed (0)