Home
last modified time | relevance | path

Searched refs:kLstmCellSimple (Results 1 – 3 of 3) sorted by relevance

/external/tensorflow/tensorflow/compiler/mlir/lite/utils/
Dlstm_utils.h35 constexpr char kLstmCellSimple[] = "LSTMCellSimple"; variable
62 virtual llvm::StringRef GetCompositeOpName() { return kLstmCellSimple; } in GetCompositeOpName()
Dlstm_utils_test.cc74 attributes.push_back(kLstmCellSimple); in createLstmCompositeFunc()
200 llvm::SmallVector<std::string, 2> attributes{kLstmCellSimple, in TEST_F()
/external/tensorflow/tensorflow/compiler/mlir/lite/transforms/
Dprepare_composite_functions_tf.cc253 } else if (attr.getValue() == mlir::TFL::kLstmCellSimple) { in ConvertTFImplements()