/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 | 44 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 …]
|
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 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
|
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 533 def _RunAndVerifyResult(self, rank, 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 70 def _testGradient(self, x, use_gpu=False, **kwargs): 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 | parameterized_truncated_normal_op_test.py | 256 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
|
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 | 53 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
|
D | bias_op_base.py | 48 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
|
D | denormal_test.py | 38 def _flushDenormalsTest(self, use_gpu, dtypes): 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 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
|
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
|
D | conv_ops_test.py | 165 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 …]
|
D | softsign_op_test.py | 36 def _testSoftsign(self, np_features, use_gpu=False): argument
|
D | softmax_op_test.py | 58 def _testSoftmax(self, np_features, dim=-1, log=False, use_gpu=False): argument 109 def _testOverflow(self, use_gpu=False): argument
|
D | softplus_op_test.py | 39 def _testSoftplus(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
|
/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 264 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
|
/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
|