Lines Matching refs:types_pb2
21 from tensorflow.core.framework import types_pb2
80 self.assertEqual(types_pb2.DT_FLOAT, x_tensor_info_actual.dtype)
88 self.assertEqual(types_pb2.DT_FLOAT, y_tensor_info_actual.dtype)
105 self.assertEqual(types_pb2.DT_STRING, x_tensor_info_actual.dtype)
113 self.assertEqual(types_pb2.DT_FLOAT, y_tensor_info_actual.dtype)
131 self.assertEqual(types_pb2.DT_STRING, x_tensor_info_actual.dtype)
139 self.assertEqual(types_pb2.DT_STRING, classes_tensor_info_actual.dtype)
144 self.assertEqual(types_pb2.DT_FLOAT, scores_tensor_info_actual.dtype)
165 self.assertEqual(types_pb2.DT_STRING, input1_tensor_info_actual.dtype)
169 self.assertEqual(types_pb2.DT_STRING, input2_tensor_info_actual.dtype)
176 self.assertEqual(types_pb2.DT_STRING, output1_tensor_info_actual.dtype)
180 self.assertEqual(types_pb2.DT_STRING, output2_tensor_info_actual.dtype)
