Searched refs:pooling_descriptor (Results 1 – 2 of 2) sorted by relevance
874 ScopedPoolingDescriptor(const PoolingDescriptor& pooling_descriptor) in ScopedPoolingDescriptor() argument882 absl::Span<const int64> strides64 = pooling_descriptor.strides(); in ScopedPoolingDescriptor()883 absl::Span<const int64> padding64 = pooling_descriptor.padding(); in ScopedPoolingDescriptor()884 absl::Span<const int64> shape64 = pooling_descriptor.window(); in ScopedPoolingDescriptor()886 const int nd = pooling_descriptor.ndims(); in ScopedPoolingDescriptor()899 (pooling_descriptor.mode() == dnn::PoolingMode::kMaximum in ScopedPoolingDescriptor()
862 const dnn::PoolingDescriptor& pooling_descriptor) in CudnnPoolingDescriptor() argument864 absl::Span<const int64> strides64 = pooling_descriptor.strides(); in CudnnPoolingDescriptor()865 absl::Span<const int64> padding64 = pooling_descriptor.padding(); in CudnnPoolingDescriptor()866 absl::Span<const int64> shape64 = pooling_descriptor.window(); in CudnnPoolingDescriptor()868 const int nd = pooling_descriptor.ndims(); in CudnnPoolingDescriptor()878 bool propagate_nans = pooling_descriptor.propagate_nans(); in CudnnPoolingDescriptor()884 (pooling_descriptor.mode() == dnn::PoolingMode::kMaximum in CudnnPoolingDescriptor()