/**
|
* @license
|
* Copyright 2020 Google LLC. All Rights Reserved.
|
* Licensed under the Apache License, Version 2.0 (the "License");
|
* you may not use this file except in compliance with the License.
|
* You may obtain a copy of the License at
|
*
|
* http://www.apache.org/licenses/LICENSE-2.0
|
*
|
* Unless required by applicable law or agreed to in writing, software
|
* distributed under the License is distributed on an "AS IS" BASIS,
|
* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
|
* See the License for the specific language governing permissions and
|
* limitations under the License.
|
* =============================================================================
|
*/
|
/// <amd-module name="@tensorflow/tfjs-core/dist/ops/dilation2d" />
|
import { Tensor3D, Tensor4D } from '../tensor';
|
import { TensorLike } from '../types';
|
/**
|
* Computes the grayscale dilation over the input `x`.
|
*
|
* @param x The input tensor, rank 3 or rank 4 of shape
|
* `[batch, height, width, depth]`. If rank 3, batch of 1 is assumed.
|
* @param filter The filter tensor, rank 3, of shape
|
* `[filterHeight, filterWidth, depth]`.
|
* @param strides The strides of the sliding window for each dimension of the
|
* input tensor: `[strideHeight, strideWidth]`.
|
* If `strides` is a single number,
|
* then `strideHeight == strideWidth`.
|
* @param pad The type of padding algorithm.
|
* - `same` and stride 1: output will be of same size as input,
|
* regardless of filter size.
|
* - `valid`: output will be smaller than input if filter is larger
|
* than 1*1x1.
|
* - For more info, see this guide:
|
* [https://www.tensorflow.org/api_docs/python/tf/nn/convolution](
|
* https://www.tensorflow.org/api_docs/python/tf/nn/convolution)
|
* @param dataFormat Specify the data format of the input and output data.
|
* Defaults to 'NHWC'. Only 'NHWC' is currently supported. With the
|
* default format "NHWC", the data is stored in the order of: [batch,
|
* height, width, channels].
|
* @param dilations The dilation rates: `[dilationHeight, dilationWidth]`
|
* in which we sample input values across the height and width dimensions
|
* for atrous morphological dilation. Defaults to `[1, 1]`. If `dilations`
|
* is a single number, then `dilationHeight == dilationWidth`. If it is
|
* greater than 1, then all values of `strides` must be 1.
|
*
|
* @doc {heading: 'Operations', subheading: 'Convolution'}
|
*/
|
declare function dilation2d_<T extends Tensor3D | Tensor4D>(x: T | TensorLike, filter: Tensor3D | TensorLike, strides: [number, number] | number, pad: 'valid' | 'same', dilations?: [number, number] | number, dataFormat?: 'NHWC'): T;
|
export declare const dilation2d: typeof dilation2d_;
|
export {};
|
