Searched refs:input_gate_scratch (Results 1 - 2 of 2) sorted by relevance
/external/tensorflow/tensorflow/contrib/lite/kernels/ |
H A D | lstm.cc | 365 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 D | unidirectional_sequence_lstm.cc | 368 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