- Sort Score
- Result 10 results
- Languages All
Results 1 - 2 of 2 for kLayerNormalizedLstmCellSimple (0.24 sec)
-
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)