Home
last modified time | relevance | path

Searched refs:normalize_tuple (Results 1 – 7 of 7) sorted by relevance

/aosp_15_r20/external/tensorflow/tensorflow/python/layers/
H A Dutils_test.py38 self.assertEqual((2, 2, 2), utils.normalize_tuple(2, n=3, name='strides'))
40 (2, 1, 2), utils.normalize_tuple((2, 1, 2), n=3, name='strides'))
43 utils.normalize_tuple((2, 1), n=3, name='strides')
46 utils.normalize_tuple(None, n=3, name='strides')
H A Dutils.py48 def normalize_tuple(value, n, name): function
/aosp_15_r20/external/tensorflow/tensorflow/python/keras/layers/
H A Dconvolutional.py144 self.kernel_size = conv_utils.normalize_tuple(
146 self.strides = conv_utils.normalize_tuple(strides, rank, 'strides')
149 self.dilation_rate = conv_utils.normalize_tuple(
959 self.output_padding = conv_utils.normalize_tuple(
1230 self.output_padding = conv_utils.normalize_tuple(
1540 self.output_padding = conv_utils.normalize_tuple(
2614 self.size = conv_utils.normalize_tuple(size, 2, 'size')
2699 self.size = conv_utils.normalize_tuple(size, 3, 'size')
2781 self.padding = conv_utils.normalize_tuple(padding, 2, 'padding')
2876 height_padding = conv_utils.normalize_tuple(padding[0], 2,
[all …]
H A Dpooling.py62 self.pool_size = conv_utils.normalize_tuple(pool_size, 1, 'pool_size')
63 self.strides = conv_utils.normalize_tuple(strides, 1, 'strides')
345 self.pool_size = conv_utils.normalize_tuple(pool_size, 2, 'pool_size')
346 self.strides = conv_utils.normalize_tuple(strides, 2, 'strides')
683 self.pool_size = conv_utils.normalize_tuple(pool_size, 3, 'pool_size')
684 self.strides = conv_utils.normalize_tuple(strides, 3, 'strides')
H A Dconvolutional_recurrent.py513 self.kernel_size = conv_utils.normalize_tuple(kernel_size, 2, 'kernel_size')
514 self.strides = conv_utils.normalize_tuple(strides, 2, 'strides')
517 self.dilation_rate = conv_utils.normalize_tuple(dilation_rate, 2,
/aosp_15_r20/external/tensorflow/tensorflow/python/keras/utils/
H A Dconv_utils.py50 def normalize_tuple(value, n, name): function
/aosp_15_r20/external/pytorch/torch/utils/
H A Dflop_counter.py552 def normalize_tuple(x): function