Searched refs:LogTensorValue (Results 1 – 4 of 4) sorted by relevance
/external/tensorflow/tensorflow/lite/tools/optimize/calibration/custom_logging_ops/ |
D | lstm.cc | 118 logger->LogTensorValue(subgraph_index, intemediate_tensor_indexes[1], in LstmStepWithAuxInput() 122 logger->LogTensorValue(subgraph_index, intemediate_tensor_indexes[4], in LstmStepWithAuxInput() 125 logger->LogTensorValue(subgraph_index, intemediate_tensor_indexes[7], in LstmStepWithAuxInput() 127 logger->LogTensorValue(subgraph_index, intemediate_tensor_indexes[10], in LstmStepWithAuxInput() 173 logger->LogTensorValue(subgraph_index, intemediate_tensor_indexes[2], in LstmStepWithAuxInput() 181 logger->LogTensorValue(subgraph_index, intemediate_tensor_indexes[5], in LstmStepWithAuxInput() 190 logger->LogTensorValue(subgraph_index, intemediate_tensor_indexes[8], in LstmStepWithAuxInput() 197 logger->LogTensorValue(subgraph_index, intemediate_tensor_indexes[11], in LstmStepWithAuxInput() 210 logger->LogTensorValue(subgraph_index, intemediate_tensor_indexes[0], in LstmStepWithAuxInput() 232 logger->LogTensorValue(subgraph_index, intemediate_tensor_indexes[3], in LstmStepWithAuxInput() [all …]
|
/external/tensorflow/tensorflow/lite/tools/optimize/calibration/ |
D | calibration_logger.h | 52 TfLiteStatus LogTensorValue(int subgraph_index, int tensor_index, in LogTensorValue() function
|
D | calibrator.cc | 234 logger->LogTensorValue(op_info.subgraph_index, i, tensor.data.f, in LoggingEval() 258 logger->LogTensorValue(op_info.subgraph_index, i, tensor.data.f, in LoggingEval() 265 logger->LogTensorValue(op_info.subgraph_index, i, tensor.data.f, in LoggingEval()
|
/external/tensorflow/tensorflow/lite/tools/optimize/calibration/builtin_logging_ops/ |
D | lstm.cc | 85 logger->LogTensorValue(subgraph_index, intermediate_tensor_index, gate, in CalculateLstmGateFloat() 135 logger->LogTensorValue(subgraph_index, intermediate_tensor_index, scratch, in CalculateLstmOutputCalibration()
|