Home
last modified time | relevance | path

Searched refs:output_step (Results 1 – 3 of 3) sorted by relevance

/external/tensorflow/tensorflow/lite/kernels/
Dlstm_eval.cc1757 const int output_step = n_batch * output_batch_leading_dim; in EvalFloat() local
1768 GetTensorData<float>(output) + t_rel * output_step + output_offset; in EvalFloat()
1804 const int output_step = output_batch_leading_dim; in EvalFloat() local
1818 time_offset * output_step + output_offset; in EvalFloat()
1966 const int output_step = n_batch * output_batch_leading_dim; in EvalHybrid() local
1977 GetTensorData<float>(output) + t_rel * output_step + output_offset; in EvalHybrid()
2049 const int output_step = output_batch_leading_dim; in EvalHybrid() local
2063 time_offset * output_step + output_offset; in EvalHybrid()
2199 const int output_step = n_batch * output_batch_leading_dim; in EvalInteger8x8_16() local
2202 int8_t* output_ptr = GetTensorData<int8_t>(output) + t_rel * output_step; in EvalInteger8x8_16()
[all …]
/external/tensorflow/tensorflow/lite/tools/optimize/calibration/custom_logging_ops/
Dlstm.cc404 const int output_step = n_batch * output_batch_leading_dim; in EvalFloat() local
415 GetTensorData<float>(output) + t_rel * output_step + output_offset; in EvalFloat()
452 const int output_step = output_batch_leading_dim; in EvalFloat() local
466 time_offset * output_step + output_offset; in EvalFloat()
/external/tensorflow/tensorflow/lite/tools/optimize/calibration/builtin_logging_ops/
Dlstm.cc335 const int output_step = n_batch * output_batch_leading_dim; in EvalCalibration() local
346 GetTensorData<float>(output) + t_rel * output_step + output_offset; in EvalCalibration()
383 const int output_step = output_batch_leading_dim; in EvalCalibration() local
397 time_offset * output_step + output_offset; in EvalCalibration()