Search Options

Results per page
Sort
Preferred Languages
Advance

Results 1 - 8 of 8 for IntegerAttr (0.42 sec)

  1. tensorflow/compiler/mlir/tensorflow/transforms/shape_inference.cc

              window_dimensions.getValues<IntegerAttr>()[i].getInt());
        }
    
        for (auto i = 0; i < window_strides.size(); ++i) {
          window_strides_vec.push_back(
              window_strides.getValues<IntegerAttr>()[i].getInt());
        }
    
        for (auto i = 0; i < base_dilations.size(); ++i) {
          base_dilations_vec.push_back(
              base_dilations.getValues<IntegerAttr>()[i].getInt());
        }
    
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Sat Jun 08 07:28:49 UTC 2024
    - 134.1K bytes
    - Viewed (0)
  2. tensorflow/compiler/mlir/tensorflow/ir/tf_ops_n_z.cc

      if (elem_type.isSignlessInteger() || elem_type.isUnsignedInteger()) {
        auto start_attr = start_tensor.getValues<IntegerAttr>()[0];
        auto limit_attr = limit_tensor.getValues<IntegerAttr>()[0];
        auto delta_attr = delta_tensor.getValues<IntegerAttr>()[0];
        int num_elements;
        if (elem_type.isUnsignedInteger()) {
          uint64_t start = start_attr.getUInt();
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu May 09 22:07:10 UTC 2024
    - 170.8K bytes
    - Viewed (0)
  3. tensorflow/compiler/mlir/lite/ir/tfl_ops.cc

        Type elem_type = getType().cast<ShapedType>().getElementType();
        if (elem_type.isSignlessInteger()) {
          auto start_attr = start_tensor.getValues<IntegerAttr>()[0];
          auto limit_attr = limit_tensor.getValues<IntegerAttr>()[0];
          auto delta_attr = delta_tensor.getValues<IntegerAttr>()[0];
          const int num_elements = GetLengthOfRange(
              start_attr.getInt(), limit_attr.getInt(), delta_attr.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)
  4. tensorflow/compiler/mlir/tensorflow/ir/tf_ops_a_m.cc

      DenseIntElementsAttr axis_attr;
      if (!matchPattern(op.getAxis(), m_Constant(&axis_attr))) return failure();
      if (axis_attr.getNumElements() != 1) return failure();
      int64_t axis =
          axis_attr.getSplatValue<IntegerAttr>().getValue().getSExtValue();
      // TODO(ezhulenev): Compute axis from rank. e.g. It might be common to concat
      // on the channels dim for NCHW layout as axis=-2.
      if (axis < 0) return failure();
    
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu Apr 25 16:01:03 UTC 2024
    - 146.7K bytes
    - Viewed (0)
  5. tensorflow/compiler/mlir/lite/stablehlo/transforms/uniform_quantized_stablehlo_to_tfl_pass.cc

          stride_w = CastI64ToI32((*window_strides)[2]).value();
        }
        auto stride_h_attr = IntegerAttr::get(rewriter.getI32Type(), stride_h);
        auto stride_w_attr = IntegerAttr::get(rewriter.getI32Type(), stride_w);
    
        ArrayRef<int64_t> window_dims = op.getWindowDimensions();
        auto window_w_attr = IntegerAttr::get(rewriter.getI32Type(),
                                              CastI64ToI32(window_dims[2]).value());
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Mon Apr 22 09:00:19 UTC 2024
    - 99.8K bytes
    - Viewed (0)
  6. tensorflow/compiler/mlir/lite/stablehlo/transforms/legalize_hlo.cc

          float_spl && sgn_cst_spl) {
        return IsSign(float_spl.getValue(), sgn_cst_spl.getValue());
      }
      if (auto int_spl = llvm::dyn_cast_if_present<IntegerAttr>(splat),
          sgn_cst_spl = llvm::dyn_cast_if_present<IntegerAttr>(sgn_splat);
          int_spl && sgn_cst_spl) {
        return IsSign(int_spl.getValue(), sgn_cst_spl.getValue());
      }
      if (mlir::isa<DenseFPElementsAttr>(float_or_int)) {
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu Apr 25 16:01:03 UTC 2024
    - 154.9K bytes
    - Viewed (0)
  7. tensorflow/compiler/mlir/lite/ir/tfl_ops.td

        "whose size is at most " # n>;
    
    // Ensures the given integer attribute has the given value.
    class TFL_IntEqualsTo<int n> : AttrConstraint<
        CPred<"$_self.isa<IntegerAttr>() && "
                "$_self.cast<IntegerAttr>().getInt() == " # n>,
        "whose value is " # n>;
    
    // Ensures the given LSTMKernelType attribute has the given value.
    class TFL_LSTMKernelTypeEqualsTo<string value> : AttrConstraint<
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu Jun 06 19:09:08 UTC 2024
    - 186K bytes
    - Viewed (0)
  8. tensorflow/compiler/mlir/tensorflow/translate/import_model.cc

             << " at index path: <value>";
          for (auto path_element : current_index_path_) {
            os << ".";
            if (auto integer = mlir::dyn_cast<mlir::IntegerAttr>(path_element)) {
              os << integer.getValue();
            } else {
              auto str = mlir::cast<mlir::StringAttr>(path_element);
              os << str.getValue();
            }
          }
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Wed May 01 11:17:36 UTC 2024
    - 183.2K bytes
    - Viewed (0)
Back to top