Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/python/layers/
Dutils_test.py28 self.assertEqual('NCDHW', utils.convert_data_format('channels_first', 5))
29 self.assertEqual('NCHW', utils.convert_data_format('channels_first', 4))
30 self.assertEqual('NCW', utils.convert_data_format('channels_first', 3))
31 self.assertEqual('NHWC', utils.convert_data_format('channels_last', 4))
32 self.assertEqual('NWC', utils.convert_data_format('channels_last', 3))
33 self.assertEqual('NDHWC', utils.convert_data_format('channels_last', 5))
36 utils.convert_data_format('invalid', 2)
Dutils.py28 def convert_data_format(data_format, ndim): function
/external/tensorflow/tensorflow/python/keras/utils/
Dconv_utils_test.py58 self.assertEqual('NCDHW', conv_utils.convert_data_format(
60 self.assertEqual('NCHW', conv_utils.convert_data_format(
62 self.assertEqual('NCW', conv_utils.convert_data_format('channels_first', 3))
63 self.assertEqual('NHWC', conv_utils.convert_data_format('channels_last', 4))
64 self.assertEqual('NWC', conv_utils.convert_data_format('channels_last', 3))
65 self.assertEqual('NDHWC', conv_utils.convert_data_format(
69 conv_utils.convert_data_format('invalid', 2)
Dconv_utils.py28 def convert_data_format(data_format, ndim): function
/external/tensorflow/tensorflow/python/keras/layers/
Dconvolutional.py191 data_format=conv_utils.convert_data_format(self.data_format,
832 data_format=conv_utils.convert_data_format(self.data_format, ndim=4))
1102 data_format=conv_utils.convert_data_format(self.data_format, ndim=5),
1114 data_format=conv_utils.convert_data_format(self.data_format, ndim=4))
1504 data_format=conv_utils.convert_data_format(self.data_format, ndim=4))
1510 data_format=conv_utils.convert_data_format(self.data_format, ndim=4))
1658 data_format=conv_utils.convert_data_format(self.data_format, ndim=4))
1664 data_format=conv_utils.convert_data_format(self.data_format, ndim=4))
Dpooling.py248 data_format=conv_utils.convert_data_format(self.data_format, 4))
/external/tensorflow/tensorflow/contrib/model_pruning/python/layers/
Dcore_layers.py186 data_format=utils.convert_data_format(self.data_format, self.rank + 2))