- Sort Score
- Result 10 results
- Languages All
Results 11 - 19 of 19 for tf_quant__ (0.14 sec)
-
tensorflow/compiler/mlir/quantization/common/quantization_lib/quantization_driver_test.cc
return %1 : tensor<1x4x4x3xf32> } func.func private @composite_fn_1(%arg0: tensor<1x4x4x3xf32>, %arg1: tensor<3x1x1x3xf32>, %arg2: tensor<3xf32>) -> tensor<1x4x4x3xf32> attributes {tf_quant.composite_function} {
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 7.9K bytes - Viewed (0) -
tensorflow/compiler/mlir/quantization/tensorflow/quantize_preprocess.h
namespace tensorflow { namespace quantization { // Default MLIR dump file prefix for TensorFlow quantization passes. inline constexpr absl::string_view kDefaultTfQuantMlirDumpFilePrefix = "tf_quant"; // Preprocesses the `module_op` for quantization. The preprocess steps include // freezing the variables in the graph into constants. `is_inliner_run` // determines whether the `InlinerPass` should be run after unfreezing.
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed Apr 24 12:49:45 UTC 2024 - 3.2K bytes - Viewed (0) -
tensorflow/compiler/mlir/quantization/common/lift_as_function_call.h
namespace mlir::quant { // This attribute will be set for functions created by this pass. // Presence of this attribute will mark the function as quantization target. inline constexpr StringRef kFusedFunctionAttr = "tf_quant.composite_function"; // The keyword to detect if this is a `NullAttribute`. inline constexpr StringRef kNullAttributeValue = "N/A"; // Prefixes attached to lifted functions.
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Fri May 10 04:07:09 UTC 2024 - 5.5K bytes - Viewed (0) -
tensorflow/compiler/mlir/quantization/common/lift_as_function_call.td
class NamedAttr<string attr_name> : NativeCodeCall<"NamedAttribute{$_builder.getStringAttr(\"" # attr_name # "\"), $0}">; // Checks if the value is not defined inside a lifted function by checking the // `tf_quant.composite_function` attribute. def IsNotInLiftedFunc : Constraint<CPred<"!IsInLiftedFunc($0.getDefiningOp())">>; // Checks if the value is not inside a StableHLO op with region. def IsNotInStableHloOpRegion :
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Mon Mar 25 00:32:20 UTC 2024 - 3.4K bytes - Viewed (0) -
tensorflow/compiler/mlir/quantization/stablehlo/tests/passes/insert_calibration_statistics_saver_with_skipping.mlir
// CHECK: return func.func private @composite_dot_general_with_relu_fn_1(%arg0: tensor<10x1x1024xf32>, %arg1: tensor<10x1024x3xf32>) -> tensor<10x1x3xf32> attributes {_from_xla_call_module, tf_quant.composite_function} { %cst = stablehlo.constant dense<0.000000e+00> : tensor<10x1x3xf32>
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue May 14 06:31:57 UTC 2024 - 6.3K bytes - Viewed (0) -
tensorflow/compiler/mlir/quantization/stablehlo/tests/passes/prepare_quantize/prepare_quantize_per_channel.mlir
// CHECK-LABEL: composite_conv2d_with_bias_and_relu6_fn_10 func.func private @composite_conv2d_with_bias_and_relu6_fn_10(%arg0: tensor<1x3x2x3xf32>, %arg1: tensor<2x3x3x2xf32>, %arg2: tensor<2xf32>) -> tensor<1x2x2x2xf32> attributes {tf.tf_quant.composite_function} { %0 = "quantfork.stats"(%arg1) {layerStats = dense<[-3.54062747, 0.54742622]> : tensor<2xf32>} : (tensor<2x3x3x2xf32>) -> tensor<2x3x3x2xf32>
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue Mar 26 07:48:15 UTC 2024 - 8.6K bytes - Viewed (0) -
tensorflow/compiler/mlir/quantization/stablehlo/passes/replace_stablehlo_ops_in_main_function_with_xla_call_module_ops.cc
stablehlo_func_id); } // TODO - b/298966126: Currently quantizable functions are identified in TF // Quantizer via the tf_quant.composite_function UnitAttr attached to // func ops. We remove this attribute as this interferes with VHLO conversion. // Remove this temporary hack. for (auto func_op : module_op.getOps<func::FuncOp>()) {
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 21K bytes - Viewed (0) -
tensorflow/compiler/mlir/quantization/tensorflow/tests/lift_quantizable_spots_as_functions.mlir
// CHECK: } // CHECK-LABEL: func private @composite_conv2d_with_bias_and_relu6_fn_1(%arg0: tensor<1x3x4x3xf32>, %arg1: tensor<2x3x3x2xf32>, %arg2: tensor<2xf32>) -> tensor<*xf32> attributes {tf_quant.composite_function} { // CHECK-NEXT: %[[CONV2D_0:.*]] = "tf.Conv2D"(%arg0, %arg1) // CHECK-SAME: data_format = "NHWC", dilations = [1, 1, 2, 1], explicit_paddings = [], padding = "SAME", strides = [1, 1, 2, 1], use_cudnn_on_gpu = true
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Fri May 10 04:07:09 UTC 2024 - 26.5K bytes - Viewed (0) -
tensorflow/compiler/mlir/quantization/tensorflow/passes/quantize_composite_functions.cc
constexpr StringRef kQuantizeFuncName = "quantize_i8"; constexpr StringRef kDequantizeFuncName = "dequantize_i8"; constexpr StringRef kAttrMapAttribute = "attr_map"; constexpr StringRef kQuantizedOpsAttribute = "tf_quant.quantized_ops"; constexpr StringRef kCompositeFuncPrefix = "composite_"; constexpr StringRef kQuantizedFuncPrefix = "quantized_"; constexpr StringRef kFloatOutputFuncSuffix = "_float_output_fn";
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 54.5K bytes - Viewed (0)