Search Options

Results per page
Sort
Preferred Languages
Advance

Results 1 - 10 of 18 for stride_w (0.57 sec)

  1. tensorflow/compiler/mlir/lite/tests/ops.mlir

      // CHECK: "VALID"
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu Jun 06 19:09:08 UTC 2024
    - 189.2K bytes
    - Viewed (0)
  2. tensorflow/compiler/mlir/lite/stablehlo/transforms/uniform_quantized_stablehlo_to_tfl_pass.cc

        // Use NHWC format.
        int32_t stride_h = 1;
        int32_t stride_w = 1;
        std::optional<ArrayRef<int64_t>> window_strides = op.getWindowStrides();
        if (window_strides.has_value()) {
          stride_h = CastI64ToI32((*window_strides)[1]).value();
          stride_w = CastI64ToI32((*window_strides)[2]).value();
        }
        auto stride_h_attr = IntegerAttr::get(rewriter.getI32Type(), stride_h);
    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/stablehlo/tests/uniform-quantized-stablehlo-to-tfl.mlir

    // Tests that the stride_w is set to 2.
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Tue May 14 17:10:32 UTC 2024
    - 106.2K bytes
    - Viewed (0)
  4. tensorflow/compiler/mlir/lite/tests/legalize-tf.mlir

      // CHECK: %[[ARG3:.*]] = "tfl.transpose_conv"(%arg0, %[[ARG0]], %arg2, %[[CST_0]]) <{fused_activation_function = "NONE", padding = "VALID", stride_h = 2 : i32, stride_w = 2 : i32}> : (tensor<4xi32>, tensor<1x3x3x32xf32>, tensor<15x14x14x32xf32>, none) -> tensor<15x28x28x1xf32>...
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Wed Jun 05 01:54:33 UTC 2024
    - 153.4K bytes
    - Viewed (0)
  5. tensorflow/compiler/mlir/lite/ir/tfl_ops.td

        I32Attr:$dilation_d_factor,
        I32Attr:$dilation_h_factor,
        I32Attr:$dilation_w_factor,
        TFL_AFAttr:$fused_activation_function,
        TFL_PaddingAttr:$padding,
        I32Attr:$stride_d,
        I32Attr:$stride_h,
        I32Attr:$stride_w
      );
    
      let results = (outs TFL_TensorOf<[F32]>:$output);
    
      let hasOptions = 1;
    
      let customOption = "Conv3DOptions";
    }
    
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu Jun 06 19:09:08 UTC 2024
    - 186K bytes
    - Viewed (0)
  6. tensorflow/compiler/mlir/lite/transforms/optimize.cc

    //                              $qtype)),
    //                          (Arith_ConstantOp F32ElementsAttr:$bias),
    //                          $h_factor, $w_factor, TFL_AF_None,
    //                          $padding, $stride_h, $stride_w),
    //                      (Arith_ConstantOp F32ElementsAttr:$value), $act_fn),
    //           (TFL_Conv2DOp $input,
    //                      (TFL_DequantizeOp (TFL_QuantizeOp
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Tue Apr 30 00:40:15 UTC 2024
    - 102.3K bytes
    - Viewed (0)
  7. tensorflow/compiler/mlir/lite/ir/tfl_ops.cc

            input.getType().cast<ShapedType>().getElementType());
        inferredReturnTypes.assign({result_type});
        return success();
      }
    
      auto stride_h = op.getStrideHAttr().getInt();
      auto stride_w = op.getStrideWAttr().getInt();
      auto dilation_h = op.getDilationHFactorAttr().getInt();
      auto dilation_w = op.getDilationWFactorAttr().getInt();
    
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu May 02 09:41:17 UTC 2024
    - 169.2K bytes
    - Viewed (0)
  8. tensorflow/compiler/mlir/tensorflow/ir/tf_ops_n_z.cc

          return op.emitOpError() << "requires begin, end and strides to have the "
                                     "same number of elements";
        }
      }
    
      // If strides are constants, verify that none of the element is zero.
      DenseIntElementsAttr strides;
      if (matchPattern(op.getStrides(), m_Constant(&strides))) {
        if (llvm::is_contained(strides.getValues<APInt>(), 0))
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu May 09 22:07:10 UTC 2024
    - 170.8K bytes
    - Viewed (0)
  9. tensorflow/compiler/mlir/lite/flatbuffer_export.cc

                                                               TFLOp op) {
      TfLitePoolParams pool_params;
      pool_params.stride_height = op.stride_h().getSExtValue();
      pool_params.stride_width = op.stride_w().getSExtValue();
      pool_params.filter_height = op.filter_h().getSExtValue();
      pool_params.filter_width = op.filter_w().getSExtValue();
      const auto padding = GetTflitePadding(inst, op.padding());
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Wed Jun 12 21:41:49 UTC 2024
    - 164.5K bytes
    - Viewed (0)
  10. tensorflow/compiler/mlir/lite/stablehlo/transforms/legalize_hlo.cc

          return failure();
        }
    
        // Constructs strides array.
        // For example, [2, 3] -> [1, 2, 3, 1].
        SmallVector<int64_t, num_spatial_dims + 2> strides({1});
        for (const auto v :
             conv_op.getWindowStrides().value().getValues<int64_t>()) {
          strides.emplace_back(v);
        }
        strides.emplace_back(1);
    
        // Constructs dilation array.
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu Apr 25 16:01:03 UTC 2024
    - 154.9K bytes
    - Viewed (0)
Back to top