Search Options

Results per page
Sort
Preferred Languages
Advance

Results 1 - 4 of 4 for IsQuantizedTensorType (0.17 sec)

  1. tensorflow/compiler/mlir/quantization/common/uniform_quantized_types.cc

      return false;
    }
    
    bool IsQuantizedTensorType(Type type) {
      if (!mlir::isa<TensorType>(type)) {
        return false;
      }
      Type element_type = mlir::cast<TensorType>(type).getElementType();
      return mlir::isa<QuantizedType>(element_type);
    }
    
    bool IsOpFullyQuantized(Operation* op) {
      return llvm::all_of(op->getOperandTypes(), IsQuantizedTensorType) &&
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu Apr 25 16:01:03 UTC 2024
    - 8.4K bytes
    - Viewed (0)
  2. tensorflow/compiler/mlir/quantization/common/attrs_and_constraints.cc

      }
      Type lhs_type = op->getOperand(0).getType();
      Type rhs_type = op->getOperand(1).getType();
      Type result_type = op->getResult(0).getType();
      return !IsQuantizedTensorType(lhs_type) && IsQuantizedTensorType(rhs_type) &&
             !IsQuantizedTensorType(result_type);
    }
    
    absl::StatusOr<bool> IsDotGeneralFullyConnected(DotGeneralOp dot_general_op) {
      if (dot_general_op == nullptr)
        return absl::InvalidArgumentError(
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu Apr 25 16:01:03 UTC 2024
    - 6.8K bytes
    - Viewed (0)
  3. tensorflow/compiler/mlir/lite/stablehlo/transforms/uniform_quantized_stablehlo_to_tfl_pass.cc

      LogicalResult match(stablehlo::SelectOp op) const override {
        if (!IsQuantizedTensorType(op.getOperand(1).getType())) {
          return failure();
        }
        if (!IsQuantizedTensorType(op.getOperand(2).getType())) {
          return failure();
        }
        if (!IsQuantizedTensorType(op.getResult().getType())) {
          return failure();
        }
        return success();
      }
    
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Mon Apr 22 09:00:19 UTC 2024
    - 99.8K bytes
    - Viewed (0)
  4. tensorflow/compiler/mlir/quantization/common/uniform_quantized_types.h

    // `tfl.quantize` or `tfl.dequantize` ops. ui8, i8 and i16 are supported.
    bool IsSupportedByTfliteQuantizeOrDequantizeOps(IntegerType storage_type);
    
    // Returns true if a type is quantized tensor type.
    bool IsQuantizedTensorType(Type type);
    
    // Returns true if all operands and results are quantized.
    bool IsOpFullyQuantized(Operation* op);
    
    // Returns true iff none among operand and result tensors are quantized.
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu Apr 25 16:01:03 UTC 2024
    - 5.7K bytes
    - Viewed (0)
Back to top