- Sort Score
- Result 10 results
- Languages All
Results 1 - 2 of 2 for kLstmCellSimple (0.15 sec)
-
tensorflow/compiler/mlir/lite/utils/lstm_utils_test.cc
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 = builder->getStringAttr(llvm::join(attributes, ","));
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
if (failed(convert_embedded_lookup.VerifySignature())) return; func.eraseBody(); func.addEntryBlock(); convert_embedded_lookup.RewriteFunc(); } else if (attr.getValue() == mlir::TFL::kLstmCellSimple) { // Check if the lstm cell simple can be fused, if not, we just don't do // anything. if (failed(CheckFusableLstmCellSimple(func))) return; func.eraseBody(); func.addEntryBlock();
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 17.6K bytes - Viewed (0)