| /external/tensorflow/tensorflow/python/kernel_tests/ |
| D | morphological_ops_test.py | 33 def _VerifyValues(self, image, kernel, strides, rates, padding, out, use_gpu): argument
58 def _testDilationValidPadding(self, use_gpu): argument
74 def _testDilationSamePadding(self, use_gpu): argument
90 def _testDilationSamePaddingDepth(self, use_gpu): argument
106 def _testDilationSamePaddingBatch(self, use_gpu): argument
122 def _testDilationValidPaddingNonSquareWindow(self, use_gpu): argument
138 def _testDilationSamePaddingRate(self, use_gpu): argument
158 def _testDilationValidPaddingUnevenStride(self, use_gpu): argument
186 padding, use_gpu): argument
233 def _testDilationGradValidPadding_1x1x1(self, use_gpu): argument
[all …]
|
| D | pooling_ops_test.py | 44 def GetDeviceScope(self, use_gpu=False): argument
119 data_format, data_type, expected, use_gpu, v2, argument
202 data_format, expected, use_gpu, v2, argument
251 use_gpu, argument
277 def _testAvgPoolValidPadding(self, use_gpu): argument
288 def _testAvgPoolEmpty(self, use_gpu): argument
299 def _testAvgPoolSamePadding(self, use_gpu): argument
310 def _testAvgPoolSamePaddingNonSquareWindow(self, use_gpu): argument
340 def _testAvgPoolSamePaddingNonSquareWindowMultiBatch(self, use_gpu): argument
364 def _testAvgPoolValidPaddingUnevenStride(self, use_gpu): argument
[all …]
|
| D | stack_ops_test.py | 36 def _testStackPushPop(self, use_gpu): argument
50 def _testStackPushPopSwap(self, use_gpu): argument
66 def _testStackWhileSwap(self, use_gpu): argument
104 def _testMultiStack(self, use_gpu): argument
124 def _testSameNameStacks(self, use_gpu): argument
148 def _testCloseStack(self, use_gpu): argument
160 def _testPushCloseStack(self, use_gpu): argument
178 def _testStackPushPop(self, use_gpu): argument
191 def _testStackPushPopSwap(self, use_gpu): argument
206 def _testMultiStack(self, use_gpu): argument
[all …]
|
| D | shape_ops_test.py | 53 def _compareShape(self, x, use_gpu=False): argument
64 def _compareShapeSparse(self, x_np, use_gpu=False): argument
73 def _compareShapeN(self, x, use_gpu=False): argument
85 def _compareRank(self, x, use_gpu=False): argument
93 def _compareRankSparse(self, x_np, use_gpu=False): argument
102 def _compareSize(self, x, use_gpu=False): argument
113 def _compareSizeSparse(self, x_np, use_gpu=False): argument
177 def _compareExpandDims(self, x, dim, use_gpu): argument
276 def _compareSqueeze(self, x, squeeze_dims, use_gpu): argument
535 def _RunAndVerifyResult(self, rank, use_gpu): argument
|
| D | dense_update_ops_test.py | 33 def _initAssignFetch(self, x, y, use_gpu): argument
43 def _initAssignAddFetch(self, x, y, use_gpu): argument
52 def _initAssignSubFetch(self, x, y, use_gpu): argument
|
| D | cumulative_logsumexp_test.py | 40 def _testLogSumExp(self, x, dtype=dtypes.float32, use_gpu=False, **kwargs): argument
49 def _testLogSumExpAllArgs(self, x, axis=0, use_gpu=False): argument
75 def _testGradient(self, x, use_gpu=False, **kwargs): argument
|
| D | cast_op_test.py | 55 def _cast(self, x, dtype, use_gpu=False): argument
61 def _test(self, x, dtype, use_gpu=False): argument
67 def _testTypes(self, x, use_gpu=False): argument
122 def _compare(self, x, dst_dtype, expected, use_gpu=False): argument
|
| D | neon_depthwise_conv_op_test.py | 95 use_gpu, argument
187 expected, use_gpu): argument
|
| D | xent_op_test.py | 60 use_gpu=False, argument
81 def _testXentWrapper(self, np_features, np_labels, dim=-1, use_gpu=False): argument
99 def _testSingleClass(self, use_gpu=False): argument
|
| D | conv_ops_3d_test.py | 57 def _DtypesToTest(self, use_gpu): argument
72 padding, data_format, dtype, use_gpu): argument
128 use_gpu): argument
465 padding, test_input, data_format, use_gpu): argument
780 strides, dilations, padding, data_format, use_gpu, argument
|
| D | depthwise_conv_op_test.py | 276 use_gpu, argument
481 expected, use_gpu): argument
594 use_gpu, argument
907 def _GetVal(use_gpu): argument
965 def _GetVal(use_gpu): argument
|
| D | cwise_ops_test.py | 229 def _compareBinary(self, x, y, np_func, tf_func, use_gpu=False): argument
240 def _not(self, x, use_gpu=False): argument
335 def _compare(self, fn, c, x, y, use_gpu): argument
640 def _compare(self, c, x, y, use_gpu): argument
752 def _compare(self, x, y, use_gpu): argument
952 def _compare(self, x, use_gpu): argument
1067 def _compareMake(self, real, imag, use_gpu): argument
1099 def _compareRealImag(self, cplx, use_gpu): argument
1128 def _compareAngle(self, cplx, use_gpu): argument
1170 def _compareConj(self, cplx, use_gpu): argument
|
| D | bias_op_base.py | 53 def _testBias(self, np_inputs, np_bias, use_gpu=False): argument
80 def _testBiasNCHW(self, np_inputs, np_bias, use_gpu): argument
206 def _testGradient(self, np_input, bias, dtype, data_format, use_gpu): argument
|
| D | conv_ops_test.py | 166 def _DtypesToTest(self, use_gpu): argument
184 strides, padding, data_format, dtype, use_gpu): argument
245 def _SetupVal(data_format, use_gpu): argument
268 use_gpu): argument
907 use_gpu, argument
954 def _GetVal(data_format, use_gpu): argument
1113 use_gpu, 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 …]
|
| D | reshape_op_test.py | 35 def _testReshape(self, x, y, use_gpu=False): argument
50 def _testZeroDimReshape(self, x, shape, expected, use_gpu=False): argument
|
| D | softmax_op_test.py | 64 use_gpu=False): argument
118 def _testOverflow(self, use_gpu=False): argument
|
| D | softsign_op_test.py | 36 def _testSoftsign(self, np_features, use_gpu=False): argument
|
| D | basic_gpu_test.py | 94 def _compare(self, x, np_func, tf_func, use_gpu): argument
108 def _testDtype(self, dtype, use_gpu): argument
|
| D | softplus_op_test.py | 39 def _testSoftplus(self, np_features, use_gpu=False): argument
|
| /external/tensorflow/tensorflow/python/training/ |
| D | training_ops_test.py | 52 def _testTypes(self, x, alpha, delta, use_gpu=None): argument
73 def _testTypesForAdagrad(self, x, y, lr, grad, use_gpu=None): argument
94 use_gpu=None, argument
139 use_gpu=None, argument
226 def _testTypesForSparseAdagrad(self, x, y, lr, grad, indices, use_gpu): argument
254 use_gpu, argument
417 def _testTypesForAdam(self, var, m, v, grad, use_gpu): argument
|
| /external/tensorflow/tensorflow/python/kernel_tests/random/ |
| D | parameterized_truncated_normal_op_test.py | 394 def parameterized_vs_naive(shape, num_iters, use_gpu=False): argument
419 def randn_sampler_switchover(shape, num_iters, use_gpu=False): argument
479 def _benchmarkParameterizedOpVsNaiveOp(self, use_gpu): argument
504 def _benchmarkRandnSampler(self, use_gpu): argument
|
| D | random_ops_test.py | 45 use_gpu, argument
65 def _Sampler(self, num, mu, sigma, dtype, use_gpu, seed=None): argument
162 def _Sampler(self, num, mu, sigma, dtype, use_gpu, seed=None): argument
264 def _Sampler(self, num, minv, maxv, dtype, use_gpu, seed=None): argument
|
| /external/tensorflow/tensorflow/python/ops/ |
| D | nn_fused_batchnorm_test.py | 65 use_gpu=True, argument
146 use_gpu=True, argument
227 use_gpu=True, argument
290 use_gpu=True, argument
|
| /external/tensorflow/tensorflow/python/kernel_tests/linalg/ |
| D | linear_operator_toeplitz_test.py | 47 def _constrain_devices_and_set_default(self, sess, use_gpu, force_gpu): argument
|
| D | linear_operator_circulant_test.py | 58 def _constrain_devices_and_set_default(self, sess, use_gpu, force_gpu): argument
410 def _constrain_devices_and_set_default(self, sess, use_gpu, force_gpu): argument
703 def _constrain_devices_and_set_default(self, sess, use_gpu, force_gpu): argument
|