Home
last modified time | relevance | path

Searched refs:expected_dtype (Results 1 – 15 of 15) sorted by relevance

/external/tensorflow/tensorflow/python/ops/ragged/
Dconvert_to_tensor_or_ragged_tensor_op_test.py82 expected_dtype=dtypes.int32),
85 expected_dtype=dtypes.string),
90 expected_dtype=dtypes.float32),
95 expected_dtype=dtypes.float32),
100 expected_dtype=dtypes.int32),
106 expected_dtype=None): argument
107 if expected_dtype is None:
108 expected_dtype = value.dtype if dtype is None else dtype
112 self.assertEqual(dtypes.as_dtype(expected_dtype), converted.dtype)
173 expected_dtype=dtypes.int32),
[all …]
Dragged_constant_value_op_test.py122 dict(pylist=[], expected_dtype=np.float64),
123 dict(pylist=[[[], [[[]], []]]], expected_dtype=np.float64),
124 dict(pylist=[[1, 2], [3], [4, 5, 6]], expected_dtype=np.int64),
125 dict(pylist=[[1., 2.], [], [4., 5., 6.]], expected_dtype=np.float64),
126 dict(pylist=[[1, 2], [3.], [4, 5, 6]], expected_dtype=np.float64),
127 dict(pylist=[[b'a', b'b'], [b'c']], expected_dtype=np.dtype('S1')),
128 dict(pylist=[[True]], expected_dtype=np.bool),
149 expected_dtype=None): argument
157 if expected_dtype is not None:
158 self.assertEqual(rt.dtype, expected_dtype)
Dragged_const_op_test.py119 dict(pylist=[], expected_dtype=dtypes.float32),
120 dict(pylist=[[[], [[[]], []]]], expected_dtype=dtypes.float32),
121 dict(pylist=[[1, 2], [3], [4, 5, 6]], expected_dtype=dtypes.int32),
122 dict(pylist=[[1., 2.], [], [4., 5., 6.]], expected_dtype=dtypes.float32),
123 dict(pylist=[[1, 2], [3.], [4, 5, 6]], expected_dtype=dtypes.float32),
124 dict(pylist=[[b'a', b'b'], [b'c']], expected_dtype=dtypes.string),
125 dict(pylist=[[True]], expected_dtype=dtypes.bool),
145 expected_dtype=None): argument
168 if expected_dtype is not None:
169 self.assertEqual(rt.dtype, expected_dtype)
/external/tensorflow/tensorflow/python/keras/mixed_precision/experimental/
Dtest_util.py26 def create_identity_with_grad_check_fn(expected_gradient, expected_dtype=None): argument
48 if expected_dtype:
49 assert dx.dtype == expected_dtype, (
50 'dx.dtype should be %s but is: %s' % (expected_dtype, dx.dtype))
Dkeras_test.py352 expected_dtype=dtypes.float16,
/external/tensorflow/tensorflow/python/training/
Dtraining_util_test.py33 def _assert_global_step(self, global_step, expected_dtype=dtypes.int64): argument
35 self.assertEqual(expected_dtype, global_step.dtype.base_dtype)
84 training_util.get_global_step(), expected_dtype=dtypes.int32)
86 training_util.get_global_step(g), expected_dtype=dtypes.int32)
/external/tensorflow/tensorflow/core/kernels/data/
Ddataset_test_base.cc242 DataType expected_dtype = RemoveRefType(input_types[inputs->size()]); in AddDatasetInput() local
243 if (expected_dtype != dtype) { in AddDatasetInput()
245 " , but expected: ", expected_dtype); in AddDatasetInput()
/external/tensorflow/tensorflow/cc/ops/
Dconst_op_test.cc38 void ExpectTypeAndShape(const Node* n, DataType expected_dtype, in ExpectTypeAndShape() argument
45 EXPECT_EQ(dtype, expected_dtype); in ExpectTypeAndShape()
/external/tensorflow/tensorflow/core/framework/
Dtensor.cc620 void Tensor::CheckType(DataType expected_dtype) const { in CheckType()
621 CHECK_EQ(dtype(), expected_dtype) << " " in CheckType()
622 << DataTypeString(expected_dtype) << " expected, got " in CheckType()
626 void Tensor::CheckTypeAndIsAligned(DataType expected_dtype) const { in CheckTypeAndIsAligned()
627 CHECK_EQ(dtype(), expected_dtype) << " " in CheckTypeAndIsAligned()
628 << DataTypeString(expected_dtype) << " expected, got " in CheckTypeAndIsAligned()
Dtensor.h593 void CheckType(DataType expected_dtype) const;
594 void CheckTypeAndIsAligned(DataType expected_dtype) const;
/external/tensorflow/tensorflow/contrib/framework/python/ops/
Dvariables_test.py187 def _assert_global_step(self, global_step, expected_dtype=dtypes.int64): argument
189 self.assertEquals(expected_dtype, global_step.dtype.base_dtype)
238 variables_lib2.get_global_step(), expected_dtype=dtypes.int32)
240 variables_lib2.get_global_step(g), expected_dtype=dtypes.int32)
/external/tensorflow/tensorflow/python/data/ops/
Ddataset_ops.py466 for (ret_array, expected_dtype, expected_shape) in zip(
468 if ret_array.dtype != expected_dtype.as_numpy_dtype:
472 expected_dtype.as_numpy_dtype))
/external/tensorflow/tensorflow/python/framework/
Dtest_util.py2431 def assertDTypeEqual(self, target, expected_dtype): argument
2443 self.assertEqual(arr.dtype, expected_dtype)
/external/tensorflow/tensorflow/python/ops/
Dimage_ops_test.py2387 expected_dtype = image.dtype
2389 expected_dtype = dtypes.float32
2390 self.assertEqual(y.dtype, expected_dtype)
2883 expected_dtype = image.dtype
2885 expected_dtype = dtypes.float32
2886 self.assertEqual(y.dtype, expected_dtype)
/external/tensorflow/tensorflow/contrib/learn/python/learn/estimators/
Destimator_test.py1232 def _assert_single_feature_column(self, expected_shape, expected_dtype, argument
1240 shape=expected_shape, dtype=expected_dtype)