- Sort Score
- Result 10 results
- Languages All
Results 1 - 2 of 2 for GetLstmOpQuantSpec (0.15 sec)
-
tensorflow/compiler/mlir/lite/quantization/tools/op_quant_spec_getters_gen.cc
// Add special handling for LSTM. OUT(2) << "if (auto lstm_op = llvm::dyn_cast<TFL::LSTMOp>(op)) {\n"; OUT(4) << "return GetLstmOpQuantSpec<TFL::LSTMOp>(lstm_op);\n"; OUT(2) << "} else if (auto lstm_op = " "llvm::dyn_cast<TFL::UnidirectionalSequenceLSTMOp>(op)) {\n"; OUT(4) << "return " "GetLstmOpQuantSpec<TFL::UnidirectionalSequenceLSTMOp>(lstm_op);\n"; OUT(2) << "}\n";
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Feb 15 11:18:44 UTC 2024 - 4.9K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/transforms/prepare_quantize_helper.h
} return {}; }; } // Returns quantization spec for LSTMs based on their operator properties. template <typename LstmOp> std::unique_ptr<quant::OpQuantSpec> GetLstmOpQuantSpec(LstmOp op) { operator_property::OpVariant lstm_variant; operator_property::OperatorProperty lstm_property; if (failed(GetLstmProperty(op, &lstm_variant, &lstm_property))) { return nullptr; }
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Fri May 03 18:01:23 UTC 2024 - 28K bytes - Viewed (0)