Home
last modified time | relevance | path

Searched defs:data_format (Results 1 – 25 of 152) 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.py48 exponential_avg_factor, data_format): argument
67 def _reference_grad(self, x, grad_y, scale, mean, var, epsilon, data_format): argument
89 def testInference(self, data_format): argument
132 def _testLearning(self, use_gradient_checker, data_format, argument
210 def testLearning(self, data_format, exponential_avg_factor): argument
214 def testLearningWithGradientChecker(self, data_format, argument
219 def testGradientTraining(self, data_format): argument
289 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/legacy_tf_layers/
Dpooling.py49 padding='valid', data_format='channels_last', argument
64 padding='valid', data_format='channels_last', argument
120 padding='valid', data_format='channels_last', argument
135 padding='valid', data_format='channels_last', argument
195 padding='valid', data_format='channels_last', argument
207 padding='valid', data_format='channels_last', argument
269 padding='valid', data_format='channels_last', argument
281 padding='valid', data_format='channels_last', argument
345 padding='valid', data_format='channels_last', argument
357 padding='valid', data_format='channels_last', argument
[all …]
Dconvolutional.py88 data_format='channels_last', argument
128 data_format='channels_last', argument
293 data_format='channels_last', argument
333 data_format='channels_last', argument
506 data_format='channels_last', argument
546 data_format='channels_last', argument
721 data_format='channels_last', argument
836 data_format='channels_last', argument
885 data_format='channels_last', argument
1008 data_format='channels_last', argument
[all …]
/external/tensorflow/tensorflow/python/keras/layers/
Dlocal_test.py91 def test_locallyconnected_1d(self, data_format, padding, implementation): argument
121 def test_locallyconnected_1d_regularization(self, data_format, padding, argument
170 def test_locallyconnected_2d(self, data_format, padding, implementation): argument
203 def test_locallyconnected_2d_channels_first(self, data_format, padding, argument
228 def test_locallyconnected_2d_regularization(self, data_format, padding, argument
283 def test_locallyconnected_implementation(self, width, data_format): argument
406 def get_inputs(data_format, filters, height, num_samples, width): argument
443 data_format): argument
Dpooling.py58 padding='valid', data_format='channels_last', argument
192 padding='valid', data_format='channels_last', **kwargs): argument
238 padding='valid', data_format='channels_last', **kwargs): argument
274 padding='valid', data_format=None, argument
458 data_format=None, argument
511 data_format=None, argument
547 padding='valid', data_format='channels_last', argument
660 data_format=None, argument
713 data_format=None, argument
724 def __init__(self, data_format='channels_last', **kwargs): argument
[all …]
Dconvolutional.py119 data_format=None, argument
489 data_format='channels_last', argument
648 data_format=None, argument
795 data_format=None, argument
924 data_format=None, argument
1195 data_format=None, argument
1505 data_format=None, argument
1769 data_format=None, argument
2002 data_format=None, argument
2192 data_format=None, argument
[all …]
/external/tensorflow/tensorflow/python/kernel_tests/
Dpooling_ops_test.py119 data_format, data_type, expected, use_gpu, v2, argument
202 data_format, expected, use_gpu, v2, argument
1117 data_format, argument
1195 data_format, argument
1263 def _testMaxPoolGradValidPadding1_1(self, data_format, use_gpu): argument
1277 def _testMaxPoolGradValidPadding2_1_6(self, data_format, use_gpu): argument
1291 def _testMaxPoolGradValidPadding2_1_7(self, data_format, use_gpu): argument
1305 def _testMaxPoolGradValidPadding1_2(self, data_format, use_gpu): argument
1319 def _testMaxPoolGradValidPadding2_2(self, data_format, use_gpu): argument
1333 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.py184 strides, padding, data_format, dtype, use_gpu): argument
245 def _SetupVal(data_format, use_gpu): argument
267 stride, dilation, padding, data_format, argument
784 strides, padding, data_format, dtype): argument
906 data_format, argument
954 def _GetVal(data_format, use_gpu): argument
1112 data_format, argument
1156 def _GetVal(data_format, use_gpu): argument
1289 padding, data_format, use_gpu, err): argument
1335 padding, data_format, use_gpu, err): argument
[all …]
/external/tensorflow/tensorflow/python/ops/
Dnn_ops.py111 data_format=None, # pylint: disable=redefined-builtin argument
191 data_format=None, argument
257 def _conv1d(self, input, filter, strides, padding, data_format, name): argument
333 data_format, argument
422 data_format=None): argument
600 data_format=None, argument
873 data_format=None, argument
1009 data_format=None, argument
1033 data_format=None, argument
1211 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.py45 def _inference_ref(self, x, scale, offset, mean, var, epsilon, data_format): argument
67 data_format='NHWC'): argument
108 exponential_avg_factor, epsilon, data_format): argument
148 data_format='NHWC'): argument
229 data_format='NHWC', argument
292 data_format='NHWC', argument
/external/tensorflow/tensorflow/python/keras/applications/
Dimagenet_utils.py104 def preprocess_input(x, data_format=None, mode='caffe'): argument
169 def _preprocess_numpy_input(x, data_format, mode): argument
242 def _preprocess_symbolic_input(x, data_format, mode): argument
299 data_format, 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/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/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.py152 def array_to_img(x, data_format=None, scale=True, dtype=None): argument
195 def img_to_array(img, data_format=None, dtype=None): argument
238 data_format=None, argument
364 data_format=None, argument
435 data_format=None, argument
547 data_format='channels_last', argument
782 data_format=None, argument
/external/tensorflow/tensorflow/compiler/tf2xla/kernels/
Dbias_ops.cc32 string data_format; in BiasOp() local
81 string data_format; in BiasAddGradOp() local
/external/tensorflow/tensorflow/core/kernels/
Dconv_grad_shape_utils.cc99 TensorFormat data_format, ConvBackpropDimensions* dims) { in ConvBackpropComputeDimensionsV2()
160 Padding padding, TensorFormat data_format, in ConvBackpropComputeDimensions()
172 const TensorFormat& data_format, in Conv2DBackpropComputeInputShape()

1234567