Searched refs:model_dims (Results 1 – 2 of 2) sorted by relevance
1426 RnnModelDims model_dims; in ExtractAndCheckRnnForward() local1427 model_dims.num_layers = rnn_desc.num_layers(); in ExtractAndCheckRnnForward()1428 model_dims.batch_size = input_desc.batch_size(); in ExtractAndCheckRnnForward()1429 model_dims.max_seq_length = input_desc.max_seq_length(); in ExtractAndCheckRnnForward()1430 model_dims.hidden_size = rnn_desc.hidden_size(); in ExtractAndCheckRnnForward()1431 model_dims.input_size = input_desc.data_size(); in ExtractAndCheckRnnForward()1432 model_dims.dir_count = in ExtractAndCheckRnnForward()1437 model_dims.num_layers * model_dims.dir_count && in ExtractAndCheckRnnForward()1438 input_h_desc.batch_size() == model_dims.batch_size && in ExtractAndCheckRnnForward()1439 input_h_desc.data_size() == model_dims.hidden_size)) { in ExtractAndCheckRnnForward()[all …]
1904 const DeviceMemory<T>& output_c_data, RnnModelDims* model_dims) { in ExtractAndCheckRnnForward() argument1906 model_dims->num_layers = rnn_desc.num_layers(); in ExtractAndCheckRnnForward()1907 model_dims->batch_size = input_desc.batch_size(); in ExtractAndCheckRnnForward()1908 model_dims->seq_length = input_desc.seq_length(); in ExtractAndCheckRnnForward()1909 model_dims->hidden_size = rnn_desc.hidden_size(); in ExtractAndCheckRnnForward()1910 model_dims->input_size = input_desc.data_size(); in ExtractAndCheckRnnForward()1911 model_dims->dir_count = in ExtractAndCheckRnnForward()1916 model_dims->num_layers * model_dims->dir_count && in ExtractAndCheckRnnForward()1917 input_h_desc.batch_size() == model_dims->batch_size && in ExtractAndCheckRnnForward()1918 input_h_desc.data_size() == model_dims->hidden_size)) { in ExtractAndCheckRnnForward()[all …]