HomeSort by relevance Sort by last modified time
    Searched full:forget_gate_scratch (Results 1 - 3 of 3) sorted by null

  /external/tensorflow/tensorflow/contrib/lite/kernels/
lstm.cc 367 float* forget_gate_scratch = nullptr; local
371 forget_gate_scratch = scratch_buffer->data.f + n_cell * n_batch;
376 forget_gate_scratch = scratch_buffer->data.f + 2 * n_cell * n_batch;
386 n_batch, forget_gate_scratch);
400 forget_gate_scratch, /*result_stride=*/1);
416 output_state->data.f, n_batch, forget_gate_scratch, /*result_stride=*/1);
439 forget_gate_scratch);
441 tensor_utils::ApplySigmoidToVector(forget_gate_scratch, n_cell * n_batch,
442 forget_gate_scratch);
445 tensor_utils::VectorVectorCwiseProduct(forget_gate_scratch,
    [all...]
unidirectional_sequence_lstm.cc 370 float* forget_gate_scratch = nullptr; local
374 forget_gate_scratch = scratch_buffer->data.f + n_cell * n_batch;
379 forget_gate_scratch = scratch_buffer->data.f + 2 * n_cell * n_batch;
391 n_batch, forget_gate_scratch);
405 n_batch, forget_gate_scratch, /*result_stride=*/1);
422 output_state->data.f, n_batch, forget_gate_scratch,
447 forget_gate_scratch);
449 tensor_utils::ApplySigmoidToVector(forget_gate_scratch, n_cell * n_batch,
450 forget_gate_scratch);
453 tensor_utils::VectorVectorCwiseProduct(forget_gate_scratch,
    [all...]
  /frameworks/ml/nn/common/operations/
LSTM.cpp 317 float* forget_gate_scratch = nullptr; local
321 forget_gate_scratch = cell_scratch + n_cell * n_batch;
326 forget_gate_scratch = input_gate_scratch + 2 * n_cell * n_batch;
336 n_cell, n_batch, forget_gate_scratch); local
350 GetBuffer<float>(input_), n_batch, forget_gate_scratch, /*result_stride*/1); local
366 GetBuffer<float>(output_state_in_), n_batch, forget_gate_scratch, /*result_stride*/1); local
390 GetBuffer<float>(cell_state_in_), n_batch, forget_gate_scratch); local
392 tflite::tensor_utils::ApplySigmoidToVector(forget_gate_scratch,
394 forget_gate_scratch);
398 forget_gate_scratch, GetBuffer<float>(cell_state_in_), n_batch * n_cell
    [all...]

Completed in 328 milliseconds