Searched refs:num_units_ (Results 1 – 3 of 3) sorted by relevance
85 int num_units(Plane plane) const { return num_units_[plane]; } in num_units()99 int num_units_[kMaxPlanes]; variable
83 num_units_[plane] = in Reset()85 total_num_units += num_units_[plane]; in Reset()99 loop_restoration_info += num_units_[plane]; in Reset()
156 num_units_(num_units), in CudnnRnnParameters()185 std::to_string(num_units_), in ToString()202 return std::make_tuple(num_layers_, input_size_, num_units_, seq_length_, in get_data_as_tuple()209 const int num_units_; member in tensorflow::__anonebbbcdba0111::CudnnRnnParameters