- Sort Score
- Result 10 results
- Languages All
Results 1 - 9 of 9 for quantization_driver (0.58 sec)
-
tensorflow/compiler/mlir/quantization/common/quantization_lib/quantization_driver.cc
std::vector<QuantState>& states, DenseMap<Value, QuantizationDriver::QuantStateIndex>& value_to_state, DenseMap<QuantizationDriver::OpWithOperandIndex, QuantizationDriver::QuantStateIndex>& operand_states, DenseMap<QuantizationDriver::OpWithResultIndex, QuantizationDriver::QuantStateIndex>& result_states) { const auto [cached, inserted] = value_to_state.try_emplace(value, 0);
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
// class QuantizationDriver { public: // Type alias of int used to access `states_`. using QuantStateIndex = int; // (op, operand index) pair. using OpWithOperandIndex = std::pair<Operation*, int>; // (op, result index) pair. using OpWithResultIndex = std::pair<Operation*, int>; explicit QuantizationDriver(func::FuncOp func_op, const bool is_signed,
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed Mar 20 11:42:17 UTC 2024 - 16.8K bytes - Viewed (0) -
tensorflow/compiler/mlir/quantization/common/quantization_lib/quantization_driver_test.cc
for (const auto& arg : quantization_driver.GetArgs()) { const QuantState& state = quantization_driver.GetArgQuantState(arg); EXPECT_TRUE(isa<quant::QuantizedType>(state.params)); } for (const auto& result : quantization_driver.GetResultStates()) { Operation* op = result.first.first; const int res_index = result.first.second; const QuantState state =
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 7.9K bytes - Viewed (0) -
tensorflow/compiler/mlir/quantization/common/quantization_lib/BUILD
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Mar 21 11:52:27 UTC 2024 - 3.9K bytes - Viewed (0) -
tensorflow/compiler/mlir/quantization/stablehlo/passes/prepare_quantize.cc
#include "stablehlo/dialect/StablehloOps.h" // from @stablehlo #include "tensorflow/compiler/mlir/lite/quantization/ir/QuantOps.h" #include "tensorflow/compiler/mlir/quantization/common/quantization_lib/quantization_driver.h" #include "tensorflow/compiler/mlir/quantization/common/quantization_lib/quantization_utils.h" #include "tensorflow/compiler/mlir/quantization/stablehlo/ops/stablehlo_op_quant_spec.h"
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed Apr 03 05:11:03 UTC 2024 - 8.1K bytes - Viewed (0) -
tensorflow/compiler/mlir/quantization/tensorflow/passes/prepare_quantize.cc
#include "tensorflow/compiler/mlir/lite/transforms/passes.h" #include "tensorflow/compiler/mlir/quantization/common/quantization_lib/quantization_config.h" #include "tensorflow/compiler/mlir/quantization/common/quantization_lib/quantization_driver.h" #include "tensorflow/compiler/mlir/quantization/tensorflow/ops/tf_op_quant_spec.h" #include "tensorflow/compiler/mlir/tensorflow/ir/tf_dialect.h" #include "tensorflow/compiler/mlir/tensorflow/ir/tf_ops.h"
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 17.2K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/transforms/prepare_quantize.cc
#include "tensorflow/compiler/mlir/quantization/common/quantization_lib/quantization_config.h" #include "tensorflow/compiler/mlir/quantization/common/quantization_lib/quantization_driver.h" #include "tensorflow/core/framework/types.pb.h" #include "tensorflow/core/lib/monitoring/counter.h" //===----------------------------------------------------------------------===// // The prepare-quantize Pass.
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 17.6K bytes - Viewed (0) -
tensorflow/compiler/mlir/quantization/tensorflow/passes/preprocess_op.td
include "tensorflow/compiler/mlir/tensorflow/ir/tf_ops.td" // Converts tf.Const to arith.constant for statically shaped, non-opaque constants. // Needed for QuantizationDriver to recognize constants. def ConvertTfConstToArithConst : Pat< (TF_ConstOp:$res DenseElementsAttr:$value), (Arith_ConstantOp $value),
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Sun Dec 10 05:52:02 UTC 2023 - 1.2K bytes - Viewed (0) -
tensorflow/compiler/mlir/quantization/tensorflow/passes/prepare_quantize.td
include "tensorflow/compiler/mlir/tensorflow/ir/tf_ops.td" // Converts tf.Const to arith.constant for statically shaped, non-opaque constants. // Needed for QuantizationDriver to recognize constants. def ConvertTfConstToArithConst : Pat< (TF_ConstOp:$res DenseElementsAttr:$value), (Arith_ConstantOp $value),
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Sun Dec 10 05:52:02 UTC 2023 - 1.2K bytes - Viewed (0)