Home
last modified time | relevance | path

Searched refs:dilation_rate (Results 1 – 25 of 70) sorted by relevance

123

/external/tensorflow/tensorflow/python/kernel_tests/
Dpool_test.py40 dilation_rate, argument
64 effective_window_size = (window_size - 1) * dilation_rate + 1
91 input_start_pos += dilation_rate
92 input_slice = np.s_[input_start_pos:input_end_pos:dilation_rate]
104 dilation_rate, argument
141 dilation_rate=dilation_rate[i],
166 dilation_rate=[1, 1],
176 for dilation_rate in [[1], [2], [3]]:
182 dilation_rate=dilation_rate,
192 dilation_rate=[1],
[all …]
Datrous_convolution_test.py99 dilation_rate, **kwargs): argument
102 filters_upsampled = upsample_filters(filters, dilation_rate)
105 input=x, filter=filters, dilation_rate=dilation_rate, **kwargs)
118 x, w, "VALID", dilation_rate=[2, 2], data_format="NHWC")
126 x, w, "VALID", dilation_rate=[2, 2], data_format="NCHW")
135 for dilation_rate in [[1, 1], [3, 2], [2, 1]]:
141 dilation_rate=dilation_rate,
152 for dilation_rate in [[1, 1, 1], [3, 3, 3], [3, 2, 3], [3, 1, 2]]:
160 dilation_rate=dilation_rate,
175 dilation_rate=[rate],
[all …]
Dconv_ops_3d_test.py155 dilation_rate=dilation,
821 dilation_rate=dilations,
/external/tensorflow/tensorflow/python/keras/legacy_tf_layers/
Dconvolutional.py89 dilation_rate=1, argument
108 dilation_rate=dilation_rate,
129 dilation_rate=1, argument
211 dilation_rate=dilation_rate,
294 dilation_rate=(1, 1), argument
313 dilation_rate=dilation_rate,
334 dilation_rate=(1, 1), argument
423 dilation_rate=dilation_rate,
507 dilation_rate=(1, 1, 1), argument
526 dilation_rate=dilation_rate,
[all …]
/external/tensorflow/tensorflow/lite/kernels/
Dpadding.h24 inline int ComputePadding(int stride, int dilation_rate, int in_size, in ComputePadding() argument
26 int effective_filter_size = (filter_size - 1) * dilation_rate + 1; in ComputePadding()
33 inline int ComputePaddingWithOffset(int stride, int dilation_rate, int in_size, in ComputePaddingWithOffset() argument
36 int effective_filter_size = (filter_size - 1) * dilation_rate + 1; in ComputePaddingWithOffset()
46 int filter_size, int stride, int dilation_rate = 1) {
47 int effective_filter_size = (filter_size - 1) * dilation_rate + 1;
/external/tensorflow/tensorflow/python/keras/layers/
Dconvolutional.py120 dilation_rate=1, argument
151 self.dilation_rate = conv_utils.normalize_tuple(
152 dilation_rate, rank, 'dilation_rate')
229 tf_dilations = list(self.dilation_rate)
288 dilation=self.dilation_rate[i])
321 self.dilation_rate,
348 left_pad = self.dilation_rate[0] * (self.kernel_size[0] - 1)
490 dilation_rate=1, argument
509 dilation_rate=dilation_rate,
649 dilation_rate=(1, 1), argument
[all …]
Dconvolutional_recurrent.py199 dilation=cell.dilation_rate[0])
204 dilation=cell.dilation_rate[1])
497 dilation_rate=(1, 1), argument
520 self.dilation_rate = conv_utils.normalize_tuple(dilation_rate, 2,
654 dilation_rate=self.dilation_rate)
672 'dilation_rate': self.dilation_rate,
834 dilation_rate=(1, 1), argument
861 dilation_rate=dilation_rate,
913 def dilation_rate(self): member in ConvLSTM2D
914 return self.cell.dilation_rate
[all …]
/external/tensorflow/tensorflow/core/framework/
Dkernel_shape_util.cc21 int64 dilation_rate, int64 stride, in GetWindowedOutputSizeVerboseV2() argument
28 if (dilation_rate < 1) { in GetWindowedOutputSizeVerboseV2()
30 dilation_rate); in GetWindowedOutputSizeVerboseV2()
34 int64 effective_filter_size = (filter_size - 1) * dilation_rate + 1; in GetWindowedOutputSizeVerboseV2()
91 int64 dilation_rate, int64 stride, in GetWindowedOutputSizeV2() argument
100 return GetWindowedOutputSizeVerboseV2(input_size, filter_size, dilation_rate, in GetWindowedOutputSizeV2()
Dkernel_shape_util.h103 int64 dilation_rate, int64 stride,
122 int64 dilation_rate, int64 stride,
Dcommon_shape_fns.h42 DimensionOrConstant filter_size, int64 dilation_rate, int64 stride,
/external/tensorflow/tensorflow/python/compiler/tensorrt/test/
Dconv2d_test.py38 dilation_rate=(1, 1)): argument
47 dilations = [1] + list(dilation_rate) + [1]
51 dilations = [1, 1] + list(dilation_rate)
75 for dilation_rate in dilation_rates:
77 data_format, dilation_rate)
158 dilation_rate=(2, 3),
/external/tensorflow/tensorflow/compiler/xla/client/lib/
Dconv_grad_size_util.cc25 int64 input_size, int64 filter_size, int64 dilation_rate, int64 stride, in GetWindowedOutputSize() argument
31 if (dilation_rate < 1) { in GetWindowedOutputSize()
33 "Dilation rate must be >= 1, but got ", dilation_rate); in GetWindowedOutputSize()
36 int64 effective_filter_size = (filter_size - 1) * dilation_rate + 1; in GetWindowedOutputSize()
/external/tensorflow/tensorflow/python/ops/
Dnn_ops.py417 dilation_rate, argument
565 dilation_rate,
595 dilation_rate, argument
603 dilation_rate = ops.convert_to_tensor(
604 dilation_rate, dtypes.int32, name="dilation_rate")
605 if dilation_rate.shape.ndims not in (None, 1):
607 "rate must be rank 1 but saw {}".format(dilation_rate.shape.ndims))
609 if not dilation_rate.shape.is_fully_defined():
611 .format(dilation_rate.shape))
613 num_spatial_dims = dilation_rate.shape.dims[0].value
[all …]
/external/tensorflow/tensorflow/tools/api/golden/v1/
Dtensorflow.layers.pbtxt101 …ters\', \'kernel_size\', \'strides\', \'padding\', \'data_format\', \'dilation_rate\', \'activatio…
105 …ters\', \'kernel_size\', \'strides\', \'padding\', \'data_format\', \'dilation_rate\', \'activatio…
113 …ters\', \'kernel_size\', \'strides\', \'padding\', \'data_format\', \'dilation_rate\', \'activatio…
145 …ters\', \'kernel_size\', \'strides\', \'padding\', \'data_format\', \'dilation_rate\', \'depth_mul…
149 …ters\', \'kernel_size\', \'strides\', \'padding\', \'data_format\', \'dilation_rate\', \'depth_mul…
Dtensorflow.keras.backend.pbtxt93 …argspec: "args=[\'x\', \'kernel\', \'strides\', \'padding\', \'data_format\', \'dilation_rate\'], …
97 …argspec: "args=[\'x\', \'kernel\', \'strides\', \'padding\', \'data_format\', \'dilation_rate\'], …
101 …nel\', \'output_shape\', \'strides\', \'padding\', \'data_format\', \'dilation_rate\'], varargs=No…
105 …argspec: "args=[\'x\', \'kernel\', \'strides\', \'padding\', \'data_format\', \'dilation_rate\'], …
137 …', \'depthwise_kernel\', \'strides\', \'padding\', \'data_format\', \'dilation_rate\'], varargs=No…
441 …', \'pointwise_kernel\', \'strides\', \'padding\', \'data_format\', \'dilation_rate\'], varargs=No…
Dtensorflow.keras.layers.-conv-l-s-t-m2-d.pbtxt41 name: "dilation_rate"
222 …ters\', \'kernel_size\', \'strides\', \'padding\', \'data_format\', \'dilation_rate\', \'activatio…
Dtensorflow.nn.pbtxt113 …argspec: "args=[\'input\', \'filter\', \'padding\', \'strides\', \'dilation_rate\', \'name\', \'da…
293 …argspec: "args=[\'input\', \'window_shape\', \'pooling_type\', \'padding\', \'dilation_rate\', \'s…
433 …argspec: "args=[\'input\', \'dilation_rate\', \'padding\', \'op\', \'filter_shape\', \'spatial_dim…
/external/tensorflow/tensorflow/compiler/tests/
Dconv_node_name_test.py58 dilation_rate=dilations,
67 dilation_rate=dilations,
/external/tensorflow/tensorflow/python/keras/
Dbackend.py5332 dilation_rate=1): argument
5358 left_pad = dilation_rate * (kernel_shape[0] - 1)
5367 dilation_rate=dilation_rate,
5384 dilation_rate=(1, 1)): argument
5412 dilation_rate=dilation_rate,
5430 dilation_rate=(1, 1)): argument
5457 if data_format == 'channels_first' and dilation_rate != (1, 1):
5479 if dilation_rate == (1, 1):
5484 assert dilation_rate[0] == dilation_rate[1]
5489 rate=dilation_rate[0],
[all …]
/external/tensorflow/tensorflow/tools/api/golden/v2/
Dtensorflow.keras.backend.pbtxt89 …argspec: "args=[\'x\', \'kernel\', \'strides\', \'padding\', \'data_format\', \'dilation_rate\'], …
93 …argspec: "args=[\'x\', \'kernel\', \'strides\', \'padding\', \'data_format\', \'dilation_rate\'], …
97 …nel\', \'output_shape\', \'strides\', \'padding\', \'data_format\', \'dilation_rate\'], varargs=No…
101 …argspec: "args=[\'x\', \'kernel\', \'strides\', \'padding\', \'data_format\', \'dilation_rate\'], …
133 …', \'depthwise_kernel\', \'strides\', \'padding\', \'data_format\', \'dilation_rate\'], varargs=No…
437 …', \'pointwise_kernel\', \'strides\', \'padding\', \'data_format\', \'dilation_rate\'], varargs=No…
Dtensorflow.keras.layers.-conv-l-s-t-m2-d.pbtxt41 name: "dilation_rate"
222 …ters\', \'kernel_size\', \'strides\', \'padding\', \'data_format\', \'dilation_rate\', \'activatio…
Dtensorflow.keras.layers.-conv1-d-transpose.pbtxt134 …e\', \'strides\', \'padding\', \'output_padding\', \'data_format\', \'dilation_rate\', \'activatio…
Dtensorflow.keras.layers.-convolution3-d.pbtxt133 …ters\', \'kernel_size\', \'strides\', \'padding\', \'data_format\', \'dilation_rate\', \'groups\',…
Dtensorflow.keras.layers.-separable-convolution2-d.pbtxt134 …ters\', \'kernel_size\', \'strides\', \'padding\', \'data_format\', \'dilation_rate\', \'depth_mul…
/external/llvm-project/mlir/include/mlir/Dialect/Linalg/IR/
DLinalgStructuredOps.td227 x[0] * strides[0] - pad_before[0] + dilation_rate[0]*z[0],
229 x[N-1]*strides[N-1] - pad_before[N-1] + dilation_rate[N-1]*z[N-1]
295 x[0] * strides[0] + dilation_rate[0] * z[0],
297 x[N-1] * strides[N-1] + dilation_rate[N-1] * z[N-1],

123