Searched refs:output_batch_leading_dim (Results 1 – 6 of 6) sorted by relevance
| /external/tensorflow/tensorflow/lite/kernels/internal/ |
| D | kernel_utils.cc | 27 int output_batch_leading_dim, in RnnBatchStep() argument
34 batch_size, output_batch_leading_dim, activation, in RnnBatchStep()
43 int batch_size, int output_batch_leading_dim, in RnnBatchStep() argument
48 if (output_batch_leading_dim == num_units) { in RnnBatchStep()
79 output_ptr_batch + k * output_batch_leading_dim); in RnnBatchStep()
87 output_ptr_batch + k * output_batch_leading_dim); in RnnBatchStep()
96 /*n_batch=*/1, output_ptr_batch + k * output_batch_leading_dim); in RnnBatchStep()
105 /*n_batch=*/1, output_ptr_batch + k * output_batch_leading_dim); in RnnBatchStep()
111 output_ptr_batch + k * output_batch_leading_dim, num_units, in RnnBatchStep()
112 activation, output_ptr_batch + k * output_batch_leading_dim); in RnnBatchStep()
[all …]
|
| D | kernel_utils.h | 42 int output_batch_leading_dim,
52 int batch_size, int output_batch_leading_dim,
70 int num_units, int batch_size, int output_batch_leading_dim,
83 int batch_size, int output_batch_leading_dim,
|
| /external/tensorflow/tensorflow/lite/tools/optimize/calibration/custom_logging_ops/ |
| D | lstm.cc | 64 int n_aux_input, int n_output, int output_batch_leading_dim, in LstmStepWithAuxInput()
302 output_ptr + k * output_batch_leading_dim); in LstmStepWithAuxInput()
306 std::fill_n(output_ptr + k * output_batch_leading_dim, n_output, 0.0f); in LstmStepWithAuxInput()
313 /*n_batch=*/1, output_ptr + k * output_batch_leading_dim); in LstmStepWithAuxInput()
315 tensor_utils::CwiseClipping(output_ptr + k * output_batch_leading_dim, in LstmStepWithAuxInput()
322 output_ptr + k * output_batch_leading_dim); in LstmStepWithAuxInput()
326 std::copy_n(output_ptr + k * output_batch_leading_dim, n_output, in LstmStepWithAuxInput()
399 const int output_batch_leading_dim = in EvalFloat() local
404 const int output_step = n_batch * output_batch_leading_dim; in EvalFloat()
443 n_input, aux_input_size, n_output, output_batch_leading_dim, in EvalFloat()
[all …]
|
| /external/tensorflow/tensorflow/lite/kernels/ |
| D | lstm_eval.cc | 832 int n_aux_input, int n_output, int output_batch_leading_dim, in LstmStepFloat()
903 output_ptr + b * output_batch_leading_dim); in LstmStepFloat()
1025 int n_aux_input, int n_output, int output_batch_leading_dim, in LstmStepHybrid()
1220 output_ptr + b * output_batch_leading_dim); in LstmStepHybrid()
1619 int n_output, int output_batch_leading_dim, int8_t* output_state_ptr, in LstmStepInteger8x8_8()
1752 const int output_batch_leading_dim = in EvalFloat() local
1757 const int output_step = n_batch * output_batch_leading_dim; in EvalFloat()
1796 n_input, aux_input_size, n_output, output_batch_leading_dim, in EvalFloat()
1804 const int output_step = output_batch_leading_dim; in EvalFloat()
1822 GetTensorData<float>(output_state) + b * output_batch_leading_dim; in EvalFloat()
[all …]
|
| D | basic_rnn.cc | 208 const int output_batch_leading_dim = in EvalFloat() local
223 input_size, num_units, batch_size, output_batch_leading_dim, in EvalFloat()
241 const int output_batch_leading_dim = in EvalHybrid() local
272 num_units, batch_size, output_batch_leading_dim, params->activation, in EvalHybrid()
|
| /external/tensorflow/tensorflow/lite/tools/optimize/calibration/builtin_logging_ops/ |
| D | lstm.cc | 181 int n_aux_input, int n_output, int output_batch_leading_dim, in LstmStepCalibration()
259 output_ptr + b * output_batch_leading_dim); in LstmStepCalibration()
330 const int output_batch_leading_dim = in EvalCalibration() local
335 const int output_step = n_batch * output_batch_leading_dim; in EvalCalibration()
374 n_input, aux_input_size, n_output, output_batch_leading_dim, in EvalCalibration()
383 const int output_step = output_batch_leading_dim; in EvalCalibration()
401 GetTensorData<float>(output_state) + b * output_batch_leading_dim; in EvalCalibration()
436 n_cell, n_input, aux_input_size, n_output, output_batch_leading_dim, in EvalCalibration()
|