Function GpuRNNSetParam

src/model/operation/rnn.cc:498–505

}

void GpuRNNSetParam(int linLayerID, int pseudoLayer, Tensor &weights,
                    Tensor &paramValues, bool is_bias, CudnnRNNHandle &h) {
  size_t offset, size;
  std::tie(offset, size) =
      h.weights_mapping[std::make_tuple(linLayerID, pseudoLayer, is_bias)];
  CHECK_EQ(size, paramValues.size()) << "param size is not expected";
  CopyDataToFrom(&weights, paramValues, size, offset, 0);
}

Tensor GpuRNNGetParamCopy(int linLayerID, int pseudoLayer, Tensor &weights,
                          bool is_bias, CudnnRNNHandle &h) {