Lines Matching full:n_batch
246 const uint32_t n_batch = SizeOfDimension(input, 0);
271 outputShape->dimensions = { n_batch, n_output };
276 outputStateShape->dimensions = { n_batch, n_output };
281 cellStateShape->dimensions = { n_batch, n_cell };
290 scratchShape->dimensions = { n_batch, n_cell * 3 };
293 scratchShape->dimensions = { n_batch, n_cell * 4 };
303 const uint32_t n_batch = input_->shape().dimensions[0];
321 forget_gate_scratch = cell_scratch + n_cell * n_batch;
322 output_gate_scratch = cell_scratch + 2 * n_cell * n_batch;
325 cell_scratch = input_gate_scratch + n_cell * n_batch;
326 forget_gate_scratch = input_gate_scratch + 2 * n_cell * n_batch;
327 output_gate_scratch = input_gate_scratch + 3 * n_cell * n_batch;
333 n_cell, n_batch, input_gate_scratch);
336 n_cell, n_batch, forget_gate_scratch);
338 n_cell, n_batch, cell_scratch);
340 n_cell, n_batch, output_gate_scratch);
346 GetBuffer<float>(input_), n_batch, input_gate_scratch, /*result_stride*/1);
350 GetBuffer<float>(input_), n_batch, forget_gate_scratch, /*result_stride*/1);
353 GetBuffer<float>(input_), n_batch, cell_scratch, /*result_stride*/1);
356 GetBuffer<float>(input_), n_batch, output_gate_scratch, /*result_stride*/1);
362 GetBuffer<float>(output_state_in_), n_batch, input_gate_scratch, /*result_stride*/1);
366 GetBuffer<float>(output_state_in_), n_batch, forget_gate_scratch, /*result_stride*/1);
369 GetBuffer<float>(output_state_in_), n_batch, cell_scratch, /*result_stride*/1);
372 GetBuffer<float>(output_state_in_), n_batch, output_gate_scratch, /*result_stride*/1);
379 GetBuffer<float>(cell_state_in_), n_batch, input_gate_scratch);
382 n_cell * n_batch,
390 GetBuffer<float>(cell_state_in_), n_batch, forget_gate_scratch);
393 n_cell * n_batch,
398 forget_gate_scratch, GetBuffer<float>(cell_state_in_), n_batch * n_cell,
401 cell_scratch, n_batch * n_cell,
404 tflite::tensor_utils::Sub1Vector(forget_gate_scratch, n_batch * n_cell,
407 cell_scratch, forget_gate_scratch, n_batch * n_cell,
411 cell_scratch, input_gate_scratch, n_batch * n_cell,
415 tflite::tensor_utils::ClipVector(GetBuffer<float>(cell_state_out_), n_batch * n_cell,
423 GetBuffer<float>(cell_state_out_), n_batch, output_gate_scratch);
425 tflite::tensor_utils::ApplySigmoidToVector(output_gate_scratch, n_batch * n_cell,
428 n_batch * n_cell,
432 cell_scratch, n_batch * n_cell,
442 n_batch, GetBuffer<float>(output_));
444 tflite::tensor_utils::ZeroVector(GetBuffer<float>(output_), n_batch * n_output);
448 output_gate_scratch, n_batch, GetBuffer<float>(output_),
451 tflite::tensor_utils::ClipVector(GetBuffer<float>(output_), n_batch * n_output,
455 tflite::tensor_utils::CopyVector(output_gate_scratch, n_batch * n_output,
458 tflite::tensor_utils::CopyVector(GetBuffer<float>(output_), n_batch * n_output,
