- Sort Score
- Result 10 results
- Languages All
Results 1 - 10 of 14 for conv_2d (0.19 sec)
-
tensorflow/compiler/mlir/lite/stablehlo/tests/uniform-quantized-stablehlo-to-tfl.mlir
} // Confirm that the `stablehlo.convolution` is not converted to `tfl.conv_2d`. // CHECK-LABEL: convolution_upstream_srq_non_const_filter // CHECK-SAME: %[[ARG:.+]]: tensor<1x3x3x4x!quant.uniform<i8:f32, 1.000000e+00:-100>> // CHECK: stablehlo.convolution // CHECK-NOT: tfl.conv_2d // ----- // Tests that if the window padding contains values of 0, tfl.pad op is not
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue May 14 17:10:32 UTC 2024 - 106.2K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/tests/ops.mlir
func.func @testConv2D(tensor<256x32x32x3xf32>, tensor<16x3x3x3xf32>, tensor<16xf32>) -> tensor<256x32x32x16xf32> { ^bb0(%arg0: tensor<256x32x32x3xf32>, %arg1: tensor<16x3x3x3xf32>, %arg2: tensor<16xf32>): // CHECK: "tfl.conv_2d"(%arg0, %arg1, %arg2)
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Jun 06 19:09:08 UTC 2024 - 189.2K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/stablehlo/transforms/uniform_quantized_stablehlo_to_tfl_pass.cc
FindUserOfType<TFL::QuantizeOp>(op) != nullptr); } }; // Rewrites `stablehlo.convolution` into fused `tfl.conv_2d`. // If available, fuse bias and activation adjacent to `stablehlo.convolution`. // This RewritePattern rewrites both the following into `tfl.conv_2d` op: // // StableHLO Quantizer output: // * input: per-tensor qi8 // * filter: per-channel qi8 (`quantization_dimension` = 3)
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Mon Apr 22 09:00:19 UTC 2024 - 99.8K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/transforms/optimize.cc
// with a 0-d constant, e.g. before this optimization, // %cst = arith.constant dense<1.0> : tensor<16x16x4xf32> // %0 = "tfl.conv_2d"... // %1 = "tfl.add"(%0, %cst) : (tensor<16x16x4xf32>, tensor<16x16x4xf32>) // After this optimization: // %cst = arith.constant dense<1.0> : tensor<f32> // %0 = "tfl.conv_2d"... // %1 = "tfl.add"(%0, %cst) : (tensor<16x16x4xf32>, tensor<f32>)
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue Apr 30 00:40:15 UTC 2024 - 102.3K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/stablehlo/transforms/legalize_hlo.cc
auto conv_result = rewriter.create<mhlo::ConvolutionOp>( conv_op.getLoc(), new_output_type, sliced_input, sliced_kernel, conv_op.getWindowStridesAttr(), conv_op.getPaddingAttr(), conv_op.getLhsDilationAttr(), conv_op.getRhsDilationAttr(), conv_op.getWindowReversalAttr(), conv_op.getDimensionNumbers(), 1, 1, conv_op.getPrecisionConfigAttr()); conv_results.push_back(conv_result);
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 154.9K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/ir/tfl_ops.td
in the flatbuffer. }]; let arguments = (ins I32Attr:$buffer_index); let results = (outs AnyTensor:$output); } def TFL_Conv2DOp : TFL_ConvOp<"conv_2d", "Convolution", 0, [DeclareOpInterfaceMethods<InferTypeOpInterface>, 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/python/integration_test/quantize_model_test.py
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Fri May 17 03:36:50 UTC 2024 - 235.6K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/tests/shape_inference.mlir
func.return %arg0 : tensor<*xi32> } // Test conv2d inferReturnTypes can infer some information when input or // filter does not have fully static shape. // CHECK-LABEL: func @conv2d_unranked_input_and_filter func.func @conv2d_unranked_input_and_filter(%arg0: tensor<*xf32>, %arg1: tensor<*xf32>) -> tensor<*xf32> { // CHECK: "tf.Conv2D" // CHECK-SAME: -> tensor<?x?x?x?xf32>
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue Jan 23 17:24:10 UTC 2024 - 167.4K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/tests/tf-ops.mlir
%0 = "tf.Conv2D"(%arg0, %arg1) {padding = "SAME", strides = [1, 1]} : (tensor<256x32x32x3xf32>, tensor<3x3x3x16xf32>) -> tensor<256x30x30x16xf32> func.return %0 : tensor<256x30x30x16xf32> } // ----- func.func @testConv2D(%arg0: tensor<256x32x32x3xf32>, %arg1: tensor<3x3x3x16xf32>) -> tensor<256x30x30x16xf32> { // expected-error @+1 {{requires positive strides}}
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Mon Oct 23 14:40:35 UTC 2023 - 236.4K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/tests/legalize-tf.mlir
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed Jun 05 01:54:33 UTC 2024 - 153.4K bytes - Viewed (0)