| /external/tensorflow/tensorflow/compiler/tests/ |
| D | conv2d_test.py | 102 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 …]
|
| D | fused_batchnorm_test.py | 45 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
|
| D | depthwise_conv_op_test.py | 32 data_format=None): argument
153 data_format="NHWC"): argument
355 data_format="NHWC"): argument
498 data_format="NHWC"): argument
632 data_format="NHWC"): argument
|
| /external/tensorflow/tensorflow/compiler/xla/client/lib/ |
| D | pooling_test.cc | 37 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 …]
|
| D | pooling.cc | 32 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()
|
| /external/tensorflow/tensorflow/python/keras/legacy_tf_layers/ |
| D | pooling.py | 46 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 …]
|
| D | convolutional.py | 85 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 …]
|
| /external/tensorflow/tensorflow/python/kernel_tests/nn_ops/ |
| D | pooling_ops_test.py | 188 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 …]
|
| D | depthwise_conv_op_d9m_test.py | 38 data_format="NHWC", argument
81 data_format="NHWC", argument
112 data_format="NHWC", argument
|
| /external/tensorflow/tensorflow/python/ops/ |
| D | nn_ops.py | 259 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 …]
|
| D | nn_fused_batchnorm_d9m_test.py | 48 def _genParams(self, data_format, x_dtype, large_batch): argument
77 def testForward(self, data_format): argument
119 def testBackward(self, data_format): argument
|
| D | conv2d_benchmark.py | 40 def build_graph(device, dtype, data_format, input_shape, filter_shape, strides, argument
93 def _run_graph(self, device, dtype, data_format, input_shape, filter_shape, argument
|
| D | nn_fused_batchnorm_test.py | 45 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/layers/ |
| D | pooling.py | 54 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 …]
|
| D | convolutional.py | 114 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/dtensor/mlir/expansions/ |
| D | bias_add_spmd_expander.cc | 40 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
|
| /external/tensorflow/tensorflow/cc/gradients/ |
| D | nn_grad.cc | 202 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 …]
|
| /external/tensorflow/tensorflow/python/eager/benchmarks/resnet50/ |
| D | resnet50_test_util.py | 28 def random_batch(batch_size, data_format, seed=None): argument
46 def report(benchmark, label, start, num_iters, device, batch_size, data_format, argument
|
| D | resnet50.py | 42 def __init__(self, kernel_size, filters, stage, block, data_format): argument
105 data_format, argument
197 data_format, argument
|
| /external/autotest/client/cros/audio/ |
| D | audio_test_data.py | 24 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/ |
| D | mkl_pooling_ops_common.cc | 232 TensorFormat data_format, in Init()
259 TensorFormat data_format, in Init()
284 TensorFormat data_format) { in Init()
|
| /external/tensorflow/tensorflow/tools/tfg_graph_transforms/ |
| D | tfg_graph_transforms_main.cc | 61 llvm::cl::opt<DataFormat> data_format( variable
131 DataFormat data_format, const std::string& input_file, in ImportModel()
155 DataFormat data_format, in ExportTFGModule()
|
| /external/tensorflow/tensorflow/python/kernel_tests/array_ops/ |
| D | depthtospace_op_test.py | 259 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
|
| D | spacetodepth_op_test.py | 210 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
|
| /external/tensorflow/tensorflow/core/kernels/ |
| D | bias_op.cc | 48 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
|