Searched refs:n_iter_scratch_gates (Results 1 – 12 of 12) sorted by relevance
162 int n_iter_scratch_gates; member
272 rnn.n_iter_scratch_gates in set_rnn_conf()274 rnn.scratch_gates_size = (size_t)rnn.n_iter_scratch_gates * rnn.gates_nld in set_rnn_conf()
100 conf.n_iter_scratch_gates = rnn.n_iter_scratch_gates; in init_conf()236 kernel_ctx.define_int("N_ITER_SCRATCH_GATES", conf.n_iter_scratch_gates); in init_kernel_ctx()
273 rnn.n_iter_scratch_gates in set_rnn_conf()275 rnn.scratch_gates_size = (size_t)rnn.n_iter_scratch_gates * rnn.gates_nld in set_rnn_conf()
185 int n_iter_scratch_gates; member
330 int n_iter_scratch_gates = 0; member936 rnn.n_iter_scratch_gates in set_conf()938 rnn.scratch_gates_size = rnn.n_iter_scratch_gates * rnn.scratch_gates_nld in set_conf()
311 auto cell_scratch_gates = rnn.n_iter_scratch_gates == 1
340 int n_iter_scratch_gates = 0; member974 rnn.n_iter_scratch_gates in set_conf()976 rnn.scratch_gates_size = rnn.n_iter_scratch_gates * rnn.scratch_gates_nld in set_conf()
315 const auto cell_scratch_gates = rnn.n_iter_scratch_gates == 1
409 int n_iter_scratch_gates; member
437 int n_iter_scratch_gates; member