- Sort Score
- Result 10 results
- Languages All
Results 1 - 2 of 2 for SetupAllStates (0.09 sec)
-
tensorflow/compiler/mlir/quantization/common/quantization_lib/quantization_driver.cc
builder_.create<arith::ConstantOp>(cst.getLoc(), cst.getValue()); user->setOperand(operand_num, new_constant_op); } } } }); } void QuantizationDriver::SetupAllStates() { for (BlockArgument arg : fn_.getArguments()) { args_.push_back(arg); Value value = arg; // If the argument is quantized, it should only has one user. if (arg.hasOneUse()) {
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 38.1K bytes - Viewed (0) -
tensorflow/compiler/mlir/quantization/common/quantization_lib/quantization_driver.h
// later. // - Adds all per-channel weights to a set for looking up later. void PreprocessConstantOps(); // Sets up all the data structures for quantization propagation. void SetupAllStates(); // Returns Whether the constant is a weight, which shouldn't be shared by // different ops. bool IsWeight(Operation* cst) { return llvm::is_contained(weights_, cst); }
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed Mar 20 11:42:17 UTC 2024 - 16.8K bytes - Viewed (0)