- Sort Score
- Result 10 results
- Languages All
Results 1 - 3 of 3 for NotTFLQuantDequantizeOp (0.35 sec)
-
tensorflow/compiler/mlir/lite/experimental/tac/common/utils.cc
#include "tensorflow/compiler/mlir/lite/ir/tfl_ops.h" #include "tensorflow/compiler/mlir/lite/utils/utils.h" namespace mlir { namespace TFL { namespace tac { bool NotTFLQuantDequantizeOp(Operation* op) { if (!op) return false; if (llvm::isa<TFL::QuantizeOp, TFL::DequantizeOp>(op)) return false; return true; } bool IsTerminatorOp(Operation* op) { if (!op) return false;
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed Mar 06 05:37:07 UTC 2024 - 2.3K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/experimental/tac/common/utils.h
} // Returns true if 'op' is a terminator op, otherwise false. bool IsTerminatorOp(Operation* op); // 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)) {
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 3.5K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/experimental/tac/transforms/target_annotation.cc
void TargetAnnotationPass::runOnFunction() { auto func = getFunction(); OpBuilder builder(func); func.walk([&](Operation* op) { // We only care about TFL dialect. if (IsNonConstOp(op) && NotTFLQuantDequantizeOp(op) && !IsTerminatorOp(op) && !llvm::isa<func::ReturnOp, func::FuncOp, CallOpInterface>(op)) { SetTargetAnnotation(op, device_specs_flag_, &builder); } }); }
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Fri May 19 19:32:06 UTC 2023 - 5.9K bytes - Viewed (0)