Searched refs:input_gate_scratch (Results 1 - 2 of 2) sorted by relevance

/external/tensorflow/tensorflow/contrib/lite/kernels/
H A Dlstm.cc365 float* input_gate_scratch = nullptr; local
374 input_gate_scratch = scratch_buffer->data.f;
383 n_batch, input_gate_scratch);
396 input_gate_scratch, /*result_stride=*/1);
412 output_state->data.f, n_batch, input_gate_scratch, /*result_stride=*/1);
429 input_gate_scratch);
431 tensor_utils::ApplySigmoidToVector(input_gate_scratch, n_cell * n_batch,
432 input_gate_scratch);
458 cell_scratch, input_gate_scratch, n_batch * n_cell, cell_state->data.f);
H A Dunidirectional_sequence_lstm.cc368 float* input_gate_scratch = nullptr; local
377 input_gate_scratch = scratch_buffer->data.f;
388 n_batch, input_gate_scratch);
401 n_batch, input_gate_scratch, /*result_stride=*/1);
417 output_state->data.f, n_batch, input_gate_scratch,
437 input_gate_scratch);
439 tensor_utils::ApplySigmoidToVector(input_gate_scratch, n_cell * n_batch,
440 input_gate_scratch);
466 cell_scratch, input_gate_scratch, n_batch * n_cell,

Completed in 78 milliseconds