Searched refs:accum_scratch_dims (Results 1 – 9 of 9) sorted by relevance
| /external/tensorflow/tensorflow/lite/kernels/ |
| D | basic_rnn.cc | 160 int accum_scratch_dims[2] = {num_units, batch_size}; in Prepare() local
162 accum_scratch_dims)) { in Prepare()
164 accum_scratch_size->data[0] = accum_scratch_dims[0]; in Prepare()
165 accum_scratch_size->data[1] = accum_scratch_dims[1]; in Prepare()
|
| D | unidirectional_sequence_rnn.cc | 167 int accum_scratch_dims[2] = {num_units, batch_size}; in Prepare() local
169 accum_scratch_dims)) { in Prepare()
171 accum_scratch_size->data[0] = accum_scratch_dims[0]; in Prepare()
172 accum_scratch_size->data[1] = accum_scratch_dims[1]; in Prepare()
|
| D | bidirectional_sequence_rnn.cc | 269 int accum_scratch_dims[2] = {std::max(fw_num_units, bw_num_units), in Prepare() local
272 accum_scratch_dims)) { in Prepare()
274 accum_scratch_size->data[0] = accum_scratch_dims[0]; in Prepare()
275 accum_scratch_size->data[1] = accum_scratch_dims[1]; in Prepare()
|
| D | batch_matmul.cc | 251 int accum_scratch_dims[2] = {num_units, batch_size}; in InitializeTemporaries() local
253 accum_scratch_dims)) { in InitializeTemporaries()
|
| D | fully_connected.cc | 303 int accum_scratch_dims[2] = {num_units, batch_size}; in PrepareImpl() local
305 accum_scratch_dims)) { in PrepareImpl()
|
| D | conv.cc | 544 int accum_scratch_dims[2] = {channels_out, scratch_width}; in Prepare() local
546 accum_scratch_dims)) { in Prepare()
|
| D | bidirectional_sequence_lstm.cc | 840 int accum_scratch_dims[2] = {n_cell, n_batch}; in Prepare() local
842 accum_scratch_dims)) { in Prepare()
|
| D | unidirectional_sequence_lstm.cc | 1101 int accum_scratch_dims[2] = {n_cell, n_batch}; in Prepare() local
1103 accum_scratch_dims)) { in Prepare()
|
| D | lstm.cc | 1559 int accum_scratch_dims[2] = {n_cell, n_batch}; in Prepare() local
1561 accum_scratch_dims)) { in Prepare()
|