Lines Matching refs:time_major
189 params_.time_major = getScalarDataWithDefault<bool>(timeMajorOperand, false); in BidirectionalSequenceLSTM()
269 const uint32_t max_time = SizeOfDimension(input_, params_.time_major ? 0 : 1); in Prepare()
270 const uint32_t n_batch = SizeOfDimension(input_, params_.time_major ? 1 : 0); in Prepare()
388 fwOutputShape->dimensions[0] = params_.time_major ? max_time : n_batch; in Prepare()
389 fwOutputShape->dimensions[1] = params_.time_major ? n_batch : max_time; in Prepare()
414 bwOutputShape->dimensions[0] = params_.time_major ? max_time : n_batch; in Prepare()
415 bwOutputShape->dimensions[1] = params_.time_major ? n_batch : max_time; in Prepare()
517 GetBuffer<float>(fw_output_), fw_scratch_buffer.data(), params_.time_major, in Eval()
571 bw_scratch_buffer.data(), params_.time_major, kBackwardSequence); in Eval()
641 GetBuffer<_Float16>(fw_output_), fw_scratch_buffer.data(), params_.time_major, in Eval()
696 bw_scratch_buffer.data(), params_.time_major, kBackwardSequence); in Eval()