Searched refs:kNumGruGates (Results 1 – 1 of 1) sorted by relevance
96 constexpr size_t kNumGruGates = 3; // Update, reset, output. variable108 rtc::CheckedDivExact(tensor_src.size(), output_size * kNumGruGates); in GetPreprocessedGruTensor()109 const size_t stride_src = kNumGruGates * output_size; in GetPreprocessedGruTensor()112 for (size_t g = 0; g < kNumGruGates; ++g) { in GetPreprocessedGruTensor()335 RTC_DCHECK_EQ(kNumGruGates * output_size_, bias_.size()) in GatedRecurrentLayer()337 RTC_DCHECK_EQ(kNumGruGates * input_size_ * output_size_, weights_.size()) in GatedRecurrentLayer()339 RTC_DCHECK_EQ(kNumGruGates * output_size_ * output_size_, in GatedRecurrentLayer()