Search Options

Results per page
Sort
Preferred Languages
Advance

Results 1 - 5 of 5 for depthwise_conv_2d (0.23 sec)

  1. tensorflow/compiler/mlir/lite/stablehlo/tests/uniform-quantized-stablehlo-to-tfl.mlir

    // CHECK: %[[PAD:.+]] = "tfl.pad"(%[[ARG_0]], %[[CONST_0]]) : (tensor<1x4x5x3x!quant.uniform<i8:f32, 2.000000e+00>>, tensor<4x2xi32>) -> tensor<1x6x7x3x!quant.uniform<i8:f32, 2.000000e+00>>
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Tue May 14 17:10:32 UTC 2024
    - 106.2K bytes
    - Viewed (0)
  2. tensorflow/compiler/mlir/lite/stablehlo/transforms/uniform_quantized_stablehlo_to_tfl_pass.cc

          for (int k = 0; k < filter_shape[2]; ++k) {
            for (int l = 0; l < filter_shape[3]; ++l) {
              // [o, 0, 1, i] for `tfl.conv_2d` case`,
              // [i, 0, 1, o] for `tfl.depthwise_conv_2d` case.
              int old_idx = get_array_idx(filter_shape, i, j, k, l);
              int new_idx = is_depthwise
                                ? get_array_idx(new_filter_shape, k, i, j, l)
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Mon Apr 22 09:00:19 UTC 2024
    - 99.8K bytes
    - Viewed (0)
  3. tensorflow/compiler/mlir/lite/tests/ops.mlir

      %w = arith.constant dense<127.0> : tensor<32x3x3x3xf32>
      %b = arith.constant dense<0.0> : tensor<32x1xf32>
      // expected-error @+1 {{'tfl.depthwise_conv_2d' op operand #2 must be 1D tensor of any type values or none type, but got 'tensor<32x1xf32>'}}
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu Jun 06 19:09:08 UTC 2024
    - 189.2K bytes
    - Viewed (0)
  4. 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)
  5. tensorflow/compiler/mlir/lite/transforms/optimize.cc

      // Returns the dimension length of the channel dimension and also the slide
      // size by each position in the channel dimension accordingly. tfl.conv2d and
      // tfl.fully_connected has heading channel dimension, but tfl.depthwise_conv2d
      // has tailing channel dimension. This function is to provide a utility to
      // create the above information from the op property.
      static std::pair<int64_t, int64_t> GetBiasDimAndSliceSize(
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Tue Apr 30 00:40:15 UTC 2024
    - 102.3K bytes
    - Viewed (0)
Back to top