Searched defs:pooling_desc (Results 1 – 4 of 4) sorted by relevance
72 se::dnn::PoolingDescriptor pooling_desc(3); in Compute() local199 se::dnn::PoolingDescriptor pooling_desc(3); in Compute() local
355 se::dnn::PoolingDescriptor pooling_desc; in Compute() local676 se::dnn::PoolingDescriptor pooling_desc; in Compute() local
4040 ScopedPoolingDescriptor pooling_desc{pooling_dimensions}; in DoPoolForward() local4197 ScopedPoolingDescriptor pooling_desc{pooling_dimensions}; in DoPoolBackward() local
6244 CudnnPoolingDescriptor pooling_desc(pooling_dimensions); in DoPoolForward() local6311 CudnnPoolingDescriptor pooling_desc(pooling_dimensions); in DoPoolBackward() local