Searched refs:num_params_ (Results 1 – 1 of 1) sorted by relevance
1077 OP_REQUIRES_OK(context, context->GetAttr("num_params", &num_params_)); in CudnnRNNParamsToCanonical()1112 const int num_params_per_layer = num_params_ / num_layers / num_dirs; in Compute()1116 CHECK(num_params_ % (num_layers * num_dirs) == 0) in Compute()1121 CHECK(num_params_ == rnn_desc->ParamsWeightRegions().size()) in Compute()1122 << "Number of params mismatch. Expected " << num_params_ << ", got " in Compute()1165 OP_REQUIRES(context, num_params_ == rnn_desc->ParamsBiasRegions().size(), in Compute()1167 num_params_, ", got ", in Compute()1178 context->allocate_output(num_params_ + i, in Compute()1188 int num_params_; member in tensorflow::CudnnRNNParamsToCanonical