Search Options

Results per page
Sort
Preferred Languages
Advance

Results 1 - 3 of 3 for GetBoolFromCompositeAttr (0.27 sec)

  1. tensorflow/compiler/mlir/lite/stablehlo/transforms/composite_utils.cc

                                                 &attr)) {
        return false;
      }
    
      return DenseI64AttrToI32Vector(attr, out_vec);
    }
    
    std::optional<bool> GetBoolFromCompositeAttr(
        const DictionaryAttr& composite_attrs, llvm::StringRef attr_name) {
      auto attr = composite_attrs.get(attr_name);
      if (!attr) return std::nullopt;
      if (auto bool_attr = mlir::dyn_cast_or_null<BoolAttr>(attr)) {
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Wed May 29 18:33:05 UTC 2024
    - 3.4K bytes
    - Viewed (0)
  2. tensorflow/compiler/mlir/lite/stablehlo/transforms/composite_utils.h

    bool DenseI64AttrToI32Vector(const DenseIntElementsAttr& dense_attr,
                                 std::vector<int32_t>* out_vec);
    
    // Gets boolean from composite attrs if it exists.
    std::optional<bool> GetBoolFromCompositeAttr(
        const DictionaryAttr& composite_attrs, llvm::StringRef attr_name);
    
    // Given a DictionaryAttr, checks if it has a DenseIntElementsAttr attribute
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Wed May 29 18:33:05 UTC 2024
    - 3.4K bytes
    - Viewed (0)
  3. tensorflow/compiler/mlir/lite/stablehlo/transforms/composite_avg_pool.cc

      std::vector<int32_t> strides;
      GetI32VectorFromDenseI64CompositeAttr(composite_attrs, "stride", &strides);
      data.sh = strides[0];
      data.sw = strides[1];
    
      data.ceil_mode =
          GetBoolFromCompositeAttr(composite_attrs, "ceil_mode").value();
    
      data.h_out = CalculateSpatialOutDim(data.h_in, data.kh, data.ph, data.sh,
                                          data.ceil_mode);
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Tue May 28 23:16:05 UTC 2024
    - 9.2K bytes
    - Viewed (0)
Back to top