Home
last modified time | relevance | path

Searched refs:test_util (Results 1 – 25 of 895) sorted by relevance

12345678910>>...36

/external/tensorflow/tensorflow/python/ops/
Dinit_ops_v2_test.py26 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 …]
Dmath_ops_test.py27 from tensorflow.python.framework import test_util
40 class ReduceTest(test_util.TensorFlowTestCase):
42 @test_util.run_in_graph_and_eager_modes
45 with test_util.device(use_gpu=True):
49 @test_util.run_in_graph_and_eager_modes
52 with test_util.device(use_gpu=True):
62 @test_util.run_in_graph_and_eager_modes
73 @test_util.run_in_graph_and_eager_modes
83 @test_util.run_in_graph_and_eager_modes
94 class LogSumExpTest(test_util.TensorFlowTestCase):
[all …]
/external/tensorflow/tensorflow/python/kernel_tests/
Dcheck_ops_test.py34 from tensorflow.python.framework import test_util
108 @test_util.run_in_graph_and_eager_modes
114 @test_util.run_in_graph_and_eager_modes
121 @test_util.run_in_graph_and_eager_modes
127 @test_util.run_in_graph_and_eager_modes
133 @test_util.run_in_graph_and_eager_modes
139 @test_util.run_in_graph_and_eager_modes
146 @test_util.run_in_graph_and_eager_modes
155 @test_util.run_in_graph_and_eager_modes
162 @test_util.run_in_graph_and_eager_modes
[all …]
Dweights_broadcast_test.py26 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 …]
Dconv_ops_test.py37 from tensorflow.python.framework import test_util
166 if use_gpu and not test_util.CudaSupportsHalfMatMulAndConv():
200 with test_util.device(use_gpu):
208 t1 = test_util.NHWCToNCHW(t1)
209 strides = test_util.NHWCToNCHW(strides)
210 dilations = test_util.NHWCToNCHW(dilations)
212 padding = test_util.NHWCToNCHW(padding)
221 conv = test_util.NCHWToNHWC(conv)
241 with test_util.device(use_gpu):
246 t1 = test_util.NHWCToNCHW(t1)
[all …]
Dvariable_ops_test.py27 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 …]
Dpooling_ops_3d_test.py24 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)
253 ksize = test_util.NHWCToNCHW(ksize)
254 strides = test_util.NHWCToNCHW(strides)
255 t = test_util.NHWCToNCHW(t)
256 output_sizes = test_util.NHWCToNCHW(output_sizes)
298 @test_util.run_deprecated_v1
[all …]
Dscatter_ops_test.py25 from tensorflow.python.framework import test_util
204 @test_util.run_deprecated_v1
208 @test_util.run_deprecated_v1
212 @test_util.run_deprecated_v1
216 @test_util.run_deprecated_v1
220 @test_util.run_deprecated_v1
224 @test_util.run_deprecated_v1
228 @test_util.run_deprecated_v1
232 @test_util.run_deprecated_v1
236 @test_util.run_deprecated_v1
[all …]
Dcwise_ops_binary_test.py28 from tensorflow.python.framework import test_util
80 with test_util.force_cpu():
177 with test_util.use_gpu():
199 @test_util.run_deprecated_v1
237 @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 …]
Dsparse_to_dense_op_py_test.py24 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 …]
Dscan_ops_test.py27 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 …]
Dstring_format_op_test.py24 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 …]
Dconv_ops_3d_test.py29 from tensorflow.python.framework import test_util
54 if not test_util.CudaSupportsHalfMatMulAndConv():
87 t1 = test_util.NHWCToNCHW(t1)
88 strides = test_util.NHWCToNCHW(strides)
92 conv = test_util.NCHWToNHWC(conv)
144 t1 = test_util.NHWCToNCHW(t1)
165 expected = test_util.NCHWToNHWC(expected)
166 computed = test_util.NCHWToNHWC(computed)
423 input_tensor = test_util.NHWCToNCHW(orig_input_tensor)
424 new_strides = test_util.NHWCToNCHW(strides)
[all …]
Dmetrics_test.py32 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 …]
Darray_ops_test.py38 from tensorflow.python.framework import test_util
51 @test_util.run_all_in_graph_and_eager_modes
52 class BatchMatrixTransposeTest(test_util.TensorFlowTestCase):
116 class BooleanMaskTest(test_util.TensorFlowTestCase):
146 @test_util.run_deprecated_v1
152 @test_util.run_deprecated_v1
158 @test_util.run_deprecated_v1
164 @test_util.run_deprecated_v1
170 @test_util.run_deprecated_v1
176 @test_util.run_deprecated_v1
[all …]
Dsplit_op_test.py27 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 …]
Dspacetobatch_op_test.py26 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 …]
Dbatchtospace_op_test.py30 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 …]
/external/tensorflow/tensorflow/lite/kernels/
DBUILD31 "//tensorflow/lite/kernels:test_util",
37 name = "test_util",
39 srcs = ["test_util.cc"],
40 hdrs = ["test_util.h"],
135 ":test_util",
308 "//tensorflow/lite/kernels:test_util",
321 "//tensorflow/lite/kernels:test_util",
334 "//tensorflow/lite/kernels:test_util",
347 "//tensorflow/lite/kernels:test_util",
359 "//tensorflow/lite/kernels:test_util",
[all …]
/external/tensorflow/tensorflow/python/framework/
Dtest_util_test.py40 from tensorflow.python.framework import test_util
50 class TestUtilTest(test_util.TensorFlowTestCase, parameterized.TestCase):
52 @test_util.run_deprecated_v1
56 test_util.assert_ops_in_graph({"hello": "Const"}, ops.get_default_graph())
58 self.assertRaises(ValueError, test_util.assert_ops_in_graph,
61 self.assertRaises(ValueError, test_util.assert_ops_in_graph,
64 @test_util.run_deprecated_v1
102 test_util.assert_equal_graph_def(def_57, def_75)
106 test_util.assert_equal_graph_def(def_57, def_empty)
111 if test_util.IsGoogleCudaEnabled():
[all …]
/external/tensorflow/tensorflow/python/training/
Dlearning_rate_decay_test.py24 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/python/eager/
Dpywrap_tfe_test.py29 from tensorflow.python.framework import test_util
38 @test_util.assert_no_new_tensors
39 @test_util.assert_no_garbage_created
60 @test_util.assert_no_new_tensors
61 @test_util.assert_no_garbage_created
75 @test_util.assert_no_new_tensors
76 @test_util.assert_no_garbage_created
89 @test_util.assert_no_new_tensors
90 @test_util.assert_no_garbage_created
105 @test_util.assert_no_new_tensors
[all …]
/external/python/cpython3/Lib/test/test_importlib/
Dtest_api.py1 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/
Dlearning_rate_schedule_test.py25 from tensorflow.python.framework import test_util
44 class LRDecayTestV2(test_util.TensorFlowTestCase, parameterized.TestCase):
46 @test_util.run_in_graph_and_eager_modes
55 @test_util.run_in_graph_and_eager_modes
78 @test_util.run_deprecated_v1
99 @test_util.run_in_graph_and_eager_modes
120 @test_util.run_in_graph_and_eager_modes
162 class LinearDecayTestV2(test_util.TensorFlowTestCase, parameterized.TestCase):
164 @test_util.run_in_graph_and_eager_modes
174 @test_util.run_in_graph_and_eager_modes
[all …]
/external/tensorflow/tensorflow/python/client/
Dsession_partial_run_test.py30 from tensorflow.python.framework import test_util
42 class PartialRunTest(test_util.TensorFlowTestCase):
191 @test_util.run_deprecated_v1
200 @test_util.run_deprecated_v1
209 @test_util.run_deprecated_v1
213 @test_util.run_deprecated_v1
217 @test_util.run_deprecated_v1
221 @test_util.run_deprecated_v1
225 @test_util.run_deprecated_v1
229 @test_util.run_deprecated_v1
[all …]

12345678910>>...36