Searched refs:test_util (Results 1 – 25 of 828) sorted by relevance
12345678910>>...34
| /external/tensorflow/tensorflow/python/ops/ |
| D | init_ops_v2_test.py | 26 from tensorflow.python.framework import test_util
84 @test_util.run_in_graph_and_eager_modes
89 @test_util.run_in_graph_and_eager_modes
94 @test_util.run_in_graph_and_eager_modes
103 @test_util.run_in_graph_and_eager_modes
110 @test_util.run_in_graph_and_eager_modes
122 with test_util.use_gpu():
131 @test_util.run_in_graph_and_eager_modes
143 with test_util.use_gpu():
149 @test_util.run_in_graph_and_eager_modes
[all …]
|
| D | math_ops_test.py | 29 from tensorflow.python.framework import test_util
38 @test_util.run_all_in_graph_and_eager_modes
39 class ReduceTest(test_util.TensorFlowTestCase):
43 with test_util.device(use_gpu=True):
49 with test_util.device(use_gpu=True):
88 @test_util.run_all_in_graph_and_eager_modes
89 class LogSumExpTest(test_util.TensorFlowTestCase):
94 with test_util.use_gpu():
102 with test_util.use_gpu():
112 with test_util.use_gpu():
[all …]
|
| /external/tensorflow/tensorflow/python/kernel_tests/ |
| D | weights_broadcast_test.py | 26 from tensorflow.python.framework import test_util
55 @test_util.run_deprecated_v1
59 @test_util.run_deprecated_v1
65 @test_util.run_deprecated_v1
71 @test_util.run_deprecated_v1
77 @test_util.run_deprecated_v1
83 @test_util.run_deprecated_v1
89 @test_util.run_deprecated_v1
96 @test_util.run_deprecated_v1
119 @test_util.run_deprecated_v1
[all …]
|
| D | check_ops_test.py | 35 from tensorflow.python.framework import test_util
110 @test_util.run_in_graph_and_eager_modes
116 @test_util.run_in_graph_and_eager_modes
123 @test_util.run_in_graph_and_eager_modes
129 @test_util.run_in_graph_and_eager_modes
135 @test_util.run_in_graph_and_eager_modes
141 @test_util.run_in_graph_and_eager_modes
148 @test_util.run_in_graph_and_eager_modes
157 @test_util.run_in_graph_and_eager_modes
164 @test_util.run_in_graph_and_eager_modes
[all …]
|
| D | variable_ops_test.py | 27 from tensorflow.python.framework import test_util
63 @test_util.run_deprecated_v1
67 @test_util.run_deprecated_v1
74 @test_util.run_deprecated_v1
82 @test_util.run_deprecated_v1
90 @test_util.run_deprecated_v1
98 @test_util.run_deprecated_v1
111 @test_util.run_deprecated_v1
120 @test_util.run_deprecated_v1
129 @test_util.run_deprecated_v1
[all …]
|
| D | pooling_ops_3d_test.py | 24 from tensorflow.python.framework import test_util
73 t = test_util.NHWCToNCHW(t)
74 window = test_util.NHWCToNCHW(window)
75 strides = test_util.NHWCToNCHW(strides)
83 t = test_util.NCHWToNHWC(t)
273 ksize = test_util.NHWCToNCHW(ksize)
274 strides = test_util.NHWCToNCHW(strides)
275 t = test_util.NHWCToNCHW(t)
276 output_sizes = test_util.NHWCToNCHW(output_sizes)
318 @test_util.run_deprecated_v1
[all …]
|
| D | conv_ops_test.py | 36 from tensorflow.python.framework import test_util
169 if use_gpu and not test_util.GpuSupportsHalfMatMulAndConv():
203 with test_util.device(use_gpu):
211 t1 = test_util.NHWCToNCHW(t1)
212 strides = test_util.NHWCToNCHW(strides)
213 dilations = test_util.NHWCToNCHW(dilations)
215 padding = test_util.NHWCToNCHW(padding)
225 conv = test_util.NCHWToNHWC(conv)
245 with test_util.device(use_gpu):
250 t1 = test_util.NHWCToNCHW(t1)
[all …]
|
| D | sparse_to_dense_op_py_test.py | 24 from tensorflow.python.framework import test_util
45 @test_util.run_deprecated_v1
52 @test_util.run_deprecated_v1
59 @test_util.run_deprecated_v1
66 @test_util.run_deprecated_v1
73 @test_util.run_deprecated_v1
80 @test_util.run_deprecated_v1
90 @test_util.run_deprecated_v1
96 @test_util.run_deprecated_v1
105 @test_util.run_deprecated_v1
[all …]
|
| D | cwise_ops_binary_test.py | 29 from tensorflow.python.framework import test_util
81 with test_util.force_cpu():
178 with test_util.use_gpu():
200 @test_util.run_deprecated_v1
238 @test_util.run_deprecated_v1
267 @test_util.run_deprecated_v1
359 @test_util.run_deprecated_v1
374 @test_util.run_deprecated_v1
392 with test_util.force_cpu():
488 @test_util.run_deprecated_v1
[all …]
|
| D | string_format_op_test.py | 24 from tensorflow.python.framework import test_util
35 @test_util.run_in_graph_and_eager_modes()
51 @test_util.run_in_graph_and_eager_modes()
62 @test_util.run_in_graph_and_eager_modes()
73 @test_util.run_in_graph_and_eager_modes()
87 @test_util.run_in_graph_and_eager_modes()
96 @test_util.run_in_graph_and_eager_modes()
105 @test_util.run_in_graph_and_eager_modes()
135 @test_util.run_in_graph_and_eager_modes()
144 @test_util.run_in_graph_and_eager_modes()
[all …]
|
| D | scan_ops_test.py | 27 from tensorflow.python.framework import test_util
92 @test_util.run_deprecated_v1
99 @test_util.run_deprecated_v1
108 @test_util.run_deprecated_v1
115 @test_util.run_deprecated_v1
122 @test_util.run_deprecated_v1
129 @test_util.run_deprecated_v1
136 @test_util.run_deprecated_v1
137 @test_util.disable_xla("b/123860949") # The computation is constant folded
169 @test_util.run_deprecated_v1
[all …]
|
| D | conv_ops_3d_test.py | 28 from tensorflow.python.framework import test_util
58 if not test_util.GpuSupportsHalfMatMulAndConv():
91 t1 = test_util.NHWCToNCHW(t1)
92 strides = test_util.NHWCToNCHW(strides)
96 conv = test_util.NCHWToNHWC(conv)
148 t1 = test_util.NHWCToNCHW(t1)
169 expected = test_util.NCHWToNHWC(expected)
170 computed = test_util.NCHWToNHWC(computed)
230 (test_util.IsMklEnabled() and is_eager is False):
258 (test_util.IsMklEnabled() and is_eager is False):
[all …]
|
| D | depthtospace_op_test.py | 29 from tensorflow.python.framework import test_util
49 input_nchw = test_util.NHWCToNCHW(input_nhwc)
52 output_nhwc = test_util.NCHWToNHWC(output_nchw)
64 input_nchw = test_util.NHWCToNCHW(input_nhwc)
67 output_nhwc = test_util.NCHWToNHWC(output_nchw)
70 @test_util.run_deprecated_v1
77 @test_util.run_deprecated_v1
86 @test_util.run_deprecated_v1
99 @test_util.run_deprecated_v1
136 @test_util.run_deprecated_v1
[all …]
|
| D | manip_ops_test.py | 25 from tensorflow.python.framework import test_util
41 class RollTest(test_util.TensorFlowTestCase):
65 @test_util.run_deprecated_v1
77 @test_util.run_deprecated_v1
85 @test_util.run_deprecated_v1
96 @test_util.run_deprecated_v1
107 @test_util.run_deprecated_v1
112 @test_util.run_deprecated_v1
119 @test_util.run_deprecated_v1
130 @test_util.run_deprecated_v1
[all …]
|
| D | broadcast_to_ops_test.py | 26 from tensorflow.python.framework import test_util
32 class BroadcastToTest(test_util.TensorFlowTestCase):
34 @test_util.run_deprecated_v1
43 @test_util.run_deprecated_v1
51 @test_util.run_deprecated_v1
59 @test_util.run_deprecated_v1
71 @test_util.run_deprecated_v1
81 @test_util.run_deprecated_v1
91 @test_util.run_deprecated_v1
101 @test_util.run_deprecated_v1
[all …]
|
| D | split_op_test.py | 27 from tensorflow.python.framework import test_util
45 @test_util.run_deprecated_v1
89 @test_util.run_deprecated_v1
100 @test_util.run_in_graph_and_eager_modes
114 @test_util.run_in_graph_and_eager_modes
121 with test_util.device(use_gpu=True):
138 with test_util.device(use_gpu=True):
150 with test_util.device(use_gpu=True):
164 with test_util.device(use_gpu=True):
173 @test_util.run_in_graph_and_eager_modes
[all …]
|
| /external/tensorflow/tensorflow/python/framework/ |
| D | test_util_test.py | 41 from tensorflow.python.framework import test_util
51 class TestUtilTest(test_util.TensorFlowTestCase, parameterized.TestCase):
53 @test_util.run_deprecated_v1
57 test_util.assert_ops_in_graph({"hello": "Const"}, ops.get_default_graph())
59 self.assertRaises(ValueError, test_util.assert_ops_in_graph,
62 self.assertRaises(ValueError, test_util.assert_ops_in_graph,
65 @test_util.run_deprecated_v1
103 test_util.assert_equal_graph_def(def_57, def_75)
107 test_util.assert_equal_graph_def(def_57, def_empty)
112 if test_util.IsGoogleCudaEnabled():
[all …]
|
| /external/tensorflow/tensorflow/python/training/ |
| D | learning_rate_decay_test.py | 24 from tensorflow.python.framework import test_util
32 class LRDecayTest(test_util.TensorFlowTestCase):
34 @test_util.run_in_graph_and_eager_modes
42 @test_util.run_in_graph_and_eager_modes
64 @test_util.run_deprecated_v1
83 @test_util.run_in_graph_and_eager_modes
103 @test_util.run_in_graph_and_eager_modes
104 @test_util.run_v1_only("b/120545219")
149 class LinearDecayTest(test_util.TensorFlowTestCase):
151 @test_util.run_in_graph_and_eager_modes
[all …]
|
| /external/tensorflow/tensorflow/lite/kernels/ |
| D | BUILD | 116 ":test_util",
164 name = "test_util",
166 srcs = ["test_util.cc"],
167 hdrs = ["test_util.h"],
195 ":test_util",
403 ":test_util",
579 ":test_util",
670 ":test_util",
684 ":test_util",
698 ":test_util",
[all …]
|
| /external/python/cpython3/Lib/test/test_importlib/ |
| D | test_api.py | 1 from . import util as test_util unknown
3 init = test_util.import_importlib('importlib')
4 util = test_util.import_importlib('importlib.util')
5 machinery = test_util.import_importlib('importlib.machinery')
21 with test_util.mock_modules('top_level') as mock:
22 with test_util.import_state(meta_path=[mock]):
31 with test_util.mock_modules(pkg_long_name, name) as mock:
32 with test_util.import_state(meta_path=[mock]):
43 with test_util.mock_modules(pkg_long_name, absolute_name) as mock:
44 with test_util.import_state(meta_path=[mock]):
[all …]
|
| /external/tensorflow/tensorflow/python/keras/optimizer_v2/ |
| D | learning_rate_schedule_test.py | 28 from tensorflow.python.framework import test_util
48 class LRDecayTestV2(test_util.TensorFlowTestCase, parameterized.TestCase):
50 @test_util.run_in_graph_and_eager_modes
59 @test_util.run_in_graph_and_eager_modes
82 @test_util.run_deprecated_v1
103 @test_util.run_in_graph_and_eager_modes
144 @test_util.run_in_graph_and_eager_modes
169 class LinearDecayTestV2(test_util.TensorFlowTestCase, parameterized.TestCase):
171 @test_util.run_in_graph_and_eager_modes
181 @test_util.run_in_graph_and_eager_modes
[all …]
|
| /external/tensorflow/tensorflow/python/eager/ |
| D | pywrap_tfe_test.py | 33 from tensorflow.python.framework import test_util
43 @test_util.assert_no_new_tensors
44 @test_util.assert_no_garbage_created
68 @test_util.assert_no_new_tensors
69 @test_util.assert_no_garbage_created
87 @test_util.assert_no_new_tensors
88 @test_util.assert_no_garbage_created
104 @test_util.assert_no_new_tensors
105 @test_util.assert_no_garbage_created
123 @test_util.assert_no_new_tensors
[all …]
|
| D | remote_test.py | 35 from tensorflow.python.framework import test_util
49 workers, _ = test_util.create_local_cluster(1, 0)
60 @test_util.eager_lazy_remote_copy_on_and_off
75 @test_util.eager_lazy_remote_copy_on_and_off
86 @test_util.eager_lazy_remote_copy_on_and_off
138 @test_util.eager_lazy_remote_copy_on_and_off
163 workers, _ = test_util.create_local_cluster(3, 0)
175 @test_util.eager_lazy_remote_copy_on_and_off
188 @test_util.eager_lazy_remote_copy_on_and_off
202 @test_util.eager_lazy_remote_copy_on_and_off
[all …]
|
| /external/tensorflow/tensorflow/python/training/experimental/ |
| D | loss_scale_test.py | 29 from tensorflow.python.framework import test_util
63 @test_util.run_in_graph_and_eager_modes
83 @test_util.run_in_graph_and_eager_modes
90 @test_util.run_in_graph_and_eager_modes
95 @test_util.run_in_graph_and_eager_modes
160 @test_util.run_in_graph_and_eager_modes
168 @test_util.run_in_graph_and_eager_modes
184 @test_util.run_in_graph_and_eager_modes
194 @test_util.run_in_graph_and_eager_modes
204 @test_util.run_in_graph_and_eager_modes
[all …]
|
| /external/tensorflow/tensorflow/python/client/ |
| D | session_partial_run_test.py | 28 from tensorflow.python.framework import test_util
35 class PartialRunTest(test_util.TensorFlowTestCase):
184 @test_util.run_deprecated_v1
193 @test_util.run_deprecated_v1
202 @test_util.run_deprecated_v1
206 @test_util.run_deprecated_v1
210 @test_util.run_deprecated_v1
214 @test_util.run_deprecated_v1
218 @test_util.run_deprecated_v1
222 @test_util.run_deprecated_v1
[all …]
|
12345678910>>...34