Searched refs:expected_types (Results 1 – 7 of 7) sorted by relevance
| /external/tensorflow/tensorflow/python/data/experimental/kernel_tests/ |
| D | unbatch_test.py | 50 expected_types = (dtypes.int32,) * 3
52 self.assertEqual(expected_types, dataset_ops.get_legacy_output_types(data))
54 self.assertEqual(expected_types, dataset_ops.get_legacy_output_types(data))
62 expected_types = (dtypes.int32, dtypes.string, dtypes.int32)
64 self.assertEqual(expected_types, dataset_ops.get_legacy_output_types(data))
66 self.assertEqual(expected_types, dataset_ops.get_legacy_output_types(data))
101 expected_types = ((dtypes.int32,),) * 3
103 self.assertEqual(expected_types, dataset_ops.get_legacy_output_types(data))
105 self.assertEqual(expected_types, dataset_ops.get_legacy_output_types(data))
113 expected_types = ((dtypes.int32, dtypes.string),) * 3
[all …]
|
| /external/tensorflow/tensorflow/core/kernels/data/experimental/ |
| D | indexed_dataset_op.cc | 378 auto expected_types = materialized_resource->output_dtypes(); in Compute() local
386 OP_REQUIRES(ctx, out_tensors[i].dtype() == expected_types[i], in Compute()
389 expected_types[i], in Compute()
|
| /external/tensorflow/tensorflow/python/debug/wrappers/ |
| D | framework.py | 132 def _check_type(obj, expected_types): argument
143 if not isinstance(obj, expected_types):
145 (expected_types, type(obj)))
|
| /external/protobuf/python/google/protobuf/pyext/ |
| D | message.h | 334 void FormatTypeError(PyObject* arg, char* expected_types);
|
| D | message.cc | 536 void FormatTypeError(PyObject* arg, char* expected_types) { in FormatTypeError() argument
543 expected_types); in FormatTypeError()
|
| /external/tensorflow/tensorflow/python/data/util/ |
| D | structure_test.py | 64 def testFlatStructure(self, value_fn, expected_structure, expected_types, argument
69 self.assertEqual(expected_types, s._flat_types)
|
| /external/tensorflow/tensorflow/contrib/meta_graph_transform/ |
| D | meta_graph_transform_test.py | 184 expected_types = [types[0], types[2]]
200 self.assertEqual(save_op.attr['dtypes'].list.type[:], expected_types)
|