- Sort Score
- Result 10 results
- Languages All
Results 1 - 4 of 4 for IsQuantizedTensorType (0.17 sec)
-
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) -
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) -
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) -
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)