Searched refs:intermediate_tensor_index (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 | 50 Logger* logger, int intermediate_tensor_index, const int subgraph_index, in CalculateLstmGateFloat() argument 85 logger->LogTensorValue(subgraph_index, intermediate_tensor_index, gate, in CalculateLstmGateFloat() 128 int intermediate_tensor_index, const int subgraph_index, in CalculateLstmOutputCalibration() argument 135 logger->LogTensorValue(subgraph_index, intermediate_tensor_index, scratch, in CalculateLstmOutputCalibration()
|