- Sort Score
- Result 10 results
- Languages All
Results 21 - 30 of 34 for depthwise_conv_2d (0.23 sec)
-
tensorflow/compiler/mlir/quantization/tensorflow/passes/lift_quantizable_spots_as_functions.cc
// Disable quantization for the DepthwiseConv since it has no benefits in // the XLA opset. if (function_name.contains("depthwise_conv2d")) { return absl::InternalError( "DepthwiseConv2D doesn't get any benefit of quantization in XLA."); } else if (function_name.contains("conv2d")) { // For Conv2D, the channel dimension must be static to calculate the
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Fri May 10 04:07:09 UTC 2024 - 16.4K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/transforms/prepare_patterns.td
(TFL_DequantizeOp (TFL_QuantizeOp (TF_ReshapeOp $input, $shape), (UpdateShapeWithAxis<3> $qtype, $old_value))), [(UsedBy<"DepthwiseConv2D"> $old_value), (CanUpdateShapeWithAxis<3> $qtype, $old_value)], [], (addBenefit 10)>; // The axis is set to 3, because this transpose is from the legalization of
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue Apr 30 00:40:15 UTC 2024 - 10.5K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/ir/tfl_ops.td
); let results = (outs TFL_TensorOf<[F32, I32, I64]>:$output); let hasOptions = 1; } def TFL_DepthwiseConv2DOp : TFL_ConvOp<"depthwise_conv_2d", "Depthwise-separable convolution", 3, [DeclareOpInterfaceMethods<TFL_ArithmeticCount>, DynamicRangeQuantizedOpInterface]> { let arguments = (
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Jun 06 19:09:08 UTC 2024 - 186K bytes - Viewed (0) -
tensorflow/compiler/mlir/quantization/tensorflow/ops/tf_op_quant_spec.cc
StringRef function_name = mlir::cast<FlatSymbolRefAttr>(call_op.getFAttr()).getValue(); if (!function_name.starts_with("composite_")) { return spec; } if (function_name.contains("depthwise_conv2d")) { spec->coeff_op_quant_dim[1] = 3; if (function_name.contains("with_bias")) { spec->biases_params[2] = {{0, 1}, quant::GetUniformQuantizedTypeForBias}; }
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 6.3K bytes - Viewed (0) -
tensorflow/compiler/mlir/quantization/tensorflow/passes/replace_cast_hacks_with_tf_xla_ops.td
(IsConstTensor $filter), (IsInt32ElementType $conv), (HasStaticShapeConstraint $filter), (HasStaticShapeAtDimsConstraint<"3"> $input)], [], (addBenefit 10)>; // Converts inlined DepthwiseConv2D pattern to TF XlaConvV2 op. This pattern // doesn't support non-constant weights. def ConvertTFDepthwiseConv2DToXLAConvOp : Pat< (TF_CastOp:$conv (TF_DepthwiseConv2dNativeOp (TF_CastOp:$cast_input
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Sun Dec 10 05:52:02 UTC 2023 - 21.1K bytes - Viewed (0) -
tensorflow/compiler/mlir/quantization/tensorflow/passes/preprocess_op.cc
// than function name. if (!function_name.starts_with("composite_")) { return failure(); } if (function_name.contains("depthwise_conv2d")) { // Uniform Quantized op requires weights of tf.DepthwiseConv2dNative to // be transformed from [H,W,C,M] to [H,W,1,CxM] where // H=height,W=width,C=channel,M=multiplier. Therefore, a reshape op is
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 11.4K bytes - Viewed (0) -
tensorflow/compiler/mlir/quantization/tensorflow/passes/quantized_function_library.mlir
attr_map = "strides:0,use_cudnn_on_gpu:1,padding:2,explicit_paddings:3,dilations:4" } : (tensor<*xi32>, tensor<*xi32>) -> tensor<*xi32> func.return %5 : tensor<*xi32> } // DepthwiseConv2D with (simulated) int32 accumulation. func.func private @internal_depthwise_conv2d_fn( %input : tensor<*xi8>, %filter : tensor<*xi8>,
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Mon Jan 08 01:16:10 UTC 2024 - 30.6K bytes - Viewed (0) -
tensorflow/compiler/mlir/quantization/tensorflow/tests/quantize_weights.mlir
// CHECK: %[[DEPTHWISE_CONV2D:.*]] = "tf.DepthwiseConv2dNative"(%arg0, %[[DEQUANTIZED]]) <{data_format = "NHWC", dilations = [1, 1, 1, 1], explicit_paddings = [], padding = "SAME", strides = [1, 1, 2, 1]}> {attr_map = "0:strides,1:padding,2:explicit_paddings,3:dilations", device = ""} : (tensor<1x3x4x512xf32>, tensor<2x3x3x512xf32>) -> tensor<*xf32> // CHECK: return %[[DEPTHWISE_CONV2D]] : tensor<*xf32>
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Mon Oct 30 06:52:55 UTC 2023 - 42K bytes - Viewed (0) -
tensorflow/compiler/mlir/quantization/tensorflow/passes/quantized_function_library_uniform_quantized.mlir
// ...${key2}... // } // ``` // The above template with generate two functions by substituting `key1` and // `key2` with given values. module { for main_op in ["Conv2D", "DepthwiseConv2D", "MatMul"] { parameters[ {"quantized_ops": ["${main_op}", "BiasAdd"], "act_func": "internal_requantize_no_activation_fn", "output_type": "!tf_type.qint8"},
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue Aug 29 01:13:58 UTC 2023 - 19.3K bytes - Viewed (0) -
tensorflow/compiler/mlir/quantization/tensorflow/passes/quantize_composite_functions.cc
if (!function_name.starts_with("quantized_") && !function_name.starts_with("composite_")) { return failure(); } if (function_name.contains("depthwise_conv2d")) { return addReshapeOpToDepthwiseWeight(call_op, rewriter); } return failure(); } }; // Prints a summary about the quantization results. class QuantizationSummary {
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 54.5K bytes - Viewed (0)