Searched refs:pooling_descriptor (Results 1 – 2 of 2) sorted by relevance
780 const dnn::PoolingDescriptor& pooling_descriptor) in CudnnPoolingDescriptor() argument782 absl::Span<const int64> strides64 = pooling_descriptor.strides(); in CudnnPoolingDescriptor()783 absl::Span<const int64> padding64 = pooling_descriptor.padding(); in CudnnPoolingDescriptor()784 absl::Span<const int64> shape64 = pooling_descriptor.window(); in CudnnPoolingDescriptor()786 const int nd = pooling_descriptor.ndims(); in CudnnPoolingDescriptor()796 bool propagate_nans = pooling_descriptor.propagate_nans(); in CudnnPoolingDescriptor()802 (pooling_descriptor.mode() == dnn::PoolingMode::kMaximum in CudnnPoolingDescriptor()
823 ScopedPoolingDescriptor(const PoolingDescriptor& pooling_descriptor) in ScopedPoolingDescriptor() argument831 absl::Span<const int64> strides64 = pooling_descriptor.strides(); in ScopedPoolingDescriptor()832 absl::Span<const int64> padding64 = pooling_descriptor.padding(); in ScopedPoolingDescriptor()833 absl::Span<const int64> shape64 = pooling_descriptor.window(); in ScopedPoolingDescriptor()835 const int nd = pooling_descriptor.ndims(); in ScopedPoolingDescriptor()848 (pooling_descriptor.mode() == dnn::PoolingMode::kMaximum in ScopedPoolingDescriptor()