Search Options

Results per page
Sort
Preferred Languages
Advance

Results 1 - 3 of 3 for IsF32ShapedType (0.19 sec)

  1. tensorflow/compiler/mlir/lite/experimental/tac/common/utils.cc

      bool float_type_observed = false;
      bool int8_type_observed = false;
      bool uint8_type_observed = false;
      for (auto& input : op->getOpOperands()) {
        auto input_type = input.get().getType();
        if (IsF32ShapedType(input_type)) {
          float_type_observed = true;
        } else if (IsQI8Type(input_type)) {
          int8_type_observed = true;
        } else if (IsQUI8Type(input_type)) {
          uint8_type_observed = true;
        }
      }
    
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Wed Mar 06 05:37:07 UTC 2024
    - 2.3K bytes
    - Viewed (0)
  2. tensorflow/compiler/mlir/lite/experimental/tac/common/utils.h

    // Returns true if 'op' is not TFL Quant / Dequant op. Returns False otherwise
    // or if 'op' is null.
    bool NotTFLQuantDequantizeOp(Operation* op);
    
    // Returns true if it is a shaped type of f32 elements.
    inline bool IsF32ShapedType(Type t) {
      if (auto shaped_type = mlir::dyn_cast_or_null<ShapedType>(t)) {
        return shaped_type.getElementType().isF32();
      }
      return false;
    }
    
    // Return true when the given element_type is QI8.
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu Apr 25 16:01:03 UTC 2024
    - 3.5K bytes
    - Viewed (0)
  3. tensorflow/compiler/mlir/lite/ir/tfl_ops.cc

      if (a.size() > b.size()) return false;
    
      return std::equal(a.rbegin(), a.rend(), b.rbegin());
    }
    
    // Returns true if it is a shaped type of f32 elements.
    inline bool IsF32ShapedType(Type t) {
      if (auto shaped_type = t.dyn_cast_or_null<ShapedType>()) {
        return shaped_type.getElementType().isF32();
      }
      return false;
    }
    
    // Returns true if it is a shaped type of bf16 elements.
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu May 02 09:41:17 UTC 2024
    - 169.2K bytes
    - Viewed (0)
Back to top