Searched refs:NodesPerFeatureMap (Results 1 – 4 of 4) sorted by relevance
332 int64 BatchDescriptor::NodesPerFeatureMap() const { in NodesPerFeatureMap() function in stream_executor::dnn::BatchDescriptor341 return NodesPerFeatureMap() * feature_map_count(); in NodesAcrossFeatureMaps()345 return count() * feature_map_count() * NodesPerFeatureMap(); in ElementCount()
318 int64 NodesPerFeatureMap() const;
3620 const int batch_count = output_dimensions.NodesPerFeatureMap(); in DoMatMul()
3380 const int batch_count = output_dimensions.NodesPerFeatureMap(); in DoMatMul()