Home
last modified time | relevance | path

Searched refs:dilation_rate (Results 1 – 25 of 61) 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,
725 dilation_rate=dilations,
/external/tensorflow/tensorflow/python/layers/
Dconvolutional.py85 dilation_rate=1, argument
104 dilation_rate=dilation_rate,
128 dilation_rate=1, argument
204 dilation_rate=dilation_rate,
284 dilation_rate=(1, 1), argument
303 dilation_rate=dilation_rate,
327 dilation_rate=(1, 1), argument
410 dilation_rate=dilation_rate,
491 dilation_rate=(1, 1, 1), argument
510 dilation_rate=dilation_rate,
[all …]
/external/tensorflow/tensorflow/python/keras/layers/
Dconvolutional.py105 dilation_rate=1, argument
134 self.dilation_rate = conv_utils.normalize_tuple(
135 dilation_rate, rank, 'dilation_rate')
188 dilation_rate=self.dilation_rate,
224 dilation=self.dilation_rate[i])
237 dilation=self.dilation_rate[i])
249 'dilation_rate': self.dilation_rate,
266 left_pad = self.dilation_rate[0] * (self.kernel_size[0] - 1)
340 dilation_rate=1, argument
358 dilation_rate=dilation_rate,
[all …]
Dconvolutional_recurrent.py200 dilation=cell.dilation_rate[0])
205 dilation=cell.dilation_rate[1])
555 dilation_rate=(1, 1), argument
578 self.dilation_rate = conv_utils.normalize_tuple(dilation_rate, 2,
712 dilation_rate=self.dilation_rate)
730 'dilation_rate': self.dilation_rate,
883 dilation_rate=(1, 1), argument
909 dilation_rate=dilation_rate,
961 def dilation_rate(self): member in ConvLSTM2D
962 return self.cell.dilation_rate
[all …]
/external/tensorflow/tensorflow/python/ops/
Dnn_ops.py324 dilation_rate, argument
472 dilation_rate,
501 dilation_rate, argument
508 dilation_rate = ops.convert_to_tensor(
509 dilation_rate, dtypes.int32, name="dilation_rate")
511 rate_shape = dilation_rate.get_shape().with_rank(1)
515 if not dilation_rate.get_shape().is_fully_defined():
546 const_rate = tensor_util.constant_value(dilation_rate)
547 rate_or_const_rate = dilation_rate
581 self.dilation_rate = dilation_rate
[all …]
/external/tensorflow/tensorflow/contrib/model_pruning/python/layers/
Dcore_layers.py93 dilation_rate=1, argument
115 self.dilation_rate = utils.normalize_tuple(dilation_rate, rank,
183 dilation_rate=self.dilation_rate,
225 dilation=self.dilation_rate[i])
238 dilation=self.dilation_rate[i])
299 dilation_rate=(1, 1), argument
317 dilation_rate=dilation_rate,
Dlayers.py225 dilation_rate=rate,
/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/compiler/tensorrt/test/
Dconv2d_test.py39 dilation_rate=(1, 1), argument
49 dilations = [1] + list(dilation_rate) + [1]
53 dilations = [1, 1] + list(dilation_rate)
82 for dilation_rate in dilation_rates:
84 data_format, dilation_rate)
173 dilation_rate=(2, 3),
/external/tensorflow/tensorflow/lite/kernels/
Dpadding.h22 inline int ComputePadding(int stride, int dilation_rate, int in_size, in ComputePadding() argument
24 int effective_filter_size = (filter_size - 1) * dilation_rate + 1; in ComputePadding()
43 int stride_height, int stride_width, int dilation_rate, int in_height, in ComputePaddingHeightWidth() argument
Ddepthwise_conv.cc136 int dilation_rate) -> int { in Prepare() argument
137 int effective_filter_size = (filter_size - 1) * dilation_rate + 1; in Prepare()
/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\'], …
429 …', \'pointwise_kernel\', \'strides\', \'padding\', \'data_format\', \'dilation_rate\'], varargs=No…
Dtensorflow.keras.layers.-conv-l-s-t-m2-d.pbtxt34 name: "dilation_rate"
179 …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…
421 …argspec: "args=[\'input\', \'dilation_rate\', \'padding\', \'op\', \'filter_shape\', \'spatial_dim…
Dtensorflow.keras.layers.-conv1-d.pbtxt90 …ters\', \'kernel_size\', \'strides\', \'padding\', \'data_format\', \'dilation_rate\', \'activatio…
/external/tensorflow/tensorflow/python/keras/
Dbackend.py4210 dilation_rate=1): argument
4236 left_pad = dilation_rate * (kernel_shape[0] - 1)
4245 dilation_rate=dilation_rate,
4260 dilation_rate=(1, 1)): argument
4290 dilation_rate=dilation_rate,
4306 dilation_rate=(1, 1)): argument
4337 if data_format == 'channels_first' and dilation_rate != (1, 1):
4357 if dilation_rate == (1, 1):
4362 assert dilation_rate[0] == dilation_rate[1]
4367 rate=dilation_rate[0],
[all …]
/external/tensorflow/tensorflow/core/framework/
Dcommon_shape_fns.h104 int64 dilation_rate, int64 stride,
123 int64 dilation_rate, int64 stride,
165 DimensionOrConstant filter_size, int64 dilation_rate, int64 stride,
Dcommon_shape_fns.cc22 int64 dilation_rate, int64 stride, in GetWindowedOutputSizeVerboseV2() argument
29 if (dilation_rate < 1) { in GetWindowedOutputSizeVerboseV2()
31 dilation_rate); in GetWindowedOutputSizeVerboseV2()
35 int64 effective_filter_size = (filter_size - 1) * dilation_rate + 1; in GetWindowedOutputSizeVerboseV2()
92 int64 dilation_rate, int64 stride, in GetWindowedOutputSizeV2() argument
101 return GetWindowedOutputSizeVerboseV2(input_size, filter_size, dilation_rate, in GetWindowedOutputSizeV2()
141 shape_inference::DimensionOrConstant filter_size, int64 dilation_rate, in GetWindowedOutputSizeFromDimsV2() argument
148 if (dilation_rate < 1) { in GetWindowedOutputSizeFromDimsV2()
150 dilation_rate); in GetWindowedOutputSizeFromDimsV2()
161 if (dilation_rate > 1) { in GetWindowedOutputSizeFromDimsV2()
[all …]
/external/tensorflow/tensorflow/tools/api/golden/v2/
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\'], …
425 …', \'pointwise_kernel\', \'strides\', \'padding\', \'data_format\', \'dilation_rate\'], varargs=No…
Dtensorflow.keras.layers.-conv-l-s-t-m2-d.pbtxt34 name: "dilation_rate"
179 …ters\', \'kernel_size\', \'strides\', \'padding\', \'data_format\', \'dilation_rate\', \'activatio…
/external/tensorflow/tensorflow/contrib/quantize/python/
Dquantize_parameterized_test.py335 dilation_rate = 2
344 rate=dilation_rate,
681 dilation_rate = 2
689 rate=dilation_rate,
Dfold_batch_norms_test.py569 dilation_rate = 2
577 rate=dilation_rate,
599 rate=dilation_rate,

123