Home
last modified time | relevance | path

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

1234567

/aosp_15_r20/external/tensorflow/tensorflow/compiler/tests/
H A Dconv2d_test.py102 def testConv2D1x1Filter(self, data_format): argument
117 def testConv2D2x2Filter(self, data_format): argument
130 def testConv2D2x2Filter2x1Dilation(self, data_format): argument
143 def testConv2D1x2Filter(self, data_format): argument
158 def testConv2D2x2FilterStride2(self, data_format): argument
170 def testConv2D2x2FilterStride2Same(self, data_format): argument
183 def testConv2DEmptyDilation(self, data_format): argument
195 def testConv2D2x2FilterDilation(self, data_format): argument
207 def testConv2D1x2FilterDilation(self, data_format): argument
220 def testConv2DKernelSizeMatchesInputSizeDilation(self, data_format): argument
[all …]
H A Dfused_batchnorm_test.py45 exponential_avg_factor, data_format): argument
64 def _reference_grad(self, x, grad_y, scale, mean, var, epsilon, data_format): argument
86 def testInference(self, data_format): argument
129 def _testLearning(self, use_gradient_checker, data_format, argument
207 def testLearning(self, data_format, exponential_avg_factor): argument
211 def testLearningWithGradientChecker(self, data_format, argument
216 def testGradientTraining(self, data_format): argument
292 def testGradientInference(self, data_format): argument
H A Ddepthwise_conv_op_test.py32 data_format=None): argument
153 data_format="NHWC"): argument
355 data_format="NHWC"): argument
498 data_format="NHWC"): argument
632 data_format="NHWC"): argument
/aosp_15_r20/external/tensorflow/tensorflow/python/kernel_tests/nn_ops/
H A Dpooling_ops_test.py188 data_format, data_type, expected, use_gpu, v2, argument
282 data_format, expected, use_gpu, v2, argument
1159 data_format, argument
1237 data_format, argument
1305 def _testMaxPoolGradValidPadding1_1(self, data_format, use_gpu): argument
1319 def _testMaxPoolGradValidPadding2_1_6(self, data_format, use_gpu): argument
1333 def _testMaxPoolGradValidPadding2_1_7(self, data_format, use_gpu): argument
1347 def _testMaxPoolGradValidPadding1_2(self, data_format, use_gpu): argument
1361 def _testMaxPoolGradValidPadding2_2(self, data_format, use_gpu): argument
1375 def _testMaxPoolGradSamePadding1_1(self, data_format, use_gpu): argument
[all …]
H A Dconv_ops_test.py183 strides, padding, data_format, dtype, use_gpu): argument
244 def _SetupVal(data_format, use_gpu): argument
266 stride, dilation, padding, data_format, argument
792 strides, padding, data_format, dtype): argument
918 data_format, argument
966 def _GetVal(data_format, use_gpu): argument
1141 data_format, argument
1185 def _GetVal(data_format, use_gpu): argument
1318 padding, data_format, use_gpu, err): argument
1364 padding, data_format, use_gpu, err): argument
[all …]
H A Ddepthwise_conv_op_d9m_test.py38 data_format="NHWC", argument
81 data_format="NHWC", argument
112 data_format="NHWC", argument
/aosp_15_r20/external/tensorflow/tensorflow/python/ops/
H A Dnn_ops.py259 data_format=None, # pylint: disable=redefined-builtin argument
339 data_format=None, argument
408 def _conv1d(self, input, filter, strides, padding, data_format, name): argument
484 data_format, argument
576 data_format=None): argument
753 data_format=None, argument
1037 data_format=None, argument
1178 data_format=None, argument
1202 data_format=None, argument
1382 data_format=None, argument
[all …]
H A 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
H A Dnn_fused_batchnorm_d9m_test.py48 def _genParams(self, data_format, x_dtype, large_batch): argument
77 def testForward(self, data_format): argument
119 def testBackward(self, data_format): argument
/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/client/lib/
H A 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 …]
H A Dpooling.cc32 const TensorFormat& data_format) { in AvgPoolDivideByCountWithGeneralPadding()
76 const TensorFormat& data_format) { in ComputeSums()
92 const TensorFormat& data_format) { in MakeSpatialPaddingConfig()
114 const TensorFormat& data_format, bool counts_include_padding) { 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()
/aosp_15_r20/external/tensorflow/tensorflow/python/keras/legacy_tf_layers/
H A Dpooling.py46 padding='valid', data_format='channels_last', argument
62 padding='valid', data_format='channels_last', argument
119 padding='valid', data_format='channels_last', argument
135 padding='valid', data_format='channels_last', argument
196 padding='valid', data_format='channels_last', argument
209 padding='valid', data_format='channels_last', argument
272 padding='valid', data_format='channels_last', argument
285 padding='valid', data_format='channels_last', argument
350 padding='valid', data_format='channels_last', argument
363 padding='valid', data_format='channels_last', argument
[all …]
H A Dconvolutional.py85 data_format='channels_last', argument
126 data_format='channels_last', argument
292 data_format='channels_last', argument
333 data_format='channels_last', argument
507 data_format='channels_last', argument
548 data_format='channels_last', argument
724 data_format='channels_last', argument
840 data_format='channels_last', argument
890 data_format='channels_last', argument
1014 data_format='channels_last', argument
[all …]
/aosp_15_r20/external/tensorflow/tensorflow/python/keras/layers/
H A 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 …]
H A 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 …]
/aosp_15_r20/external/tensorflow/tensorflow/cc/gradients/
H A 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, StringPiece data_format, in FusedBatchNormGradAttrs()
433 std::string data_format; in BaseFusedBatchNormGrad() local
494 std::vector<Output>* grad_outputs) { in FusedBatchNormV3Grad()
[all …]
/aosp_15_r20/external/tensorflow/tensorflow/python/keras/
H A Dbackend.py3175 def resize_images(x, height_factor, width_factor, data_format, argument
3228 def resize_volumes(x, depth_factor, height_factor, width_factor, data_format): argument
3526 def spatial_2d_padding(x, padding=((1, 1), (1, 1)), data_format=None): argument
3559 def spatial_3d_padding(x, padding=((1, 1), (1, 1), (1, 1)), data_format=None): argument
5169 def _preprocess_conv1d_input(x, data_format): argument
5188 def _preprocess_conv2d_input(x, data_format, force_transpose=False): argument
5211 def _preprocess_conv3d_input(x, data_format): argument
5258 data_format=None, argument
5310 data_format=None, argument
5356 data_format=None, argument
[all …]
/aosp_15_r20/external/tensorflow/tensorflow/dtensor/mlir/expansions/
H A Dbias_add_spmd_expander.cc40 int get_c_dimension_idx(const Layout& layout, llvm::StringRef data_format) { in get_c_dimension_idx()
56 const llvm::StringRef data_format = bias_add_op.data_format(); in ExpandOp() local
124 llvm::StringRef data_format = bias_add_op.data_format(); in ComputeLayoutForward() local
161 llvm::StringRef data_format = bias_add_op.data_format(); in ComputeLayoutBackward() local
/aosp_15_r20/external/tensorflow/tensorflow/core/kernels/
H A Dbias_op.cc48 void GetBiasValueDims(const Tensor& value_tensor, TensorFormat data_format, in GetBiasValueDims()
93 string data_format; in BiasOp() local
163 string data_format; in BiasGradOp() local
242 string data_format; in BiasOp() local
335 BiasAddParams(const SpatialArray& in_shape, TensorFormat data_format, in BiasAddParams()
390 string data_format; in BiasGradOp() local
H A Davgpooling_op.cc60 string data_format; in AvgPoolingOp() local
138 string data_format; in AvgPoolingOp() local
264 string data_format; in AvgPoolingGradOp() local
430 string data_format; in AvgPoolingGradOp() local
517 string data_format; in AvgPoolingGradOpCustomGPUKernel() local
/aosp_15_r20/external/tensorflow/tensorflow/core/framework/
H A Dcommon_shape_fns.cc386 string data_format; in BiasAddShape() local
444 string data_format; in BiasAddGradShape() local
595 TensorFormat data_format; in Conv2DShapeImpl() local
777 string data_format; in Conv3DShape() local
890 TensorFormat data_format; in Conv2DBackpropInputShape() local
960 string data_format; in Conv2DBackpropFilterWithBiasShape() local
1009 TensorFormat data_format; in DepthwiseConv2DNativeShapeImpl() local
1111 TensorFormat data_format; in AvgPoolShape() local
1189 TensorFormat data_format; in FusedBatchNormShape() local
1239 TensorFormat data_format; in FusedBatchNormExShape() local
[all …]
/aosp_15_r20/external/tensorflow/tensorflow/python/kernel_tests/array_ops/
H A Ddepthtospace_op_test.py259 def depthToSpaceUsingTranspose(self, tensor, block_size, data_format): argument
280 block_size, data_format, use_gpu): argument
336 def _checkGrad(self, x, block_size, data_format): argument
358 def _compare(self, b, h, w, d, block_size, data_format): argument
H A Dspacetodepth_op_test.py210 def spaceToDepthUsingTranspose(self, tensor, block_size, data_format): argument
241 block_size, data_format, data_type, use_gpu): argument
320 def _checkGrad(self, x, block_size, data_format): argument
338 def _compare(self, b, h, w, d, block_size, data_format): argument
/aosp_15_r20/external/autotest/client/cros/multimedia/
H A Daudio_facade.py214 def playback(self, file_path, data_format, blocking=False, node_type=None, argument
260 def start_recording(self, data_format, node_type=None, block_size=None): argument
347 def start_listening(self, data_format): argument
580 def start(self, data_format, pin_device, block_size): argument
687 def start(self, data_format): argument
/aosp_15_r20/external/tensorflow/tensorflow/python/ops/parallel_for/
H A Dgradients_test.py215 def __init__(self, data_format): argument
260 def create_mnist_autobatch(batch_size, data_format, training): argument
276 def create_mnist_per_eg_grad(batch_size, data_format, training): argument
299 def create_mnist_batch_jacobian(batch_size, data_format, training): argument
309 def create_mnist_per_eg_jacobian(batch_size, data_format, training): argument

1234567