- Sort Score
- Result 10 results
- Languages All
Results 1 - 2 of 2 for LSTMOp (0.49 sec)
-
tensorflow/compiler/mlir/lite/quantization/tools/op_quant_spec_getters_gen.cc
// TODO(b/176258587): Move to OpTrait if this should be generalized. // 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 "
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/utils/lstm_utils.h
// layer norm coefficients Value input_layer_norm_coefficients_; Value forget_layer_norm_coefficients_; Value cell_layer_norm_coefficients_; Value output_layer_norm_coefficients_; mlir::TFL::LSTMOp lstm_; Value none_; SmallVector<int64_t, 1> bias_slice_shape_; SmallVector<int64_t, 1> bias_size_values_; SmallVector<int64_t, 2> weight_slice_shape_;
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Sat Jun 03 00:14:05 UTC 2023 - 7.3K bytes - Viewed (0)