Home
last modified time | relevance | path

Searched defs:data_format (Results 1 – 25 of 155) sorted by relevance

1234567

/external/tensorflow/tensorflow/compiler/tests/
Dconv2d_test.py106 def testConv2D1x1Filter(self, data_format): argument
121 def testConv2D2x2Filter(self, data_format): argument
134 def testConv2D2x2Filter2x1Dilation(self, data_format): argument
147 def testConv2D1x2Filter(self, data_format): argument
162 def testConv2D2x2FilterStride2(self, data_format): argument
174 def testConv2D2x2FilterStride2Same(self, data_format): argument
187 def testConv2DEmptyDilation(self, data_format): argument
199 def testConv2D2x2FilterDilation(self, data_format): argument
211 def testConv2D1x2FilterDilation(self, data_format): argument
224 def testConv2DKernelSizeMatchesInputSizeDilation(self, data_format): argument
[all …]
Dfused_batchnorm_test.py49 exponential_avg_factor, data_format): argument
68 def _reference_grad(self, x, grad_y, scale, mean, var, epsilon, data_format): argument
90 def testInference(self, data_format): argument
133 def _testLearning(self, use_gradient_checker, data_format, argument
211 def testLearning(self, data_format, exponential_avg_factor): argument
215 def testLearningWithGradientChecker(self, data_format, argument
220 def testGradientTraining(self, data_format): argument
296 def testGradientInference(self, data_format): argument
Ddepthwise_conv_op_test.py36 data_format=None): argument
157 data_format="NHWC"): argument
355 data_format="NHWC"): argument
498 data_format="NHWC"): argument
628 data_format="NHWC"): argument
/external/tensorflow/tensorflow/compiler/xla/client/lib/
Dpooling_test.cc37 const xla::TensorFormat& data_format) { in MakeGeneralPadding()
50 const xla::TensorFormat& data_format) { in ExpandWithBatchAndFeatureDimensions()
70 auto data_format = MakeNCHWFormat(2); in XLA_TEST_F() local
83 auto data_format = MakeNCHWFormat(2); in XLA_TEST_F() local
96 auto data_format = MakeNCHWFormat(2); in XLA_TEST_F() local
110 auto data_format = MakeNCHWFormat(2); in XLA_TEST_F() local
126 auto data_format = MakeNCHWFormat(2); in XLA_TEST_F() local
142 auto data_format = MakeNCHWFormat(2); in XLA_TEST_F() local
160 auto data_format = MakeNCHWFormat(2); in XLA_TEST_F() local
175 auto data_format = MakeNCHWFormat(2); in XLA_TEST_F() local
[all …]
Dpooling.cc32 const TensorFormat& data_format) { in AvgPoolDivideByCountWithGeneralPadding()
76 const TensorFormat& data_format) { in ComputeSums()
92 const TensorFormat& data_format) { in MakeSpatialPaddingConfig()
113 PrimitiveType dtype, const TensorFormat& data_format, in AvgPoolDivideByCount()
136 const TensorFormat& data_format) { in MaxPool()
151 const TensorFormat& data_format, in AvgPool()
176 const TensorFormat& data_format) { in MakeSpatialPadding()
197 const TensorFormat& data_format, in AvgPoolGrad()
/external/tensorflow/tensorflow/python/keras/layers/
Dlocal_test.py89 def test_locallyconnected_1d(self, data_format, padding, implementation): argument
119 def test_locallyconnected_1d_regularization(self, data_format, padding, argument
168 def test_locallyconnected_2d(self, data_format, padding, implementation): argument
201 def test_locallyconnected_2d_channels_first(self, data_format, padding, argument
226 def test_locallyconnected_2d_regularization(self, data_format, padding, argument
281 def test_locallyconnected_implementation(self, width, data_format): argument
385 def test_locallyconnected_save(self, width, data_format): argument
497 def get_inputs(data_format, filters, height, num_samples, width): argument
534 data_format): argument
567 num_classes, data_format): argument
Dpooling.py54 padding='valid', data_format='channels_last', argument
188 padding='valid', data_format='channels_last', **kwargs): argument
301 padding='valid', data_format='channels_last', **kwargs): argument
337 padding='valid', data_format=None, argument
525 data_format=None, argument
639 data_format=None, argument
675 padding='valid', data_format='channels_last', argument
806 data_format=None, argument
877 data_format=None, argument
888 def __init__(self, data_format='channels_last', keepdims=False, **kwargs): argument
[all …]
Dconvolutional.py114 data_format=None, argument
491 data_format='channels_last', argument
651 data_format=None, argument
798 data_format=None, argument
927 data_format=None, argument
1198 data_format=None, argument
1508 data_format=None, argument
1772 data_format=None, argument
2005 data_format=None, argument
2196 data_format=None, argument
[all …]
/external/tensorflow/tensorflow/python/keras/legacy_tf_layers/
Dpooling.py50 padding='valid', data_format='channels_last', argument
66 padding='valid', data_format='channels_last', argument
123 padding='valid', data_format='channels_last', argument
139 padding='valid', data_format='channels_last', argument
200 padding='valid', data_format='channels_last', argument
213 padding='valid', data_format='channels_last', argument
276 padding='valid', data_format='channels_last', argument
289 padding='valid', data_format='channels_last', argument
354 padding='valid', data_format='channels_last', argument
367 padding='valid', data_format='channels_last', argument
[all …]
Dconvolutional.py89 data_format='channels_last', argument
130 data_format='channels_last', argument
296 data_format='channels_last', argument
337 data_format='channels_last', argument
511 data_format='channels_last', argument
552 data_format='channels_last', argument
728 data_format='channels_last', argument
844 data_format='channels_last', argument
894 data_format='channels_last', argument
1018 data_format='channels_last', argument
[all …]
/external/tensorflow/tensorflow/python/kernel_tests/
Dpooling_ops_test.py191 data_format, data_type, expected, use_gpu, v2, argument
285 data_format, expected, use_gpu, v2, argument
1137 data_format, argument
1215 data_format, argument
1283 def _testMaxPoolGradValidPadding1_1(self, data_format, use_gpu): argument
1297 def _testMaxPoolGradValidPadding2_1_6(self, data_format, use_gpu): argument
1311 def _testMaxPoolGradValidPadding2_1_7(self, data_format, use_gpu): argument
1325 def _testMaxPoolGradValidPadding1_2(self, data_format, use_gpu): argument
1339 def _testMaxPoolGradValidPadding2_2(self, data_format, use_gpu): argument
1353 def _testMaxPoolGradSamePadding1_1(self, data_format, use_gpu): argument
[all …]
Ddepthtospace_op_test.py263 def depthToSpaceUsingTranspose(self, tensor, block_size, data_format): argument
284 block_size, data_format, use_gpu): argument
340 def _checkGrad(self, x, block_size, data_format): argument
362 def _compare(self, b, h, w, d, block_size, data_format): argument
Dspacetodepth_op_test.py230 def spaceToDepthUsingTranspose(self, tensor, block_size, data_format): argument
261 block_size, data_format, data_type, use_gpu): argument
340 def _checkGrad(self, x, block_size, data_format): argument
358 def _compare(self, b, h, w, d, block_size, data_format): argument
Dconv_ops_test.py186 strides, padding, data_format, dtype, use_gpu): argument
247 def _SetupVal(data_format, use_gpu): argument
269 stride, dilation, padding, data_format, argument
786 strides, padding, data_format, dtype): argument
912 data_format, argument
960 def _GetVal(data_format, use_gpu): argument
1118 data_format, argument
1162 def _GetVal(data_format, use_gpu): argument
1295 padding, data_format, use_gpu, err): argument
1341 padding, data_format, use_gpu, err): argument
[all …]
/external/tensorflow/tensorflow/python/ops/
Dnn_ops.py231 data_format=None, # pylint: disable=redefined-builtin argument
311 data_format=None, argument
377 def _conv1d(self, input, filter, strides, padding, data_format, name): argument
453 data_format, argument
542 data_format=None): argument
720 data_format=None, argument
993 data_format=None, argument
1129 data_format=None, argument
1153 data_format=None, argument
1331 data_format=None, argument
[all …]
Dconv2d_benchmark.py44 def build_graph(device, dtype, data_format, input_shape, filter_shape, strides, argument
97 def _run_graph(self, device, dtype, data_format, input_shape, filter_shape, argument
Dnn_fused_batchnorm_test.py46 def _inference_ref(self, x, scale, offset, mean, var, epsilon, data_format): argument
68 data_format='NHWC'): argument
109 exponential_avg_factor, epsilon, data_format): argument
149 data_format='NHWC'): argument
230 data_format='NHWC', argument
293 data_format='NHWC', argument
/external/tensorflow/tensorflow/python/keras/applications/
Dimagenet_utils.py101 def preprocess_input(x, data_format=None, mode='caffe'): argument
166 def _preprocess_numpy_input(x, data_format, mode): argument
239 def _preprocess_symbolic_input(x, data_format, mode): argument
299 data_format, argument
/external/tensorflow/tensorflow/cc/gradients/
Dnn_grad.cc202 string data_format; in BiasAddGradHelper() local
216 string data_format; in Conv2DGrad() local
243 string data_format; in MaxPoolGradHelper() local
263 string data_format; in MaxPoolGradV2Helper() local
284 string data_format; in MaxPool3DGradHelper() local
305 string data_format; in AvgPoolGradHelper() local
326 string data_format; in AvgPool3DGradHelper() local
399 T FusedBatchNormGradAttrs(float epsilon, std::string data_format, in FusedBatchNormGradAttrs()
433 std::string data_format; in BaseFusedBatchNormGrad() local
494 std::vector<Output>* grad_outputs) { in FusedBatchNormV3Grad()
[all …]
/external/autotest/client/cros/audio/
Daudio_test_data.py24 data_format=None, argument
62 def convert(self, data_format, volume_scale, path=None): argument
165 data_format=None, argument
/external/tensorflow/tensorflow/python/eager/benchmarks/resnet50/
Dresnet50_test_util.py32 def random_batch(batch_size, data_format, seed=None): argument
50 def report(benchmark, label, start, num_iters, device, batch_size, data_format, argument
Dresnet50.py46 def __init__(self, kernel_size, filters, stage, block, data_format): argument
109 data_format, argument
201 data_format, argument
/external/tensorflow/tensorflow/core/kernels/mkl/
Dmkl_pooling_ops_common.cc228 TensorFormat data_format, in Init()
255 TensorFormat data_format, in Init()
280 TensorFormat data_format) { in Init()
/external/tensorflow/tensorflow/python/keras/preprocessing/
Dimage.py162 def array_to_img(x, data_format=None, scale=True, dtype=None): argument
206 def img_to_array(img, data_format=None, dtype=None): argument
250 data_format=None, argument
378 data_format=None, argument
449 data_format=None, argument
563 data_format='channels_last', argument
798 data_format=None, argument
/external/tensorflow/tensorflow/core/kernels/
Dconv_grad_shape_utils.cc100 TensorFormat data_format, ConvBackpropDimensions* dims) { in ConvBackpropComputeDimensionsV2()
165 Padding padding, TensorFormat data_format, in ConvBackpropComputeDimensions()
177 const TensorFormat& data_format, in Conv2DBackpropComputeInputShape()

1234567