- Sort Score
- Result 10 results
- Languages All
Results 31 - 40 of 49 for serving_default (0.27 sec)
-
tensorflow/compiler/mlir/quantization/tensorflow/tests/lift_quantizable_spots_as_functions_xla_selective_quantization.mlir
return %2 : tensor<1x3x2x2xf32> } func.func @serving_default(%arg0: tensor<1x3x4x3xf32>) -> (tensor<1x3x2x2xf32>) { %0 = "tf.PartitionedCall"(%arg0) {config = "", config_proto = "", executor_type = "", f = @conv2d_with_inliner} : (tensor<1x3x4x3xf32>) -> tensor<1x3x2x2xf32> return %0 : tensor<1x3x2x2xf32> } // CHECK-LABEL: func @serving_default // CHECK: "tf.PartitionedCall"
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Mon Oct 30 06:52:55 UTC 2023 - 6.8K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/tests/mlir2flatbuffer/signature_def.mlir
// CHECK-NEXT: tensor_index: 5 // CHECK-NEXT: }, { // CHECK-NEXT: name: "start_logits", // CHECK-NEXT: tensor_index: 6 // CHECK-NEXT: } ], // CHECK-NEXT: signature_key: "serving_default" // CHECK-NEXT: } ] // CHECK-NEXT:} module attributes {tf.versions = {bad_consumers = [], min_consumer = 12 : i32, producer = 554 : i32}, tf_saved_model.semantics} {
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed Dec 06 18:55:51 UTC 2023 - 4.9K 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/tensorflow/python/representative_dataset.py
tf.quantization.experimental.TfRecordRepresentativeDatasetSaver( path_map={'serving_default': '/tmp/representative_dataset_path'} ).save({'serving_default': representative_dataset}) ) # Using in QuantizationOptions. quantization_options = tf.quantization.experimental.QuantizationOptions( signature_keys=['serving_default'], representative_datasets=dataset_file_map, )
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Fri Mar 22 22:55:22 UTC 2024 - 14.2K 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/tfrt/tests/saved_model/saved_model_test.cc
TF_ASSERT_OK(MapFunctionSignaturesFromTFSavedModelMLIR( module.get(), [&](const TFRTSavedModelSignatureInfo& sig_info) { // Only check the signature of "serving_default". if (sig_info.func_name != "serving_default") return; transform(sig_info.input_names, std::back_inserter(inputs), [](llvm::StringRef x) { return x.str(); }); in_specs.assign(sig_info.input_specs.begin(),
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Fri Oct 13 01:17:29 UTC 2023 - 9K bytes - Viewed (0) -
tensorflow/compiler/mlir/quantization/tensorflow/python/integration_test/concurrency_test.py
preset_method=quant_opts_pb2.QuantizationMethod.PresetMethod.METHOD_STATIC_RANGE_INT8 ), tags={tag_constants.SERVING}, signature_keys=['serving_default'], ) model = quantize_model.quantize( temp_path, quantization_options=quantization_options, representative_dataset=data_gen(), ) return model
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Mon Sep 11 00:47:05 UTC 2023 - 3.6K bytes - Viewed (0) -
tensorflow/compiler/mlir/quantization/stablehlo/python/integration_test/quantize_model_test_base.py
self, output_saved_model_path: str ) -> str: """Extracts the first XlaCallModule op from output saved model to string.""" root = load.load(output_saved_model_path) tf_graph_def = root.signatures['serving_default'].graph.as_graph_def() for function in tf_graph_def.library.function: for node_def in function.node_def: if node_def.op == 'XlaCallModule': with ir.Context() as context:
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue May 14 06:31:57 UTC 2024 - 18.2K 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/tfrt/tests/mlrt/async_while.mlir
// CHECK-NEXT: "tf_mlrt.tf_promise"(%arg4, %5) : (!mlrt.promise, tensor<3x3xf32>) -> () // CHECK-NEXT: return //CHECK-LABEL: func.func @serving_default func.func @serving_default(%max_iterations: tensor<i32>, %array_handle: tensor<?x!tf_type.resource>, %array_flow: tensor<*xf32>, %matrix: tensor<3x3xf32>) -> (tensor<3x3xf32>, tensor<*xf32>) {
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Mon Oct 30 06:52:55 UTC 2023 - 22.2K bytes - Viewed (0)