- Sort Score
- Result 10 results
- Languages All
Results 1 - 3 of 3 for GetOpQuantSpec (0.3 sec)
-
tensorflow/compiler/mlir/quantization/common/quantization_lib/quantization_driver_test.cc
return %0 : tensor<1x4x4x3xf32> } } )mlir"; // TOOD: b/323478683 - Directly use types rather than creating a `unique_ptr`. std::unique_ptr<quant::OpQuantSpec> GetOpQuantSpec( const mlir::Operation* op, bool disable_per_channel_for_dense_layers = false) { auto spec = std::make_unique<quant::OpQuantSpec>(); spec->coeff_op_quant_dim[1] = 3;
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/lite/transforms/default_quant_params.cc
namespace mlir { namespace TFL { // Includs an auto-generated function, which can retrieve the quantization // specification for an TFL operation. The signature of the function is // std::unique_pointer<OpQuantSpec> TFL::GetOpQuantSpec(Operation *) #include "tensorflow/compiler/mlir/lite/utils/generated_op_quant_spec_getters.inc" namespace { #define GEN_PASS_DEF_DEFAULTQUANTPARAMSPASS
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 9.4K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/quantization/tools/op_quant_spec_getters_gen.cc
// name. std::vector<Record *> defs = records.getAllDerivedDefinitions("Op"); llvm::sort(defs, LessRecord()); OUT(0) << "static std::unique_ptr<quant::OpQuantSpec> " "GetOpQuantSpec(mlir::Operation *op, bool " "disable_per_channel_for_dense_layers = false) {\n"; // TODO(b/176258587): Move to OpTrait if this should be generalized. // Add special handling for LSTM.
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Feb 15 11:18:44 UTC 2024 - 4.9K bytes - Viewed (0)