| 414 | } |
| 415 | |
| 416 | int64 BatchDescriptor::ElementCount() const { |
| 417 | return count() * feature_map_count() * NodesPerFeatureMap(); |
| 418 | } |
| 419 | |
| 420 | int64 BatchDescriptor::FullyConnectedWeightCount( |
| 421 | const BatchDescriptor& input, const BatchDescriptor& output) { |