Searched refs:test_util (Results 1 – 25 of 871) sorted by relevance
12345678910>>...35
| /external/tensorflow/tensorflow/python/ops/ |
| D | init_ops_v2_test.py | 26 from tensorflow.python.framework import test_util
96 @test_util.run_in_graph_and_eager_modes
101 @test_util.run_in_graph_and_eager_modes
106 @test_util.run_in_graph_and_eager_modes
113 @test_util.run_in_graph_and_eager_modes
118 @test_util.run_in_graph_and_eager_modes
123 @test_util.run_in_graph_and_eager_modes
132 @test_util.run_in_graph_and_eager_modes
140 @test_util.run_in_graph_and_eager_modes
147 @test_util.run_in_graph_and_eager_modes
[all …]
|
| D | math_ops_test.py | 29 from tensorflow.python.framework import test_util
39 @test_util.run_all_in_graph_and_eager_modes
40 class ReduceTest(test_util.TensorFlowTestCase):
44 with test_util.device(use_gpu=True):
60 with test_util.device(use_gpu=True):
136 @test_util.run_all_in_graph_and_eager_modes
137 class LogSumExpTest(test_util.TensorFlowTestCase):
142 with test_util.use_gpu():
150 with test_util.use_gpu():
160 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 | 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 | variable_ops_test.py | 27 from tensorflow.python.framework import test_util
69 @test_util.run_deprecated_v1
73 @test_util.run_deprecated_v1
80 @test_util.run_deprecated_v1
88 @test_util.run_deprecated_v1
96 @test_util.run_deprecated_v1
104 @test_util.run_deprecated_v1
117 @test_util.run_deprecated_v1
126 @test_util.run_deprecated_v1
135 @test_util.run_deprecated_v1
[all …]
|
| D | conv_ops_test.py | 36 from tensorflow.python.framework import test_util
170 if use_gpu and not test_util.GpuSupportsHalfMatMulAndConv():
204 with test_util.device(use_gpu):
212 t1 = test_util.NHWCToNCHW(t1)
213 strides = test_util.NHWCToNCHW(strides)
214 dilations = test_util.NHWCToNCHW(dilations)
216 padding = test_util.NHWCToNCHW(padding)
226 conv = test_util.NCHWToNHWC(conv)
246 with test_util.device(use_gpu):
251 t1 = test_util.NHWCToNCHW(t1)
[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
364 @test_util.run_deprecated_v1
379 @test_util.run_deprecated_v1
397 with test_util.force_cpu():
493 @test_util.run_deprecated_v1
[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
117 @test_util.run_deprecated_v1
124 @test_util.run_deprecated_v1
131 @test_util.run_deprecated_v1
138 @test_util.run_deprecated_v1
145 @test_util.run_deprecated_v1
146 @test_util.disable_xla("b/123860949") # The computation is constant folded
[all …]
|
| D | conv_ops_3d_test.py | 28 from tensorflow.python.framework import test_util
51 @test_util.run_all_without_tensor_float_32(
62 if not test_util.GpuSupportsHalfMatMulAndConv():
91 t1 = test_util.NHWCToNCHW(t1)
92 strides = test_util.NHWCToNCHW(strides)
96 conv = test_util.NCHWToNHWC(conv)
144 t1 = test_util.NHWCToNCHW(t1)
165 expected = test_util.NCHWToNHWC(expected)
166 computed = test_util.NCHWToNHWC(computed)
200 @test_util.run_in_graph_and_eager_modes
[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 | 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 | batchtospace_op_test.py | 30 from tensorflow.python.framework import test_util
54 @test_util.run_deprecated_v1
75 @test_util.run_deprecated_v1
84 @test_util.run_deprecated_v1
94 @test_util.run_deprecated_v1
104 @test_util.run_deprecated_v1
114 @test_util.run_deprecated_v1
123 @test_util.run_deprecated_v1
171 @test_util.run_deprecated_v1
176 @test_util.run_deprecated_v1
[all …]
|
| D | metrics_test.py | 32 from tensorflow.python.framework import test_util
179 @test_util.run_deprecated_v1
184 @test_util.run_deprecated_v1
191 @test_util.run_deprecated_v1
198 @test_util.run_deprecated_v1
216 @test_util.run_deprecated_v1
238 @test_util.run_deprecated_v1
281 @test_util.run_deprecated_v1
287 @test_util.run_deprecated_v1
293 @test_util.run_deprecated_v1
[all …]
|
| D | spacetobatch_op_test.py | 26 from tensorflow.python.framework import test_util
123 @test_util.run_deprecated_v1
131 @test_util.run_deprecated_v1
142 @test_util.run_deprecated_v1
151 @test_util.run_deprecated_v1
162 @test_util.run_deprecated_v1
175 @test_util.run_deprecated_v1
220 @test_util.run_deprecated_v1
228 @test_util.run_deprecated_v1
243 @test_util.run_deprecated_v1
[all …]
|
| D | py_func_test.py | 38 from tensorflow.python.framework import test_util
137 @test_util.run_v1_only("b/120545219")
204 @test_util.run_v1_only("b/120545219")
222 @test_util.run_v1_only("b/120545219")
240 @test_util.run_v1_only("b/120545219")
256 @test_util.run_v1_only("b/120545219")
263 @test_util.run_v1_only("b/120545219")
271 @test_util.run_v1_only("b/120545219")
279 @test_util.run_v1_only("b/120545219")
293 @test_util.run_v1_only("b/120545219")
[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 | 44 from tensorflow.python.framework import test_util
55 class TestUtilTest(test_util.TensorFlowTestCase, parameterized.TestCase):
60 test_util.assert_ops_in_graph({"hello": "Const"}, ops.get_default_graph())
62 self.assertRaises(ValueError, test_util.assert_ops_in_graph,
65 self.assertRaises(ValueError, test_util.assert_ops_in_graph,
68 @test_util.run_deprecated_v1
106 test_util.assert_equal_graph_def(def_57, def_75)
110 test_util.assert_equal_graph_def(def_57, def_empty)
127 test_util.assert_equal_graph_def(def_1, def_2,
130 test_util.assert_equal_graph_def(def_1, def_2,
[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/eager/ |
| D | pywrap_tfe_test.py | 37 from tensorflow.python.framework import test_util
46 @test_util.assert_no_new_tensors
47 @test_util.assert_no_garbage_created
70 @test_util.assert_no_new_tensors
71 @test_util.assert_no_garbage_created
87 @test_util.assert_no_new_tensors
88 @test_util.assert_no_garbage_created
103 @test_util.assert_no_new_tensors
104 @test_util.assert_no_garbage_created
121 @test_util.assert_no_new_tensors
[all …]
|
| /external/tensorflow/tensorflow/lite/kernels/ |
| D | BUILD | 120 ":test_util",
170 name = "test_util",
172 srcs = ["test_util.cc"],
173 hdrs = ["test_util.h"],
203 # based on SingleOpModel or its derivatives defined in test_util.h/cc.
246 ":test_util",
508 ":test_util",
840 ":test_util",
854 ":test_util",
868 ":test_util",
[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/distribute/ |
| D | test_util_test.py | 25 from tensorflow.python.distribute import test_util
51 results = test_util.gather(strategy, strategy.run(f))
68 results = test_util.gather(strategy, strategy.run(f))
81 test_util.set_logical_devices_to_at_least('CPU', 3)
102 order = test_util.topological_sort_operations(graph.get_operations())
108 test_util.assert_sequential_execution(order, [a, d])
109 test_util.assert_sequential_execution(order, [e, a, f])
111 test_util.assert_sequential_execution(order, [a, c])
113 test_util.assert_sequential_execution(order, [f, a, c])
115 test_util.assert_sequential_execution(order, [d, e, a, c])
[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>>...35