Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/python/kernel_tests/
Dconstant_op_test.py28 from tensorflow.python.framework import dtypes as dtypes_lib unknown
48 dtype = dtypes_lib.as_dtype(np_ans.dtype)
58 dtype = dtypes_lib.as_dtype(np_ans.dtype)
71 constant_op.constant(dtypes_lib.string, "[,]")
75 bfloat16 = dtypes_lib.bfloat16.as_numpy_dtype
149 dtype=dtypes_lib.variant.as_datatype_enum,
283 c = constant_op.constant([[1, 2], [3]], dtype=dtypes_lib.int32)
308 self.assertEqual(dtypes_lib.int32, x.dtype)
312 self.assertEqual(dtypes_lib.int32, x.dtype)
316 self.assertEqual(dtypes_lib.int32, x.dtype)
[all …]
Dconstant_op_eager_test.py26 from tensorflow.python.framework import dtypes as dtypes_lib unknown
72 self.assertEqual(dtypes_lib.float32, tf_ans.dtype)
78 self.assertEqual(dtypes_lib.float32, tf_ans.dtype)
83 self.assertEqual(dtypes_lib.float32, tf_ans.dtype)
89 self.assertEqual(dtypes_lib.float32, tf_ans.dtype)
99 tf_ans = constant_op.constant(orig, dtypes_lib.float64)
100 self.assertEqual(dtypes_lib.float64, tf_ans.dtype)
104 tf_ans = constant_op.constant(2**54 + 1, dtypes_lib.float64)
110 constant_op.constant(10**310, dtypes_lib.float64)
127 self.assertEqual(dtypes_lib.int64, tf_ans.dtype)
[all …]
Dpadding_fifo_queue_test.py28 from tensorflow.python.framework import dtypes as dtypes_lib unknown
44 10, dtypes_lib.float32, ((None,),), name="Q")
58 5, (dtypes_lib.int32, dtypes_lib.float32), ((), ()),
76 5, (dtypes_lib.int32, dtypes_lib.float32),
101 q = data_flow_ops.PaddingFIFOQueue(10, dtypes_lib.float32, ((),))
108 10, dtypes_lib.float32, shapes=((3, 2),))
118 10, [dtypes_lib.int32, dtypes_lib.int32], shapes=[(), (2,)])
124 q = data_flow_ops.PaddingFIFOQueue(10, dtypes_lib.float32, ((),))
150 q = data_flow_ops.PaddingFIFOQueue(10, dtypes_lib.float32, ((),))
174 q = data_flow_ops.PaddingFIFOQueue(10, dtypes_lib.float32, ((),))
[all …]
Dfifo_queue_test.py30 from tensorflow.python.framework import dtypes as dtypes_lib unknown
47 q = data_flow_ops.FIFOQueue(10, dtypes_lib.float32, name="Q")
61 5, (dtypes_lib.int32, dtypes_lib.float32),
79 5, (dtypes_lib.int32, dtypes_lib.float32),
104 q = data_flow_ops.FIFOQueue(10, dtypes_lib.float32)
110 q = data_flow_ops.FIFOQueue(10, dtypes_lib.float16)
116 q = data_flow_ops.FIFOQueue(10, dtypes_lib.float32, shapes=(3, 2))
126 10, [dtypes_lib.int32, dtypes_lib.int32], shapes=[(), (2,)])
132 q = data_flow_ops.FIFOQueue(10, [dtypes_lib.int32], shapes=[()])
139 q = data_flow_ops.FIFOQueue(10, [dtypes_lib.int32], shapes=[()])
[all …]
Dsparse_conditional_accumulator_test.py25 from tensorflow.python.framework import dtypes as dtypes_lib unknown
63 dtypes_lib.float32, name="Q")
79 dtypes_lib.float32, name="Q", reduction_type="Invalid")
84 dtypes_lib.float32,
106 dtypes_lib.float32, name="Q")
113 dtypes_lib.float32, name="Q", shape=tensor_shape.TensorShape([1]))
121 dtypes_lib.float32, name="Q", shape=tensor_shape.TensorShape([3, 3]))
132 dtypes = [dtypes_lib.float16, dtypes_lib.float32, dtypes_lib.float64]
156 dtypes_lib.float32, name="Q", shape=tensor_shape.TensorShape([2, 2]))
158 dtypes_lib.float32, name="Q", shape=tensor_shape.TensorShape([2, 2]))
[all …]
Dconditional_accumulator_test.py25 from tensorflow.python.framework import dtypes as dtypes_lib unknown
44 q = data_flow_ops.ConditionalAccumulator(dtypes_lib.float32, name="Q")
60 dtypes_lib.float32, name="Q", reduction_type="Invalid")
65 dtypes_lib.float32,
86 q = data_flow_ops.ConditionalAccumulator(dtypes_lib.float32, name="Q")
93 dtypes_lib.float32, name="Q", shape=tensor_shape.TensorShape([1]))
101 dtypes_lib.float32, name="Q", shape=tensor_shape.TensorShape([1]))
108 dtypes = [dtypes_lib.float16, dtypes_lib.float32, dtypes_lib.float64]
127 dtypes_lib.float32, name="Q", shape=tensor_shape.TensorShape([1]))
129 dtypes_lib.float32, name="Q", shape=tensor_shape.TensorShape([1]))
[all …]
Dsegment_reduction_ops_test.py27 from tensorflow.python.framework import dtypes as dtypes_lib unknown
38 def _input(self, input_shape, dtype=dtypes_lib.int32):
90 dtypes_lib.float32, dtypes_lib.float64, dtypes_lib.int64,
91 dtypes_lib.int32, dtypes_lib.complex64, dtypes_lib.complex128
112 if dtype in (dtypes_lib.complex64, dtypes_lib.complex128):
155 tf_x, _ = self._input(shape, dtype=dtypes_lib.float32)
164 tf_x, np_x = self._input(shape, dtype=dtypes_lib.float32)
175 tf_x, np_x = self._input(shape, dtype=dtypes_lib.float32)
221 tf_x, _ = self._input(shape, dtype=dtypes_lib.float32)
232 tf_x, _ = self._input(shape, dtype=dtypes_lib.float32)
[all …]
Dmetrics_test.py28 from tensorflow.python.framework import dtypes as dtypes_lib unknown
202 4, dtypes=dtypes_lib.float32, shapes=(1, 2))
220 4, dtypes=dtypes_lib.float32, shapes=(1, 2))
328 values_placeholder = array_ops.placeholder(dtype=dtypes_lib.float32)
383 4, dtypes=dtypes_lib.float32, shapes=(1, 2))
401 2, dtypes=dtypes_lib.float32, shapes=(2, 2, 2))
424 4, dtypes=dtypes_lib.float32, shapes=(1, 2))
447 4, dtypes=dtypes_lib.float32, shapes=(1, 2))
456 4, dtypes=dtypes_lib.float32, shapes=(1, 1))
475 4, dtypes=dtypes_lib.float32, shapes=(1, 2))
[all …]
Dcholesky_op_test.py26 from tensorflow.python.framework import dtypes as dtypes_lib unknown
208 shapes, dtypes=(dtypes_lib.float32, dtypes_lib.float64))
215 shapes, dtypes=(dtypes_lib.complex64, dtypes_lib.complex128))
223 dtypes=(dtypes_lib.float32, dtypes_lib.float64),
231 shapes, dtypes=(dtypes_lib.float32,), scalarTest=True)
238 shapes, dtypes=(dtypes_lib.float64,), scalarTest=True)
245 shapes, dtypes=(dtypes_lib.complex64,), scalarTest=True)
252 shapes, dtypes=(dtypes_lib.complex128,), scalarTest=True)
261 x = constant_op.constant(data, dtypes_lib.float32)
270 dtypes=(dtypes_lib.float32, dtypes_lib.float64,
[all …]
Dcwise_ops_test.py24 from tensorflow.python.framework import dtypes as dtypes_lib unknown
229 self.assertEqual(out.dtype, dtypes_lib.bool)
238 self.assertEqual(out.dtype, dtypes_lib.bool)
676 dtypes_lib.float16,
677 dtypes_lib.float32,
678 dtypes_lib.float64,
679 dtypes_lib.int32,
680 dtypes_lib.int64,
681 dtypes_lib.complex64,
682 dtypes_lib.complex128,
[all …]
Dweights_broadcast_test.py23 from tensorflow.python.framework import dtypes as dtypes_lib unknown
44 weights_placeholder = array_ops.placeholder(dtypes_lib.float32)
45 values_placeholder = array_ops.placeholder(dtypes_lib.float32)
108 weights_placeholder = array_ops.placeholder(dtypes_lib.float32)
109 values_placeholder = array_ops.placeholder(dtypes_lib.float32)
172 weights_placeholder = array_ops.placeholder(dtypes_lib.float32)
173 values_placeholder = array_ops.placeholder(dtypes_lib.float32)
250 weights_placeholder = array_ops.placeholder(dtypes_lib.float32)
251 values_placeholder = array_ops.placeholder(dtypes_lib.float32)
Dself_adjoint_eig_op_test.py24 from tensorflow.python.framework import dtypes as dtypes_lib unknown
153 if dtype_ in (dtypes_lib.float32, dtypes_lib.complex64):
209 if dtype_ in (dtypes_lib.float32, dtypes_lib.complex64):
244 for dtype in (dtypes_lib.float32, dtypes_lib.float64, dtypes_lib.complex64,
245 dtypes_lib.complex128):
Ddiag_op_test.py23 from tensorflow.python.framework import dtypes as dtypes_lib unknown
71 v = array_ops.placeholder(dtype=dtypes_lib.float32)
166 v = array_ops.placeholder(dtype=dtypes_lib.float32)
179 np.random.rand(*shape), dtype=dtypes_lib.float32)
182 np.random.rand(*diag_shape), dtype=dtypes_lib.float32)
198 v = array_ops.placeholder(dtype=dtypes_lib.float32)
199 mat = array_ops.placeholder(dtype=dtypes_lib.float32)
200 grad_input = array_ops.placeholder(dtype=dtypes_lib.float32)
276 v = array_ops.placeholder(dtype=dtypes_lib.float32)
526 dtypes = (dtypes_lib.float32, dtypes_lib.float64)
[all …]
Dmatrix_band_part_op_test.py24 from tensorflow.python.framework import dtypes as dtypes_lib unknown
61 for index_dtype in [dtypes_lib.int32, dtypes_lib.int64]:
145 dtypes_lib.bfloat16.as_numpy_dtype, np.float32, np.float64,
Daccumulate_n_test.py22 from tensorflow.python.framework import dtypes as dtypes_lib unknown
58 x0 = array_ops.placeholder(dtype=dtypes_lib.int32, shape=[None])
88 ops.convert_to_tensor(x, dtype=dtypes_lib.float32)
Dcwise_ops_binary_test.py24 from tensorflow.python.framework import dtypes as dtypes_lib unknown
108 dtypes_lib.float16: 1e-3,
109 dtypes_lib.float32: 1e-3,
110 dtypes_lib.complex64: 1e-2,
111 dtypes_lib.float64: 1e-5,
112 dtypes_lib.complex128: 1e-4
142 tol = self._GRAD_TOL[dtypes_lib.as_dtype(x.dtype)]
172 tol = self._GRAD_TOL[dtypes_lib.as_dtype(x.dtype)]
Dcwise_ops_unary_test.py26 from tensorflow.python.framework import dtypes as dtypes_lib unknown
82 dtypes_lib.bfloat16.as_numpy_dtype):
91 elif x.dtype == dtypes_lib.bfloat16.as_numpy_dtype:
/external/tensorflow/tensorflow/python/kernel_tests/random/
Drandom_shuffle_queue_test.py28 from tensorflow.python.framework import dtypes as dtypes_lib unknown
52 q = data_flow_ops.RandomShuffleQueue(10, 5, dtypes_lib.float32)
61 10, 5, dtypes_lib.float32, shapes=tensor_shape.TensorShape([3, 2]))
71 10, 5, [dtypes_lib.int32, dtypes_lib.int32], shapes=[(), (2,)])
76 10, 5, dtypes_lib.int32, shapes=tensor_shape.TensorShape([3]))
83 10, 0, [dtypes_lib.int32, dtypes_lib.int32], shapes=[(), (1,)])
99 q = data_flow_ops.RandomShuffleQueue(10, 0, dtypes_lib.float32)
125 q = data_flow_ops.RandomShuffleQueue(10, 0, dtypes_lib.float32)
149 q = data_flow_ops.RandomShuffleQueue(10, 0, dtypes_lib.float32)
162 q = data_flow_ops.RandomShuffleQueue(3, 0, dtypes_lib.float32)
[all …]
/external/tensorflow/tensorflow/contrib/metrics/python/ops/
Dmetric_ops_test.py28 from tensorflow.python.framework import dtypes as dtypes_lib unknown
183 4, dtypes=dtypes_lib.float32, shapes=(1, 2))
200 4, dtypes=dtypes_lib.float32, shapes=(1, 2))
222 4, dtypes=dtypes_lib.float32, shapes=(1, 2))
231 4, dtypes=dtypes_lib.float32, shapes=(1, 1))
249 values = array_ops.placeholder(dtype=dtypes_lib.float32)
253 4, dtypes=dtypes_lib.float32, shapes=(1,))
271 4, dtypes=dtypes_lib.float32, shapes=(1, 2))
280 4, dtypes=dtypes_lib.float32, shapes=(1, 2))
298 values = array_ops.placeholder(dtype=dtypes_lib.float32)
[all …]
Dmetric_ops_large_test.py24 from tensorflow.python.framework import dtypes as dtypes_lib unknown
41 shape, 0.0, 1.0, dtype=dtypes_lib.float32)
/external/tensorflow/tensorflow/compiler/tests/
Dfifo_queue_test.py26 from tensorflow.python.framework import dtypes as dtypes_lib unknown
35 q = data_flow_ops.FIFOQueue(10, dtypes_lib.float32)
41 q = data_flow_ops.FIFOQueue(10, dtypes_lib.float32, shapes=(3, 2))
50 q = data_flow_ops.FIFOQueue(10, [dtypes_lib.int32], shapes=[()])
59 q = data_flow_ops.FIFOQueue(10, [dtypes_lib.int32], shapes=[()])
61 q2 = data_flow_ops.FIFOQueue(10, [dtypes_lib.int32], shapes=[()])
68 q = data_flow_ops.FIFOQueue(10, dtypes_lib.float32)
74 q = data_flow_ops.FIFOQueue(10, dtypes_lib.float32)
99 q = data_flow_ops.FIFOQueue(10, dtypes_lib.float32)
123 q = data_flow_ops.FIFOQueue(10, dtypes_lib.float32)
[all …]
/external/tensorflow/tensorflow/contrib/slim/python/slim/data/
Dparallel_reader_test.py24 from tensorflow.python.framework import dtypes as dtypes_lib unknown
134 dtypes=[dtypes_lib.string, dtypes_lib.string])
139 capacity=256, dtypes=[dtypes_lib.string, dtypes_lib.string])
146 dtypes=[dtypes_lib.string, dtypes_lib.string],
153 dtypes=[dtypes_lib.string, dtypes_lib.string],
/external/tensorflow/tensorflow/contrib/training/python/training/
Dbucket_ops_test.py25 from tensorflow.python.framework import dtypes as dtypes_lib unknown
65 self.scalar_int_feed = array_ops.placeholder(dtypes_lib.int32, ())
66 self.unk_int64_feed = array_ops.placeholder(dtypes_lib.int64, (None,))
67 self.vec3_str_feed = array_ops.placeholder(dtypes_lib.string, (3,))
78 dtypes=[dtypes_lib.int32, dtypes_lib.int64, dtypes_lib.string],
338 lengths = array_ops.placeholder(dtypes_lib.int32, ())
339 data = array_ops.placeholder(dtypes_lib.int64, (data_len,))
340 labels = array_ops.placeholder(dtypes_lib.string, (labels_len,))
349 5000, (dtypes_lib.int32, dtypes_lib.int64, dtypes_lib.string), (
Dsampling_ops_threading_test.py22 from tensorflow.python.framework import dtypes as dtypes_lib unknown
42 dtypes_lib.int32)
/external/tensorflow/tensorflow/contrib/learn/python/learn/learn_io/
Dgraph_io_test.py31 from tensorflow.python.framework import dtypes as dtypes_lib unknown
207 parsing_ops.FixedLenFeature(shape=shape, dtype=dtypes_lib.float32)
375 features = {"sequence": parsing_ops.FixedLenFeature([], dtypes_lib.string)}
430 features = {"sequence": parsing_ops.FixedLenFeature([], dtypes_lib.string)}
718 dtypes = {"age": parsing_ops.FixedLenFeature([1], dtypes_lib.int64)}
768 features = {"age": parsing_ops.FixedLenFeature([], dtypes_lib.int64)}
829 shape=shape, dtype=dtypes_lib.float32)