Searched refs:intermediate_tensor_indexes (Results 1 – 1 of 1) sorted by relevance
| /aosp_15_r20/external/tensorflow/tensorflow/lite/tools/optimize/calibration/builtin_logging_ops/ |
| H A D | lstm.cc | 185 Logger* logger, const std::vector<int>& intermediate_tensor_indexes, in LstmStepCalibration()
214 intermediate_tensor_indexes[0], subgraph_index, error_reporter); in LstmStepCalibration()
224 is_aux_input_all_zeros, logger, intermediate_tensor_indexes[1], in LstmStepCalibration()
234 is_aux_input_all_zeros, logger, intermediate_tensor_indexes[2], in LstmStepCalibration()
248 is_aux_input_all_zeros, logger, intermediate_tensor_indexes[3], in LstmStepCalibration()
255 intermediate_tensor_indexes[4], subgraph_index, error_reporter); in LstmStepCalibration()
291 const std::vector<int>& intermediate_tensor_indexes, in EvalCalibration()
379 intermediate_tensor_indexes, subgraph_index, error_reporter); in EvalCalibration()
441 intermediate_tensor_indexes, subgraph_index, error_reporter); in EvalCalibration()
574 std::vector<int> intermediate_tensor_indexes(node->intermediates->size); in lstm_eval() local
[all …]
|