Searched refs:kCellToForgetWeightsTensor (Results 1 – 8 of 8) sorted by relevance
| /external/tensorflow/tensorflow/lite/kernels/ |
| D | lstm_shared.h | 44 constexpr int kCellToForgetWeightsTensor = 10; // Optional variable
|
| D | unidirectional_sequence_lstm.cc | 129 context, node, lstm::full::kCellToForgetWeightsTensor); in PopulateQuantizedLstmParams8x8_16()
497 context, node, lstm::full::kCellToForgetWeightsTensor); in CheckInputTensorDimensions()
1254 context, node, lstm::full::kCellToForgetWeightsTensor); in Eval()
|
| D | lstm.cc | 214 GetOptionalInputTensor(context, node, kCellToForgetWeightsTensor); in PopulateQuantizedLstmParams8x8_16()
502 GetOptionalInputTensor(context, node, kCellToForgetWeightsTensor); in PopulateQuantizedLstmParams8x8_8()
951 GetOptionalInputTensor(context, node, kCellToForgetWeightsTensor); in CheckInputTensorDimensions()
1865 GetOptionalInputTensor(context, node, kCellToForgetWeightsTensor); in Eval()
|
| /external/tensorflow/tensorflow/lite/toco/graph_transformations/ |
| D | lstm_utils.h | 42 kCellToForgetWeightsTensor = 10, // Optional enumerator
|
| D | identify_lstm_split_inputs.cc | 128 &(lstm_cell_op->inputs[kCellToForgetWeightsTensor]), in Run()
|
| /external/tensorflow/tensorflow/lite/delegates/gpu/common/ |
| D | lstm_parser.cc | 83 node, tflite::ops::builtin::lstm::full::kCellToForgetWeightsTensor); in HasPeephole()
470 tflite::ops::builtin::lstm::full::kCellToForgetWeightsTensor, in ParseLSTMAttributes()
|
| /external/tensorflow/tensorflow/lite/tools/optimize/calibration/builtin_logging_ops/ |
| D | lstm.cc | 516 context, node, ops::builtin::lstm::full::kCellToForgetWeightsTensor); in lstm_eval()
|
| /external/tensorflow/tensorflow/lite/tools/optimize/calibration/custom_logging_ops/ |
| D | lstm.cc | 560 context, node, ops::builtin::lstm::full::kCellToForgetWeightsTensor); in lstm_eval()
|