- Sort Score
- Result 10 results
- Languages All
Results 1 - 3 of 3 for kLstmCellSimple (0.16 sec)
-
tensorflow/compiler/mlir/lite/utils/lstm_utils.h
#include "mlir/Support/LogicalResult.h" // from @llvm-project #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";
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
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)