| 402 | } |
| 403 | |
| 404 | int64 BatchDescriptor::NodesPerFeatureMap() const { |
| 405 | int64 ret = 1; |
| 406 | for (int i = 0; i < ndims(); i++) { |
| 407 | ret *= spatial_size()[i]; |
| 408 | } |
| 409 | return ret; |
| 410 | } |
| 411 | |
| 412 | int64 BatchDescriptor::NodesAcrossFeatureMaps() const { |
| 413 | return NodesPerFeatureMap() * feature_map_count(); |