- Sort Score
- Result 10 results
- Languages All
Results 1 - 3 of 3 for kLayerNormalizedLstmCellSimple (0.36 sec)
-
tensorflow/compiler/mlir/lite/utils/lstm_utils.h
#include "tensorflow/compiler/mlir/lite/ir/tfl_ops.h" namespace mlir { namespace TFL { constexpr char kTFImplements[] = "tf._implements"; constexpr char kLstmCellSimple[] = "LSTMCellSimple"; constexpr char kLayerNormalizedLstmCellSimple[] = "LayerNormalizedLstmCellSimple"; constexpr char kCoupleInputForgetGates[] = "CoupleInputForgetGates"; // A utility class that enables the conversion of the LSTMCellSimple composite
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Sat Jun 03 00:14:05 UTC 2023 - 7.3K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/utils/lstm_utils_test.cc
mlir::NameLoc::get(builder->getStringAttr("fused_func")), "fused_func", func_type, {}); func.addEntryBlock(); std::vector<std::string> attributes; if (ln) { attributes.push_back(kLayerNormalizedLstmCellSimple); } else { attributes.push_back(kLstmCellSimple); } if (cifg) { attributes.push_back(kCoupleInputForgetGates); } mlir::StringAttr attr_values =
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 10K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/transforms/prepare_composite_functions_tf.cc
ConvertLSTMCellSimpleToFusedLSTM convert_lstm_cell_simple(func); if (failed(convert_lstm_cell_simple.RewriteFunc())) { return signalPassFailure(); } } else if (attr.getValue() == mlir::TFL::kLayerNormalizedLstmCellSimple) { // Check if the layer normalized lstm cell simple can be fused, if not, we // just don't do anything. if (failed(CheckFusableLayerNormalizedLstmCellSimple(func))) return; func.eraseBody();
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 17.6K bytes - Viewed (0)