Searched defs:scratch_cell (Results 1 – 8 of 8) sorted by relevance
152 rnn_utils::ws_gates_aoc<scratch_t> scratch_cell(rnn, scratch_cell_); in postgemm_fwd_call() local229 ws_gates_aoc<scratch_t> scratch_cell(rnn, scratch_cell_); in rnn_postgemm_sig() local
158 const rnn_utils::ws_gates_aoc<scratch_t> scratch_cell( in postgemm_fwd_call() local237 const ws_gates_aoc<scratch_t> scratch_cell(rnn, scratch_cell_); in rnn_postgemm_sig() local
54 ws_gates_aoc<scratch_data_t> scratch_cell(rnn, scratch_cell_); in gru_lbr_fwd_postgemm_template() local
1165 auto scratch_cell = scratchpad.template get<scratch_t>(key_rnn_cell); in execute_() local
61 const ws_gates_aoc<scratch_data_t> scratch_cell(rnn, scratch_cell_); in gru_lbr_fwd_postgemm_template() local
1224 auto scratch_cell = scratchpad.template get<scratch_t>(key_rnn_cell); in execute_() local
1082 const memory_storage_t &scratch_cell, in gates_reduction()1413 auto &scratch_cell in execute_() local
1100 const memory_storage_t &scratch_cell, in gates_reduction()1445 auto &scratch_cell = this->pd()->is_lbr() || is_vanilla_gru in execute_() local