| /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
232 def _testDilationGradValidPadding_1x1x1(self, use_gpu): argument
[all …]
|
| D | pooling_ops_test.py | 43 def GetDeviceScope(self, use_gpu=False): argument
109 data_format, data_type, expected, use_gpu, v2): argument
186 data_format, expected, use_gpu, v2): argument
223 use_gpu, argument
242 def _testAvgPoolValidPadding(self, use_gpu): argument
253 def _testAvgPoolSamePadding(self, use_gpu): argument
264 def _testAvgPoolSamePaddingNonSquareWindow(self, use_gpu): argument
294 def _testAvgPoolSamePaddingNonSquareWindowMultiBatch(self, use_gpu): argument
318 def _testAvgPoolValidPaddingUnevenStride(self, use_gpu): argument
336 def _testAvgPoolSamePadding4(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 | depthwise_conv_op_test.py | 104 use_gpu, argument
245 expected, use_gpu): argument
358 use_gpu, argument
528 def _GetVal(use_gpu): argument
548 def _GetVal(use_gpu): argument
580 def _GetVal(use_gpu): argument
600 def _GetVal(use_gpu): argument
|
| D | dense_update_ops_test.py | 33 def _initAssignFetch(self, x, y, use_gpu=False): argument
43 def _initAssignAddFetch(self, x, y, use_gpu=False): argument
52 def _initAssignSubFetch(self, x, y, use_gpu=False): argument
|
| 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
531 def _RunAndVerifyResult(self, rank, use_gpu): argument
|
| D | cast_op_test.py | 57 def _cast(self, x, dtype, use_gpu=False): argument
62 def _test(self, x, dtype, use_gpu=False): argument
68 def _testTypes(self, x, use_gpu=False): argument
126 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 | 55 def _testXent(self, np_features, np_labels, use_gpu=False): argument
64 def _testXentWrapper(self, np_features, np_labels, dim=-1, use_gpu=False): argument
78 def _testSingleClass(self, use_gpu=False): argument
|
| D | parameterized_truncated_normal_op_test.py | 243 def parameterized_vs_naive(shape, num_iters, use_gpu=False): argument
268 def randn_sampler_switchover(shape, num_iters, use_gpu=False): argument
328 def _benchmarkParameterizedOpVsNaiveOp(self, use_gpu): argument
353 def _benchmarkRandnSampler(self, use_gpu): argument
|
| D | conv_ops_3d_test.py | 52 def _DtypesToTest(self, use_gpu): argument
64 padding, data_format, dtype, use_gpu): argument
124 use_gpu): argument
369 padding, test_input, data_format, use_gpu): argument
684 strides, dilations, padding, data_format, use_gpu, argument
|
| D | bias_op_test.py | 48 def _testBias(self, np_inputs, np_bias, use_gpu=False): argument
75 def _testBiasNCHW(self, np_inputs, np_bias, use_gpu): argument
141 def _testGradient(self, np_input, bias, dtype, data_format, use_gpu): argument
|
| D | cwise_ops_test.py | 222 def _compareBinary(self, x, y, np_func, tf_func, use_gpu=False): argument
233 def _not(self, x, use_gpu=False): argument
320 def _compare(self, c, x, y, use_gpu): argument
465 def _compare(self, c, x, y, use_gpu): argument
574 def _compare(self, x, y, use_gpu): argument
737 def _compare(self, x, use_gpu): argument
835 def _compareMake(self, real, imag, use_gpu): argument
855 def _compareRealImag(self, cplx, use_gpu): argument
884 def _compareAngle(self, cplx, use_gpu): argument
917 def _compareConj(self, cplx, use_gpu): argument
|
| D | denormal_test.py | 38 def _flushDenormalsTest(self, use_gpu, dtypes): argument
|
| D | softsign_op_test.py | 36 def _testSoftsign(self, np_features, use_gpu=False): argument
|
| D | softmax_op_test.py | 57 def _testSoftmax(self, np_features, dim=-1, log=False, use_gpu=False): argument
108 def _testOverflow(self, use_gpu=False): argument
|
| D | conv_ops_test.py | 165 def _DtypesToTest(self, use_gpu): argument
180 strides, padding, data_format, dtype, use_gpu): argument
240 def _SetupVal(data_format, use_gpu): argument
263 use_gpu): argument
711 use_gpu, argument
757 def _GetVal(data_format, use_gpu): argument
900 use_gpu, argument
943 def _GetVal(data_format, use_gpu): argument
1076 padding, data_format, use_gpu, err): argument
1122 padding, data_format, use_gpu, err): argument
[all …]
|
| /external/tensorflow/tensorflow/contrib/sparsemax/python/kernel_tests/ |
| D | sparsemax_test.py | 72 def _tf_sparsemax(self, z, dtype, use_gpu): argument
79 def _test_sparsemax_against_numpy(self, dtype, random, use_gpu): argument
90 def _test_sparsemax_of_nan(self, dtype, random, use_gpu): argument
103 def _test_sparsemax_of_inf(self, dtype, random, use_gpu): argument
130 def _test_sparsemax_of_zero(self, dtype, random, use_gpu): argument
140 def _test_sparsemax_of_to_inf(self, dtype, random, use_gpu): argument
160 def _test_constant_add(self, dtype, random, use_gpu): argument
172 def _test_permutation(self, dtype, random, use_gpu): argument
188 def _test_diffrence(self, dtype, random, use_gpu): argument
207 def _test_two_dimentional(self, dtype, random, use_gpu): argument
[all …]
|
| D | sparsemax_loss_test.py | 81 def _tf_sparsemax(self, z, dtype, use_gpu): argument
88 def _tf_sparsemax_loss(self, z, q, dtype, use_gpu): argument
99 def _test_sparsemax_loss_against_numpy(self, dtype, random, use_gpu): argument
112 def _test_sparsemax_loss_of_nan(self, dtype, random, use_gpu): argument
121 def _test_sparsemax_loss_of_inf(self, dtype, random, use_gpu): argument
148 def _test_constant_add(self, dtype, random, use_gpu): argument
167 def _test_sparsemax_loss_positive(self, dtype, random, use_gpu): argument
178 def _test_sparsemax_loss_zero(self, dtype, random, use_gpu): argument
197 def _test_gradient_against_estimate(self, dtype, random, use_gpu): argument
213 def _test_gradient_against_numpy(self, dtype, random, use_gpu): argument
|
| /external/tensorflow/tensorflow/contrib/rnn/python/kernel_tests/ |
| D | rnn_test.py | 44 use_gpu, argument
101 def _testStackBidirectionalRNN(self, use_gpu, use_shape): argument
153 def _testStackBidirectionalRNNStates(self, use_gpu): argument
224 use_gpu, argument
282 def _testStackBidirectionalDynamicRNN(self, use_gpu, use_shape, argument
336 def _testStackBidirectionalDynamicRNNStates(self, use_gpu): argument
|
| D | gru_ops_test.py | 339 use_gpu=False, argument
414 use_gpu=False, argument
471 use_gpu=False, argument
|
| D | benchmarking.py | 27 def device(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
72 def _testTypesForAdagrad(self, x, y, lr, grad, use_gpu=None): argument
93 use_gpu=None, argument
268 def _testTypesForAdam(self, var, m, v, grad, use_gpu): argument
|
| /external/tensorflow/tensorflow/python/kernel_tests/random/ |
| 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
263 def _Sampler(self, num, minv, maxv, dtype, use_gpu, seed=None): argument
|
| /external/tensorflow/tensorflow/python/ops/ |
| D | nn_fused_batchnorm_test.py | 61 use_gpu=True, argument
113 use_gpu=True, argument
186 use_gpu=True, argument
246 use_gpu=True, argument
|