Searched defs:pooling_desc (Results 1 – 4 of 4) sorted by relevance
71 se::dnn::PoolingDescriptor pooling_desc(3); in Compute() local204 se::dnn::PoolingDescriptor pooling_desc(3); in Compute() local
352 se::dnn::PoolingDescriptor pooling_desc; in Compute() local667 se::dnn::PoolingDescriptor pooling_desc; in Compute() local
4274 CudnnPoolingDescriptor pooling_desc(pooling_dimensions); in DoPoolForward() local4299 CudnnPoolingDescriptor pooling_desc(pooling_dimensions); in DoPoolForward() local4325 CudnnPoolingDescriptor pooling_desc(pooling_dimensions); in DoPoolForward() local4349 CudnnPoolingDescriptor pooling_desc(pooling_dimensions); in DoPoolForward() local4377 CudnnPoolingDescriptor pooling_desc(pooling_dimensions); in DoPoolBackward() local4407 CudnnPoolingDescriptor pooling_desc(pooling_dimensions); in DoPoolBackward() local4437 CudnnPoolingDescriptor pooling_desc(pooling_dimensions); in DoPoolBackward() local
4109 ScopedPoolingDescriptor pooling_desc{pooling_dimensions}; in DoPoolForward() local4177 ScopedPoolingDescriptor pooling_desc{pooling_dimensions}; in DoPoolForward() local4214 ScopedPoolingDescriptor pooling_desc{pooling_dimensions}; in DoPoolBackwardImpl() local