Searched refs:out_shard (Results 1 – 2 of 2) sorted by relevance
212 EigenMatrixMap out_shard(out_mat.data() + start * output_image_size, in SpatialMaxPool()214 out_shard.setConstant(Eigen::NumTraits<T>::lowest()); in SpatialMaxPool()459 EigenMatrixMap out_shard(out_mat.data() + start * output_image_size,461 out_shard.setConstant(Eigen::NumTraits<T>::lowest());
118 EigenMatrixMap out_shard(out_mat.data() + start * output_image_size, 1, in SpatialMaxPoolWithArgMaxHelper() local120 out_shard.setConstant(Eigen::NumTraits<T>::lowest()); in SpatialMaxPoolWithArgMaxHelper()