Searched refs:pooling_descriptor (Results 1 – 2 of 2) sorted by relevance
719 const dnn::PoolingDescriptor& pooling_descriptor) in CudnnPoolingDescriptor() argument721 absl::Span<const int64> strides64 = pooling_descriptor.strides(); in CudnnPoolingDescriptor()722 absl::Span<const int64> padding64 = pooling_descriptor.padding(); in CudnnPoolingDescriptor()723 absl::Span<const int64> shape64 = pooling_descriptor.window(); in CudnnPoolingDescriptor()725 const int nd = pooling_descriptor.ndims(); in CudnnPoolingDescriptor()735 bool propagate_nans = pooling_descriptor.propagate_nans(); in CudnnPoolingDescriptor()741 (pooling_descriptor.mode() == dnn::PoolingMode::kMaximum in CudnnPoolingDescriptor()
708 ScopedPoolingDescriptor(const PoolingDescriptor& pooling_descriptor) in ScopedPoolingDescriptor() argument716 absl::Span<const int64> strides64 = pooling_descriptor.strides(); in ScopedPoolingDescriptor()717 absl::Span<const int64> padding64 = pooling_descriptor.padding(); in ScopedPoolingDescriptor()718 absl::Span<const int64> shape64 = pooling_descriptor.window(); in ScopedPoolingDescriptor()720 const int nd = pooling_descriptor.ndims(); in ScopedPoolingDescriptor()738 (pooling_descriptor.mode() == dnn::PoolingMode::kMaximum in ScopedPoolingDescriptor()