Home
last modified time | relevance | path

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

12345

/external/tensorflow/tensorflow/core/kernels/
Dmkl_pooling_ops_common.cc31 TensorFormat data_format, in Init()
50 TensorFormat data_format, in Init()
65 TensorFormat data_format, in Init()
80 TensorFormat data_format) { in Init()
151 void ExtractMklOpParams(OpKernelContext* context, TensorFormat data_format, in ExtractMklOpParams()
Dbias_op.cc44 void GetBiasValueDims(const Tensor& value_tensor, TensorFormat data_format, in GetBiasValueDims()
88 string data_format; in BiasOp() local
208 string data_format; in BiasGradOp() local
312 string data_format; in BiasOp() local
371 string data_format; in BiasGradOp() local
Dpooling_ops_common.cc34 Padding padding, TensorFormat data_format, in PoolParameters()
145 Padding padding, TensorFormat data_format, const Tensor& tensor_in, in Compute()
242 Padding padding, TensorFormat data_format, const Tensor* tensor_in, in Compute()
/external/tensorflow/tensorflow/python/layers/
Dpooling.py54 padding='valid', data_format='channels_last', argument
120 padding='valid', data_format='channels_last', argument
134 padding='valid', data_format='channels_last', argument
187 padding='valid', data_format='channels_last', argument
201 padding='valid', data_format='channels_last', argument
260 padding='valid', data_format='channels_last', argument
329 padding='valid', data_format='channels_last', argument
340 padding='valid', data_format='channels_last', argument
399 padding='valid', data_format='channels_last', argument
410 padding='valid', data_format='channels_last', argument
[all …]
Dconvolutional.py91 data_format='channels_last', argument
281 data_format='channels_last', argument
322 data_format='channels_last', argument
479 data_format='channels_last', argument
520 data_format='channels_last', argument
685 data_format='channels_last', argument
726 data_format='channels_last', argument
899 data_format='channels_last', argument
1054 data_format='channels_last', argument
1203 data_format='channels_last', argument
[all …]
/external/tensorflow/tensorflow/python/kernel_tests/
Dpooling_ops_test.py96 data_format, data_type, expected, use_gpu, v2): argument
172 data_format, expected, use_gpu, v2): argument
887 data_format, argument
963 data_format, argument
1031 def _testMaxPoolGradValidPadding1_1(self, data_format, use_gpu): argument
1045 def _testMaxPoolGradValidPadding2_1_6(self, data_format, use_gpu): argument
1059 def _testMaxPoolGradValidPadding2_1_7(self, data_format, use_gpu): argument
1073 def _testMaxPoolGradValidPadding1_2(self, data_format, use_gpu): argument
1087 def _testMaxPoolGradValidPadding2_2(self, data_format, use_gpu): argument
1101 def _testMaxPoolGradSamePadding1_1(self, data_format, use_gpu): argument
[all …]
Ddepthtospace_op_test.py210 def depthToSpaceUsingTranspose(self, tensor, block_size, data_format): argument
231 block_size, data_format, use_gpu): argument
287 def _checkGrad(self, x, block_size, data_format): argument
309 def _compare(self, b, h, w, d, block_size, data_format): argument
Dspacetodepth_op_test.py205 def spaceToDepthUsingTranspose(self, tensor, block_size, data_format): argument
228 block_size, data_format, use_gpu): argument
280 def _checkGrad(self, x, block_size, data_format): argument
302 def _compare(self, b, h, w, d, block_size, data_format): argument
Dconv_ops_test.py169 strides, padding, data_format, dtype, use_gpu): argument
233 def _SetupVal(data_format, use_gpu): argument
255 stride, dilation, padding, data_format, argument
521 strides, padding, expected, data_format, argument
559 def _GetVal(data_format, use_gpu): argument
695 strides, padding, expected, data_format, argument
735 def _GetVal(data_format, use_gpu): argument
868 padding, data_format, use_gpu, err): argument
922 padding, data_format, use_gpu, err): argument
1117 stride_cols, padding, test_input, data_format, argument
[all …]
/external/tensorflow/tensorflow/python/keras/_impl/keras/applications/
Dimagenet_utils.py38 def _preprocess_numpy_input(x, data_format, mode): argument
109 def _preprocess_symbolic_input(x, data_format, mode): argument
169 def preprocess_input(x, data_format=None, mode='caffe'): argument
251 data_format, argument
/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/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/python/ops/
Dnn_ops.py54 data_format=None, # pylint: disable=redefined-builtin argument
132 data_format=None, argument
187 def _conv1d(self, input, filter, strides, padding, data_format, name): argument
216 data_format=None): argument
393 data_format=None): argument
654 data_format=None): argument
808 data_format=None): argument
881 data_format=None): argument
1189 data_format="NHWC", argument
1415 data_format="NDHWC", argument
[all …]
Dnn_fused_batchnorm_test.py44 def _inference_ref(self, x, scale, offset, mean, var, epsilon, data_format): argument
61 data_format='NHWC'): argument
94 def _training_ref(self, x, scale, offset, epsilon, data_format): argument
113 data_format='NHWC'): argument
186 data_format='NHWC', argument
246 data_format='NHWC', argument
/external/tensorflow/tensorflow/compiler/tf2xla/kernels/
Dbias_ops.cc31 string data_format; in BiasOp() local
78 string data_format; in BiasAddGradOp() local
/external/tensorflow/tensorflow/contrib/eager/python/examples/resnet50/
Dresnet50.py45 def __init__(self, kernel_size, filters, stage, block, data_format): argument
115 data_format, argument
208 data_format, argument
/external/tensorflow/tensorflow/python/keras/_impl/keras/layers/
Dconvolutional.py237 data_format=None, argument
378 data_format=None, argument
520 data_format=None, argument
670 data_format=None, argument
788 data_format=None, argument
945 data_format=None, argument
1099 def __init__(self, size=(2, 2), data_format=None, **kwargs): argument
1167 def __init__(self, size=(2, 2, 2), data_format=None, **kwargs): argument
1289 def __init__(self, padding=(1, 1), data_format=None, **kwargs): argument
1393 def __init__(self, padding=(1, 1, 1), data_format=None, **kwargs): argument
[all …]
Dpooling.py143 data_format=None, argument
209 data_format=None, argument
271 data_format=None, argument
333 data_format=None, argument
406 def __init__(self, data_format=None, **kwargs): argument
504 def __init__(self, data_format=None, **kwargs): argument
/external/tensorflow/tensorflow/compiler/tests/
Dfused_batchnorm_test.py33 def _reference_training(self, x, scale, offset, epsilon, data_format): argument
45 def _reference_grad(self, x, grad_y, scale, mean, var, epsilon, data_format): argument
Dunary_ops_test.py36 def nhwc_to_format(x, data_format): argument
687 def make_op(data_format): argument
730 def make_op(data_format): argument
/external/tensorflow/tensorflow/contrib/eager/python/examples/mnist/
Dmnist_graph_test.py25 def data_format(): function
/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
1183 data_format=DATA_FORMAT_NHWC, argument
1299 data_format=DATA_FORMAT_NDHWC, argument
1808 data_format='channels_last', argument
1978 data_format='channels_last', argument
[all …]
/external/tensorflow/tensorflow/contrib/eager/python/examples/gan/
Dmnist.py44 def __init__(self, data_format): argument
102 def __init__(self, data_format): argument
/external/tensorflow/tensorflow/cc/gradients/
Dnn_grad.cc109 string data_format; in BiasAddGradHelper() local
124 string data_format; in Conv2DGrad() local
154 string data_format; in MaxPoolGradHelper() local
178 string data_format; in MaxPoolGradV2Helper() local
/external/tensorflow/tensorflow/python/keras/_impl/keras/
Dbackend.py240 def set_image_data_format(data_format): argument
2192 def resize_images(x, height_factor, width_factor, data_format): argument
2237 def resize_volumes(x, depth_factor, height_factor, width_factor, data_format): argument
2462 def spatial_2d_padding(x, padding=((1, 1), (1, 1)), data_format=None): argument
2493 def spatial_3d_padding(x, padding=((1, 1), (1, 1), (1, 1)), data_format=None): argument
3503 def _preprocess_conv1d_input(x, data_format): argument
3522 def _preprocess_conv2d_input(x, data_format): argument
3541 def _preprocess_conv3d_input(x, data_format): argument
3586 data_format=None, argument
3636 data_format=None, argument
[all …]

12345