Searched refs:scratch3 (Results 1 – 6 of 6) sorted by relevance
| /external/tensorflow/tensorflow/lite/kernels/ |
| D | lstm_eval.cc | 440 float* scratch2, int32_t* scratch3, int32_t* scratch4) { in CalculateLstmOutputHybrid() argument
459 scratch2, scratch3, in CalculateLstmOutputHybrid()
473 /*per_channel_scale=*/nullptr, scratch3, scratch4, in CalculateLstmOutputHybrid()
834 float* scratch1, float* scratch2, float* scratch3, float* output_ptr) { in LstmStepFloat()
844 float* output_gate_scratch = scratch3; in LstmStepFloat()
1026 float* scratch0, float* scratch1, float* scratch2, float* scratch3, in LstmStepHybrid()
1044 float* output_gate_scratch = scratch3; in LstmStepHybrid()
1378 int16_t* scratch0, int16_t* scratch1, int16_t* scratch2, int16_t* scratch3, in LstmStepInteger8x8_16()
1385 int16_t* output_gate_scratch = scratch3; in LstmStepInteger8x8_16()
1621 int8_t* scratch0, int8_t* scratch1, int16_t* scratch2, int16_t* scratch3, in LstmStepInteger8x8_8()
[all …]
|
| D | lstm_eval.h | 195 TfLiteTensor* scratch3, TfLiteTensor* scratch4, TfLiteTensor* scratch5,
221 TfLiteTensor* scratch3, TfLiteTensor* scratch4, TfLiteTensor* scratch5,
|
| D | lstm.cc | 2086 TfLiteTensor* scratch3; in Eval() local
2088 GetTemporarySafe(context, node, 3, &scratch3)); in Eval()
2107 cell_state, output, scratch0, scratch1, scratch2, scratch3, in Eval()
2119 TfLiteTensor* scratch3; in Eval() local
2121 GetTemporarySafe(context, node, 3, &scratch3)); in Eval()
2146 scratch3, scratch4, scratch5, scratch6, scratch7); in Eval()
|
| D | unidirectional_sequence_lstm.cc | 1407 TfLiteTensor* scratch3; in Eval() local
1409 GetTemporarySafe(context, node, 3, &scratch3)); in Eval()
1428 output, scratch0, scratch1, scratch2, scratch3, scratch4, scratch5, in Eval()
|
| /external/tensorflow/tensorflow/lite/tools/optimize/calibration/builtin_logging_ops/ |
| D | lstm.cc | 183 float* scratch1, float* scratch2, float* scratch3, float* output_ptr, in LstmStepCalibration()
195 float* output_gate_scratch = scratch3; in LstmStepCalibration()
|
| /external/tensorflow/tensorflow/core/kernels/ |
| D | fused_batch_norm_op.cc | 559 typename TTypes<U, 2>::Tensor scratch3(scratch3_tensor.tensor<U, 2>()); in operator ()() local
595 scratch3.device(d) = in operator ()()
|