| /external/tensorflow/tensorflow/python/kernel_tests/math_ops/ |
| D | bincount_op_test.py | 189 def test_bincount_all_count(self, dtype): argument
205 def test_bincount_all_count_with_weights(self, dtype): argument
223 def test_bincount_all_binary(self, dtype): argument
240 def test_bincount_all_binary_with_weights(self, dtype): argument
253 def _test_bincount_col_count(self, num_rows, num_cols, size, dtype): argument
266 def _test_bincount_col_binary(self, num_rows, num_cols, size, dtype): argument
283 dtype): argument
314 def test_col_reduce_shared_memory(self, dtype): argument
326 def test_col_reduce_global_memory(self, dtype): argument
338 def test_col_reduce_shared_memory_with_weights(self, dtype): argument
[all …]
|
| D | sets_test.py | 38 def _values(values, dtype): argument
44 def _constant(values, dtype): argument
48 def _dense_to_sparse(dense, dtype): argument
76 def _test_set_size_2d(self, dtype): argument
90 def _test_set_size_duplicates_2d(self, dtype): argument
108 def _test_set_size_3d(self, dtype, invalid_indices=False): argument
172 def _test_set_intersection_multirow_2d(self, dtype): argument
209 def _test_dense_set_intersection_multirow_2d(self, dtype): argument
234 def _test_set_intersection_duplicates_2d(self, dtype): argument
286 def _test_set_intersection_3d(self, dtype, invalid_indices=False): argument
[all …]
|
| D | batch_matmul_op_test.py | 31 def GetRandomNormalInput(shape, dtype): argument
74 def _testNonEmpty(self, dtype, adjoint_a, adjoint_b, use_static_shape): argument
94 def _testBroadcasting(self, dtype, adjoint_a, adjoint_b, use_static_shape): argument
112 def _testEmpty(self, dtype, adjoint_a, adjoint_b, use_static_shape): argument
127 def _GetBatchMatmulOpTest(dtype, adjoint_a, adjoint_b, use_static_shape): argument
138 def _GetBatchMatmulOpBroadcastingTest(dtype, adjoint_a, adjoint_b, argument
174 def _compare(self, a_shape, b_shape, dtype, adjoint_a, adjoint_b): argument
181 def _GetBatchMatmulGradientTest(dtype, adjoint_a, adjoint_b): argument
193 def _GetBatchMatmulGradientWithBroadcastingTest(dtype, adjoint_a, adjoint_b): argument
|
| /external/tensorflow/tensorflow/compiler/tests/ |
| D | special_math_test.py | 83 def adjust_tolerance_for_tpu(self, dtype, rtol, atol): argument
91 def _test_range(self, low, high, dtype, rtol, atol, is_negative=False): argument
107 def testSmallX(self, dtype, rtol, atol): argument
113 def testGreaterThanNegativeTwentyExponent(self, dtype, rtol, atol): argument
119 def testGreaterThanNegativeTenExponent(self, dtype, rtol, atol): argument
125 def testGreaterThanNegativeFiveExponent(self, dtype, rtol, atol): argument
131 def testXGreaterThanOneTenth(self, dtype, rtol, atol): argument
137 def testXGreaterThanOne(self, dtype, rtol, atol): argument
150 def adjust_tolerance_for_tpu(self, dtype, rtol, atol): argument
206 def testLargeXSmallQ(self, dtype, rtol, atol): argument
[all …]
|
| D | stateful_random_ops_test.py | 155 def testXLAEqualsCPU(self, dtype): argument
203 def _testRngIsNotConstant(self, rng, dtype): argument
211 def check_dtype(self, dtype): argument
217 def testUniformIsNotConstant(self, alg, dtype): argument
221 def rng(dtype): argument
228 def testNormalIsNotConstant(self, alg, dtype): argument
232 def rng(dtype): argument
238 def testUniformIsInRange(self, alg, dtype): argument
251 def testNormalIsFinite(self, alg, dtype): argument
259 def testDistributionOfUniform(self, alg, dtype): argument
[all …]
|
| D | ftrl_test.py | 30 def initVariableAndGradient(self, dtype): argument
38 def equivAdagradTest_FtrlPart(self, steps, dtype): argument
58 def equivAdagradTest_AdagradPart(self, steps, dtype): argument
73 def equivGradientDescentTest_FtrlPart(self, steps, dtype): argument
93 def equivGradientDescentTest_GradientDescentPart(self, steps, dtype): argument
|
| /external/tensorflow/tensorflow/python/kernel_tests/strings_ops/ |
| D | substr_op_test.py | 38 def testScalarString(self, dtype, pos, unit): argument
60 def testScalarString_EdgeCases(self, dtype, unit): argument
124 def testVectorStrings(self, dtype, pos, unit): argument
147 def testMatrixStrings(self, dtype, unit): argument
197 def testElementWisePosLen(self, dtype, unit): argument
238 def testBroadcast(self, dtype, unit): argument
320 def testBadBroadcast(self, dtype, unit): argument
340 def testOutOfRangeError_Scalar(self, dtype, pos, unit): argument
364 def testOutOfRangeError_VectorScalar(self, dtype, pos, unit): argument
385 def testOutOfRangeError_MatrixMatrix(self, dtype, unit): argument
[all …]
|
| /external/tensorflow/tensorflow/python/ops/signal/ |
| D | window_ops.py | 31 def _check_params(window_length, dtype): argument
54 def kaiser_window(window_length, beta=12., dtype=dtypes.float32, name=None): argument
95 dtype=dtypes.float32, name=None): argument
122 def vorbis_window(window_length, dtype=dtypes.float32, name=None): argument
147 def hann_window(window_length, periodic=True, dtype=dtypes.float32, name=None): argument
173 def hamming_window(window_length, periodic=True, dtype=dtypes.float32, argument
200 dtype, a, b): argument
|
| /external/tensorflow/tensorflow/python/ops/ |
| D | init_ops.py | 54 def __call__(self, shape, dtype=None, partition_info=None): argument
169 def __init__(self, dtype=dtypes.float32): argument
172 def __call__(self, shape, dtype=None, partition_info=None): argument
206 def __init__(self, dtype=dtypes.float32): argument
209 def __call__(self, shape, dtype=None, partition_info=None): argument
369 def __init__(self, value=0, dtype=dtypes.float32, verify_shape=False): argument
380 def __call__(self, shape, dtype=None, partition_info=None, verify_shape=None): argument
465 def __init__(self, minval=.0, maxval=None, seed=None, dtype=dtypes.float32): argument
471 def __call__(self, shape, dtype=None, partition_info=None): argument
555 def __init__(self, mean=0.0, stddev=1.0, seed=None, dtype=dtypes.float32): argument
[all …]
|
| D | special_math_ops_test.py | 188 def test_dawsn_odd(self, dtype): argument
195 def test_dawsn_small(self, dtype): argument
205 def test_dawsn_larger(self, dtype): argument
237 def test_expint_small(self, dtype): argument
247 def test_expint_larger(self, dtype): argument
273 def test_fresnel_cos_odd(self, dtype): argument
280 def test_fresnel_cos_small(self, dtype): argument
290 def test_fresnel_cos_larger(self, dtype): argument
318 def test_fresnel_sin_odd(self, dtype): argument
325 def test_fresnel_sin_small(self, dtype): argument
[all …]
|
| D | random_ops.py | 46 dtype=dtypes.float32, argument
107 dtype=dtypes.float32, argument
163 dtype=dtypes.float32, argument
218 dtype=dtypes.float32, argument
505 def categorical(logits, num_samples, dtype=None, seed=None, name=None): argument
533 def multinomial_categorical_impl(logits, num_samples, dtype, seed): argument
566 dtype=dtypes.float32, argument
657 def random_poisson(lam, shape, dtype=dtypes.float32, seed=None, name=None): argument
697 def random_poisson_v2(shape, lam, dtype=dtypes.float32, seed=None, name=None): argument
|
| D | bincount_ops.py | 36 dtype=dtypes.int32, argument
225 dtype=dtypes.int32): argument
455 def validate_dense_weights(values, weights, dtype=None): argument
470 def validate_sparse_weights(values, weights, dtype=None): argument
506 def validate_ragged_weights(values, weights, dtype=None): argument
|
| D | init_ops_v2.py | 48 def __call__(self, shape, dtype=None, **kwargs): argument
135 def __call__(self, shape, dtype=dtypes.float32, **kwargs): argument
182 def __call__(self, shape, dtype=dtypes.float32, **kwargs): argument
275 def __call__(self, shape, dtype=None, **kwargs): argument
336 def __call__(self, shape, dtype=dtypes.float32, **kwargs): argument
405 def __call__(self, shape, dtype=dtypes.float32, **kwargs): argument
474 def __call__(self, shape, dtype=dtypes.float32, **kwargs): argument
573 def __call__(self, shape, dtype=dtypes.float32, **kwargs): argument
664 def __call__(self, shape, dtype=dtypes.float32, **kwargs): argument
737 def __call__(self, shape, dtype=dtypes.float32, **kwargs): argument
[all …]
|
| D | bincount_ops_test.py | 556 def test_sparse_input_all_count(self, dtype): argument
576 def test_sparse_input_all_count_with_weights(self, dtype): argument
601 def test_sparse_input_all_binary(self, dtype): argument
622 def test_sparse_input_col_reduce_count(self, dtype): argument
648 def test_sparse_input_col_reduce_binary(self, dtype): argument
678 def test_ragged_input_count(self, dtype): argument
697 def test_ragged_input_binary(self, dtype): argument
717 def test_ragged_input_count_with_weights(self, dtype): argument
738 def test_ragged_input_count_np(self, dtype): argument
758 def test_ragged_input_count_np_with_weights(self, dtype): argument
|
| /external/tensorflow/tensorflow/python/ops/ragged/ |
| D | convert_to_tensor_or_ragged_tensor_op_test.py | 51 def testConvertRaggedTensor(self, pylist, dtype=None, preferred_dtype=None): argument
77 dtype=None, argument
115 dtype=None, argument
136 dtype=None, argument
150 def testConvertTensor(self, pylist, dtype=None, preferred_dtype=None): argument
171 dtype=None, argument
203 dtype=None, argument
222 dtype=None, argument
|
| D | row_partition.py | 196 dtype=None, argument
315 dtype=None, argument
378 dtype=None, argument
434 dtype=None, argument
485 dtype=None, argument
533 dtype=None, argument
666 def _convert_row_partition(cls, partition, name, dtype=None, dtype_hint=None): argument
729 def dtype(self): member in RowPartition
963 def with_dtype(self, dtype): argument
1220 dtype=dtypes.int64): argument
[all …]
|
| /external/tensorflow/tensorflow/python/kernel_tests/linalg/ |
| D | linalg_ops_test.py | 35 def _RandomPDMatrix(n, rng, dtype=np.float64): argument
207 def test_eye_no_placeholder(self, num_rows, num_columns, batch_shape, dtype): argument
238 self, num_rows, num_columns, batch_shape, dtype): argument
316 dtype = np.float32 variable in MatrixRankStatic32Test
322 dtype = np.float64 variable in MatrixRankDynamic64Test
379 dtype = np.float32 variable in PinvTestDynamic32DefaultRcond
386 dtype = np.float64 variable in PinvTestStatic64DefaultRcond
393 dtype = np.float32 variable in PinvTestDynamic32CustomtRcond
400 dtype = np.float64 variable in PinvTestStatic64CustomRcond
414 dtype = np.float32 variable in _LUReconstruct
[all …]
|
| /external/tensorflow/tensorflow/python/ops/linalg/ |
| D | linear_operator_test_util.py | 131 self, shapes_info, dtype, use_placeholder, argument
313 def _test_slicing(use_placeholder, shapes_info, dtype): argument
337 def _test_to_dense(use_placeholder, shapes_info, dtype): argument
351 def _test_det(use_placeholder, shapes_info, dtype): argument
366 def _test_log_abs_det(use_placeholder, shapes_info, dtype): argument
383 def _test_operator_matmul_with_same_type(use_placeholder, shapes_info, dtype): argument
402 def _test_operator_solve_with_same_type(use_placeholder, shapes_info, dtype): argument
425 dtype, argument
497 dtype, argument
517 dtype, argument
[all …]
|
| /external/tensorflow/tensorflow/python/framework/ |
| D | constant_op.py | 74 def convert_to_eager_tensor(value, ctx, dtype=None): argument
107 value, dtype=None, shape=None, name="Const", verify_shape=False): argument
171 def constant(value, dtype=None, shape=None, name="Const"): argument
272 value, dtype, shape, name, verify_shape, allow_broadcast): argument
302 def _constant_eager_impl(ctx, value, dtype, shape, verify_shape): argument
340 def _constant_tensor_conversion_function(v, dtype=None, name=None, argument
353 dtype=None, argument
387 dtype=None, argument
|
| /external/tensorflow/tensorflow/python/keras/ |
| D | metrics.py | 147 def __init__(self, name=None, dtype=None, **kwargs): argument
234 def dtype(self): member in Metric
294 dtype=None): argument
364 def __init__(self, reduction, name, dtype=None): argument
485 def __init__(self, name='sum', dtype=None): argument
527 def __init__(self, name='mean', dtype=None): argument
570 def __init__(self, normalizer, name=None, dtype=None): argument
640 def __init__(self, fn, name=None, dtype=None, **kwargs): argument
740 def __init__(self, name='accuracy', dtype=None): argument
784 def __init__(self, name='binary_accuracy', dtype=None, threshold=0.5): argument
[all …]
|
| /external/tensorflow/tensorflow/python/keras/initializers/ |
| D | initializers_v2.py | 77 def __call__(self, shape, dtype=None, **kwargs): argument
134 def __call__(self, shape, dtype=None, **kwargs): argument
171 def __call__(self, shape, dtype=None, **kwargs): argument
220 def __call__(self, shape, dtype=None, **kwargs): argument
273 def __call__(self, shape, dtype=None, **kwargs): argument
336 def __call__(self, shape, dtype=None, **kwargs): argument
401 def __call__(self, shape, dtype=None, **kwargs): argument
491 def __call__(self, shape, dtype=None, **kwargs): argument
578 def __call__(self, shape, dtype=None, **kwargs): argument
645 def __call__(self, shape, dtype=None, **kwargs): argument
[all …]
|
| /external/tensorflow/tensorflow/python/kernel_tests/random/ |
| D | stateful_random_ops_test.py | 409 def compare(dtype, old, new): argument
436 def old_normal(dtype, seed1, seed2): argument
439 def new_normal(dtype, gen): argument
441 def old_truncated_normal(dtype, seed1, seed2): argument
444 def new_truncated_normal(dtype, gen): argument
446 def old_uniform_int(dtype, seed1, seed2): argument
451 def new_uniform_int(dtype, gen): argument
453 def old_uniform(dtype, seed1, seed2): argument
456 def new_uniform(dtype, gen): argument
486 def testGPUEqualsCPU(self, dtype): argument
[all …]
|
| /external/tensorflow/tensorflow/python/training/ |
| D | warm_starting_util_test.py | 564 partitioner = lambda shape, dtype: [1] * len(shape) argument
593 partitioner = lambda shape, dtype: [1] * len(shape) argument
627 partitioner = lambda shape, dtype: [1] * len(shape) argument
663 partitioner = lambda shape, dtype: [1] * len(shape) argument
708 partitioner = lambda shape, dtype: [1] * len(shape) argument
751 partitioner = lambda shape, dtype: [1] * len(shape) argument
822 partitioner = lambda shape, dtype: [1] * len(shape) argument
894 def _partitioner(shape, dtype): # pylint:disable=unused-argument argument
1020 def _partitioner(shape, dtype): # pylint:disable=unused-argument argument
1080 def _partitioner(shape, dtype): # pylint:disable=unused-argument argument
[all …]
|
| /external/tensorflow/tensorflow/python/ops/numpy_ops/ |
| D | np_array_ops.py | 46 def empty(shape, dtype=float): # pylint: disable=redefined-outer-name argument
51 def empty_like(a, dtype=None): argument
56 def zeros(shape, dtype=float): # pylint: disable=redefined-outer-name argument
63 def zeros_like(a, dtype=None): # pylint: disable=missing-docstring argument
71 def ones(shape, dtype=float): # pylint: disable=redefined-outer-name argument
78 def ones_like(a, dtype=None): argument
84 def eye(N, M=None, k=0, dtype=float): # pylint: disable=invalid-name,missing-docstring argument
115 def identity(n, dtype=float): argument
120 def full(shape, fill_value, dtype=None): # pylint: disable=redefined-outer-name argument
131 def full_like(a, fill_value, dtype=None, order='K', subok=True, shape=None): # pylint: disable=mis… argument
[all …]
|
| /external/tensorflow/tensorflow/python/kernel_tests/array_ops/ |
| D | one_hot_op_test.py | 32 dtype=None, argument
54 def _testBasic(self, dtype): argument
84 def _testDefaultBasic(self, dtype): argument
127 def _testBatch(self, dtype): argument
158 def _testDefaultValuesBatch(self, dtype): argument
178 def _testValueTypeBatch(self, dtype): argument
210 def _testEmpty(self, dtype): argument
|