Home
last modified time | relevance | path

Searched defs:use_gpu (Results 1 – 25 of 51) sorted by relevance

123

/external/tensorflow/tensorflow/python/kernel_tests/
Dmorphological_ops_test.py33 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 …]
Dpooling_ops_test.py44 def GetDeviceScope(self, use_gpu=False): argument
110 data_format, data_type, expected, use_gpu, v2): argument
187 data_format, expected, use_gpu, v2): argument
226 use_gpu, argument
245 def _testAvgPoolValidPadding(self, use_gpu): argument
256 def _testAvgPoolEmpty(self, use_gpu): argument
267 def _testAvgPoolSamePadding(self, use_gpu): argument
278 def _testAvgPoolSamePaddingNonSquareWindow(self, use_gpu): argument
308 def _testAvgPoolSamePaddingNonSquareWindowMultiBatch(self, use_gpu): argument
332 def _testAvgPoolValidPaddingUnevenStride(self, use_gpu): argument
[all …]
Dstack_ops_test.py36 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 …]
Ddepthwise_conv_op_test.py104 use_gpu, argument
273 expected, use_gpu): argument
386 use_gpu, argument
629 def _GetVal(use_gpu): argument
649 def _GetVal(use_gpu): argument
685 def _GetVal(use_gpu): argument
705 def _GetVal(use_gpu): argument
Ddense_update_ops_test.py33 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
Dshape_ops_test.py53 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
533 def _RunAndVerifyResult(self, rank, use_gpu): argument
Dcumulative_logsumexp_test.py40 def _testLogSumExp(self, x, dtype=dtypes.float32, use_gpu=False, **kwargs): argument
49 def _testLogSumExpAllArgs(self, x, axis=0, use_gpu=False): argument
70 def _testGradient(self, x, use_gpu=False, **kwargs): argument
Dcast_op_test.py57 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
Dneon_depthwise_conv_op_test.py95 use_gpu, argument
187 expected, use_gpu): argument
Dparameterized_truncated_normal_op_test.py256 def parameterized_vs_naive(shape, num_iters, use_gpu=False): argument
281 def randn_sampler_switchover(shape, num_iters, use_gpu=False): argument
341 def _benchmarkParameterizedOpVsNaiveOp(self, use_gpu): argument
366 def _benchmarkRandnSampler(self, use_gpu): argument
Dxent_op_test.py60 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
Dconv_ops_3d_test.py53 def _DtypesToTest(self, use_gpu): argument
68 padding, data_format, dtype, use_gpu): argument
128 use_gpu): argument
406 padding, test_input, data_format, use_gpu): argument
721 strides, dilations, padding, data_format, use_gpu, argument
Dbias_op_base.py48 def _testBias(self, np_inputs, np_bias, use_gpu=False): argument
75 def _testBiasNCHW(self, np_inputs, np_bias, use_gpu): argument
143 def _testGradient(self, np_input, bias, dtype, data_format, use_gpu): argument
Ddenormal_test.py38 def _flushDenormalsTest(self, use_gpu, dtypes): argument
Dcwise_ops_test.py223 def _compareBinary(self, x, y, np_func, tf_func, use_gpu=False): argument
234 def _not(self, x, use_gpu=False): argument
321 def _compare(self, fn, c, x, y, use_gpu): argument
617 def _compare(self, c, x, y, use_gpu): argument
726 def _compare(self, x, y, use_gpu): argument
889 def _compare(self, x, use_gpu): argument
988 def _compareMake(self, real, imag, use_gpu): argument
1018 def _compareRealImag(self, cplx, use_gpu): argument
1047 def _compareAngle(self, cplx, use_gpu): argument
1080 def _compareConj(self, cplx, use_gpu): argument
Dreshape_op_test.py33 def _testReshape(self, x, y, use_gpu=False): argument
48 def _testZeroDimReshape(self, x, shape, expected, use_gpu=False): argument
Dconv_ops_test.py165 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
830 use_gpu, argument
876 def _GetVal(data_format, use_gpu): argument
1019 use_gpu, argument
1062 def _GetVal(data_format, use_gpu): argument
1195 padding, data_format, use_gpu, err): argument
1241 padding, data_format, use_gpu, err): argument
[all …]
Dsoftsign_op_test.py36 def _testSoftsign(self, np_features, use_gpu=False): argument
Dsoftmax_op_test.py58 def _testSoftmax(self, np_features, dim=-1, log=False, use_gpu=False): argument
109 def _testOverflow(self, use_gpu=False): argument
Dsoftplus_op_test.py39 def _testSoftplus(self, np_features, use_gpu=False): argument
Dbasic_gpu_test.py94 def _compare(self, x, np_func, tf_func, use_gpu): argument
108 def _testDtype(self, dtype, use_gpu): argument
/external/tensorflow/tensorflow/python/training/
Dtraining_ops_test.py52 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/
Drandom_ops_test.py45 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/
Dnn_fused_batchnorm_test.py61 use_gpu=True, argument
113 use_gpu=True, argument
186 use_gpu=True, argument
246 use_gpu=True, argument
/external/tensorflow/tensorflow/python/kernel_tests/linalg/
Dlinear_operator_toeplitz_test.py47 def _constrain_devices_and_set_default(self, sess, use_gpu, force_gpu): argument

123