Home
last modified time | relevance | path

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

123456

/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.py40 def _reference_training(self, x, scale, offset, epsilon, data_format): argument
52 def _reference_grad(self, x, grad_y, scale, mean, var, epsilon, data_format): argument
74 def testInference(self, data_format): argument
115 def _testLearning(self, use_gradient_checker, data_format): argument
174 def testLearning(self, data_format): argument
178 def testLearningWithGradientChecker(self, data_format): argument
182 def testGradientTraining(self, data_format): argument
252 def testGradientInference(self, data_format): argument
Ddepthwise_conv_op_test.py35 data_format=None): argument
130 data_format="NHWC"): argument
359 data_format="NHWC"): argument
/external/tensorflow/tensorflow/python/layers/
Dpooling.py48 padding='valid', data_format='channels_last', argument
66 padding='valid', data_format='channels_last', argument
119 padding='valid', data_format='channels_last', argument
137 padding='valid', data_format='channels_last', argument
194 padding='valid', data_format='channels_last', argument
209 padding='valid', data_format='channels_last', argument
268 padding='valid', data_format='channels_last', argument
283 padding='valid', data_format='channels_last', argument
344 padding='valid', data_format='channels_last', argument
359 padding='valid', data_format='channels_last', argument
[all …]
Dconvolutional.py84 data_format='channels_last', argument
127 data_format='channels_last', argument
283 data_format='channels_last', argument
326 data_format='channels_last', argument
490 data_format='channels_last', argument
533 data_format='channels_last', argument
699 data_format='channels_last', argument
811 data_format='channels_last', argument
863 data_format='channels_last', argument
983 data_format='channels_last', argument
[all …]
/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()
112 PrimitiveType dtype, const TensorFormat& data_format, in AvgPoolDivideByCount()
135 const TensorFormat& data_format) { in MaxPool()
150 const TensorFormat& data_format, in AvgPool()
175 const TensorFormat& data_format) { in MakeSpatialPadding()
196 const TensorFormat& data_format, in AvgPoolGrad()
/external/tensorflow/tensorflow/python/kernel_tests/
Dpooling_ops_test.py109 data_format, data_type, expected, use_gpu, v2): argument
186 data_format, expected, use_gpu, v2): argument
953 data_format, argument
1029 data_format, argument
1097 def _testMaxPoolGradValidPadding1_1(self, data_format, use_gpu): argument
1111 def _testMaxPoolGradValidPadding2_1_6(self, data_format, use_gpu): argument
1125 def _testMaxPoolGradValidPadding2_1_7(self, data_format, use_gpu): argument
1139 def _testMaxPoolGradValidPadding1_2(self, data_format, use_gpu): argument
1153 def _testMaxPoolGradValidPadding2_2(self, data_format, use_gpu): argument
1167 def _testMaxPoolGradSamePadding1_1(self, data_format, use_gpu): argument
[all …]
Ddepthtospace_op_test.py249 def depthToSpaceUsingTranspose(self, tensor, block_size, data_format): argument
270 block_size, data_format, use_gpu): argument
326 def _checkGrad(self, x, block_size, data_format): argument
348 def _compare(self, b, h, w, d, block_size, data_format): argument
Dspacetodepth_op_test.py236 def spaceToDepthUsingTranspose(self, tensor, block_size, data_format): argument
259 block_size, data_format, use_gpu): argument
311 def _checkGrad(self, x, block_size, data_format): argument
333 def _compare(self, b, h, w, d, block_size, data_format): argument
/external/tensorflow/tensorflow/core/kernels/
Dmkl_pooling_ops_common.cc227 TensorFormat data_format, in Init()
255 TensorFormat data_format, in Init()
270 TensorFormat data_format, in Init()
295 TensorFormat data_format) { in Init()
405 void ExtractMklOpParams(OpKernelContext* context, TensorFormat data_format, in ExtractMklOpParams()
/external/tensorflow/tensorflow/python/ops/
Dnn_ops.py84 data_format=None, # pylint: disable=redefined-builtin argument
162 data_format=None, argument
215 def _conv1d(self, input, filter, strides, padding, data_format, name): argument
242 data_format, argument
329 data_format=None): argument
506 data_format=None): argument
767 data_format=None, argument
902 data_format=None, argument
926 data_format=None, argument
1014 data_format=None): argument
[all …]
Dnn_fused_batchnorm_test.py45 def _inference_ref(self, x, scale, offset, mean, var, epsilon, data_format): argument
62 data_format='NHWC'): argument
95 def _training_ref(self, x, scale, offset, epsilon, data_format): argument
114 data_format='NHWC'): argument
187 data_format='NHWC', argument
247 data_format='NHWC', argument
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
/external/tensorflow/tensorflow/python/keras/layers/
Dpooling.py58 padding='valid', data_format='channels_last', argument
143 padding='valid', data_format='channels_last', **kwargs): argument
186 padding='valid', data_format='channels_last', **kwargs): argument
222 padding='valid', data_format=None, argument
323 data_format=None, argument
373 data_format=None, argument
409 padding='valid', data_format='channels_last', argument
519 data_format=None, argument
569 data_format=None, argument
580 def __init__(self, data_format='channels_last', **kwargs): argument
[all …]
Dconvolutional.py104 data_format=None, argument
339 data_format='channels_last', argument
455 data_format=None, argument
573 data_format=None, argument
701 data_format=None, argument
976 data_format=None, argument
1234 data_format=None, argument
1437 data_format=None, argument
1606 data_format=None, argument
1741 data_format=None, argument
[all …]
/external/tensorflow/tensorflow/contrib/fused_conv/python/ops/
Dfused_conv2d_bias_activation_benchmark.py33 padding, num_iters, data_format): argument
78 padding, num_iters, data_format): argument
135 num_iters, data_format): argument
/external/tensorflow/tensorflow/python/keras/preprocessing/
Dimage.py48 def array_to_img(x, data_format=None, scale=True, dtype=None): argument
78 def img_to_array(img, data_format=None, dtype=None): argument
107 data_format=None, argument
195 data_format=None, argument
266 data_format=None, argument
486 data_format=None, argument
/external/tensorflow/tensorflow/contrib/eager/python/examples/densenet/
Ddensenet.py41 def __init__(self, num_filters, data_format, bottleneck, weight_decay=1e-4, argument
92 def __init__(self, num_filters, data_format, argument
127 def __init__(self, num_layers, growth_rate, data_format, bottleneck, argument
174 output_classes, num_layers_in_each_block, data_format, argument
/external/autotest/client/cros/audio/
Daudio_test_data.py23 def __init__(self, data_format=None, path=None, frequencies=None, argument
61 def convert(self, data_format, volume_scale, path=None): argument
159 def GenerateAudioTestData(data_format, path, frequencies=None, argument
/external/tensorflow/tensorflow/contrib/eager/python/examples/resnet50/
Dresnet50.py46 def __init__(self, kernel_size, filters, stage, block, data_format): argument
109 data_format, argument
197 data_format, 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/contrib/layers/python/layers/
Dlayers.py82 data_format=DATA_FORMAT_NHWC, argument
131 data_format=DATA_FORMAT_NDHWC, argument
189 data_format=DATA_FORMAT_NHWC, argument
446 data_format=DATA_FORMAT_NHWC, argument
849 data_format=DATA_FORMAT_NHWC, argument
923 data_format=None, argument
1079 data_format=None, argument
1122 data_format=None, argument
1165 data_format=None, argument
1315 data_format=DATA_FORMAT_NHWC, argument
[all …]
/external/tensorflow/tensorflow/contrib/eager/python/examples/revnet/
Dblocks.py47 data_format="channels_first", argument
122 data_format="channels_first", argument
272 data_format="channels_first", argument
354 data_format="channels_first", argument
/external/autotest/client/cros/multimedia/
Daudio_facade_native.py194 def playback(self, file_path, data_format, blocking=False): argument
230 def start_recording(self, data_format): argument
276 def start_listening(self, data_format): argument
496 def start(self, data_format): argument
601 def start(self, data_format): argument

123456