| /external/tensorflow/tensorflow/python/kernel_tests/nn_ops/ |
| D | morphological_ops_test.py | 31 def _VerifyValues(self, image, kernel, strides, rates, padding, out, use_gpu): argument
56 def _testDilationValidPadding(self, use_gpu): argument
72 def _testDilationSamePadding(self, use_gpu): argument
88 def _testDilationSamePaddingDepth(self, use_gpu): argument
104 def _testDilationSamePaddingBatch(self, use_gpu): argument
120 def _testDilationValidPaddingNonSquareWindow(self, use_gpu): argument
136 def _testDilationSamePaddingRate(self, use_gpu): argument
156 def _testDilationValidPaddingUnevenStride(self, use_gpu): argument
184 padding, use_gpu): argument
231 def _testDilationGradValidPadding_1x1x1(self, use_gpu): argument
[all …]
|
| D | pooling_ops_test.py | 42 def GetDeviceScope(self, use_gpu=False): argument
188 data_format, data_type, expected, use_gpu, v2, argument
282 data_format, expected, use_gpu, v2, argument
331 use_gpu, argument
878 use_gpu=False): argument
1160 use_gpu, argument
1238 use_gpu, 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
[all …]
|
| D | conv_ops_3d_test.py | 54 def _DtypesToTest(self, use_gpu): argument
69 padding, data_format, dtype, use_gpu): argument
125 use_gpu): argument
472 padding, test_input, data_format, use_gpu): argument
787 strides, dilations, padding, data_format, use_gpu, argument
|
| D | depthwise_conv_op_base.py | 305 use_gpu, argument
531 expected, use_gpu): argument
644 use_gpu, argument
962 def _GetVal(use_gpu): argument
1025 def _GetVal(use_gpu, data_format="NHWC"): argument
|
| D | bias_op_base.py | 49 def _testBias(self, np_inputs, np_bias, use_gpu=False): argument
76 def _testBiasNCHW(self, np_inputs, np_bias, use_gpu): argument
202 def _testGradient(self, np_input, bias, dtype, data_format, use_gpu): argument
|
| D | conv_ops_test.py | 163 def _DtypesToTest(self, use_gpu): argument
183 strides, padding, data_format, dtype, use_gpu): argument
244 def _SetupVal(data_format, use_gpu): argument
267 use_gpu): argument
919 use_gpu, argument
966 def _GetVal(data_format, use_gpu): argument
1142 use_gpu, 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 …]
|
| D | softmax_op_test.py | 60 use_gpu=False): argument
114 def _testOverflow(self, use_gpu=False): argument
|
| D | softsign_op_test.py | 32 def _testSoftsign(self, np_features, use_gpu=False): argument
|
| D | softplus_op_test.py | 35 def _testSoftplus(self, np_features, use_gpu=False): argument
|
| D | rnn_test.py | 450 def static_vs_dynamic_rnn_benchmark(batch_size, max_time, num_units, use_gpu): argument
513 use_gpu): argument
581 use_gpu): argument
|
| D | pooling_ops_3d_test.py | 52 data_format, expected, use_gpu): argument
256 use_gpu): argument
|
| /external/tensorflow/tensorflow/python/kernel_tests/data_structures/ |
| D | stack_ops_test.py | 32 def _testStackPushPop(self, use_gpu): argument
46 def _testStackPushPopSwap(self, use_gpu): argument
62 def _testStackWhileSwap(self, use_gpu): argument
100 def _testMultiStack(self, use_gpu): argument
120 def _testSameNameStacks(self, use_gpu): argument
144 def _testCloseStack(self, use_gpu): argument
156 def _testPushCloseStack(self, use_gpu): argument
174 def _testStackPushPop(self, use_gpu): argument
187 def _testStackPushPopSwap(self, use_gpu): argument
202 def _testMultiStack(self, use_gpu): argument
[all …]
|
| /external/tensorflow/tensorflow/python/kernel_tests/variables/ |
| D | dense_update_ops_test.py | 27 def _initAssignFetch(self, x, y, use_gpu): argument
37 def _initAssignAddFetch(self, x, y, use_gpu): argument
46 def _initAssignSubFetch(self, x, y, use_gpu): argument
|
| /external/tensorflow/tensorflow/python/kernel_tests/array_ops/ |
| D | shape_ops_test.py | 49 def _compareShape(self, x, use_gpu=False): argument
60 def _compareShapeSparse(self, x_np, use_gpu=False): argument
69 def _compareShapeN(self, x, use_gpu=False): argument
81 def _compareRank(self, x, use_gpu=False): argument
89 def _compareRankSparse(self, x_np, use_gpu=False): argument
98 def _compareSize(self, x, use_gpu=False): argument
109 def _compareSizeSparse(self, x_np, use_gpu=False): argument
173 def _compareExpandDims(self, x, dim, use_gpu): argument
272 def _compareSqueeze(self, x, squeeze_dims, use_gpu): argument
531 def _RunAndVerifyResult(self, rank, use_gpu): argument
|
| D | cast_op_test.py | 53 def _cast(self, x, dtype, use_gpu=False): argument
59 def _test(self, x, dtype, use_gpu=False): argument
65 def _testTypes(self, x, use_gpu=False): argument
120 def _compare(self, x, dst_dtype, expected, use_gpu=False): argument
|
| D | reshape_op_test.py | 33 def _testReshape(self, x, y, use_gpu=False): argument
48 def _testZeroDimReshape(self, x, shape, expected, use_gpu=False): argument
|
| /external/tensorflow/tensorflow/python/training/ |
| D | training_ops_test.py | 48 def _testTypes(self, x, alpha, delta, use_gpu=None): argument
69 def _testTypesForAdagrad(self, x, y, lr, grad, use_gpu=None): argument
90 use_gpu=None, argument
135 use_gpu=None, argument
222 def _testTypesForSparseAdagrad(self, x, y, lr, grad, indices, use_gpu): argument
250 use_gpu, argument
413 def _testTypesForAdam(self, var, m, v, grad, use_gpu): argument
|
| /external/tensorflow/tensorflow/python/kernel_tests/math_ops/ |
| D | cumulative_logsumexp_test.py | 36 def _testLogSumExp(self, x, dtype=dtypes.float32, use_gpu=False, **kwargs): argument
45 def _testLogSumExpAllArgs(self, x, axis=0, use_gpu=False): argument
71 def _testGradient(self, x, use_gpu=False, **kwargs): argument
|
| D | cwise_ops_test.py | 223 def _compareBinary(self, x, y, np_func, tf_func, use_gpu=False): argument
234 def _not(self, x, use_gpu=False): argument
329 def _compare(self, fn, c, x, y, use_gpu): argument
634 def _compare(self, c, x, y, use_gpu): argument
747 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 | basic_gpu_test.py | 89 def _compare(self, x, np_func, tf_func, use_gpu): argument
103 def _testDtype(self, dtype, use_gpu): argument
|
| /external/tensorflow/tensorflow/python/kernel_tests/random/ |
| D | parameterized_truncated_normal_op_test.py | 412 def parameterized_vs_naive(shape, num_iters, use_gpu=False): argument
437 def randn_sampler_switchover(shape, num_iters, use_gpu=False): argument
497 def _benchmarkParameterizedOpVsNaiveOp(self, use_gpu): argument
522 def _benchmarkRandnSampler(self, use_gpu): argument
|
| D | random_ops_test.py | 43 use_gpu, argument
63 def _Sampler(self, num, mu, sigma, dtype, use_gpu, seed=None): argument
161 def _Sampler(self, num, mu, sigma, dtype, use_gpu, seed=None): argument
276 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/sparse_ops/ |
| D | sparse_xent_op_test.py | 71 def sparse_vs_dense_xent_benchmark(batch_size, num_entries, use_gpu): argument
|
| /external/tensorflow/tensorflow/python/kernel_tests/linalg/ |
| D | linear_operator_toeplitz_test.py | 43 def _constrain_devices_and_set_default(self, sess, use_gpu, force_gpu): argument
|