Searched refs:lstm_op (Results 1 – 3 of 3) sorted by relevance
| /external/tensorflow/tensorflow/lite/tools/optimize/ |
| D | quantization_wrapper_utils_test.cc | 36 auto lstm_op = absl::make_unique<OperatorT>(); in TEST() local
42 lstm_op->opcode_index = 0; in TEST()
43 lstm_op->inputs = {0, 1, 2, 3, 4, 5, 6, 7, 8, -1, -1, -1, in TEST()
45 lstm_op->outputs = {24}; in TEST()
48 for (int i = 0; i < lstm_op->inputs.size(); ++i) { in TEST()
49 const int index = lstm_op->inputs[i]; in TEST()
59 model->subgraphs[0]->operators.push_back(std::move(lstm_op)); in TEST()
|
| D | quantization_wrapper_utils_custom_test.cc | 35 auto lstm_op = absl::make_unique<OperatorT>(); in TEST() local
41 lstm_op->opcode_index = 0; in TEST()
42 lstm_op->inputs = {0, 1, 2, 3, 4, 5, 6, 7, 8, -1, -1, -1, in TEST()
44 lstm_op->outputs = {24}; in TEST()
47 for (int i = 0; i < lstm_op->inputs.size(); ++i) { in TEST()
48 const int index = lstm_op->inputs[i]; in TEST()
58 model->subgraphs[0]->operators.push_back(std::move(lstm_op)); in TEST()
|
| /external/tensorflow/tensorflow/lite/toco/tflite/ |
| D | operator.cc | 854 const auto& lstm_op = in GetVersion() local
858 op_sig.options.lstm.kernel_type = GetKernelType(lstm_op.kernel_type); in GetVersion()
864 const auto& lstm_op = static_cast<const LstmCellOperator&>(op); in GetMutatingInputVariables() local
867 switch (lstm_op.kernel_type) { in GetMutatingInputVariables()
|