| /external/tensorflow/tensorflow/python/framework/ |
| dtypes.py | 22 from tensorflow.core.framework import types_pb2
74 type_enum: A `types_pb2.DataType` enum value.
77 TypeError: If `type_enum` is not a value `types_pb2.DataType`.
83 if (type_enum not in types_pb2.DataType.values() or
84 type_enum == types_pb2.DT_INVALID):
86 "type_enum is not a valid types_pb2.DataType: %s" % type_enum)
124 types_pb2.DT_VARIANT, types_pb2.DT_VARIANT_REF, types_pb2.DT_RESOURCE,
125 types_pb2.DT_RESOURCE_RE
[all...] |
| dtypes_test.py | 23 from tensorflow.core.framework import types_pb2
30 non_numeric_dtypes = [types_pb2.DT_VARIANT,
31 types_pb2.DT_VARIANT_REF,
32 types_pb2.DT_INVALID,
33 types_pb2.DT_RESOURCE,
34 types_pb2.DT_RESOURCE_REF]
41 for datatype_enum in types_pb2.DataType.values():
42 if datatype_enum == types_pb2.DT_INVALID:
48 for datatype_enum in types_pb2.DataType.values():
49 if datatype_enum == types_pb2.DT_INVALID
[all...] |
| op_def_library.py | 28 from tensorflow.core.framework import types_pb2
73 if arg.type != types_pb2.DT_INVALID: num += 1
431 if input_arg.type != types_pb2.DT_INVALID:
475 if input_arg.type != types_pb2.DT_INVALID:
494 if input_arg.type != types_pb2.DT_INVALID:
531 if input_arg.type != types_pb2.DT_INVALID:
577 if input_arg.type != types_pb2.DT_INVALID:
|
| constant_op.py | 48 from tensorflow.core.framework import types_pb2
75 attrs = ("T", attr_t, "index_type", types_pb2.DT_INT32)
|
| importer.py | 26 from tensorflow.core.framework import types_pb2
56 assert arg_def.type != types_pb2.DT_INVALID
64 assert arg_def.type != types_pb2.DT_INVALID
|
| /external/tensorflow/tensorflow/contrib/summary/ |
| summary_ops_test.py | 28 from tensorflow.core.framework import types_pb2
42 types_pb2.DT_HALF: np.float16,
43 types_pb2.DT_FLOAT: np.float32,
44 types_pb2.DT_DOUBLE: np.float64,
45 types_pb2.DT_INT8: np.int8,
46 types_pb2.DT_INT16: np.int16,
47 types_pb2.DT_INT32: np.int32,
48 types_pb2.DT_INT64: np.int64,
49 types_pb2.DT_UINT8: np.uint8,
50 types_pb2.DT_UINT16: np.uint16
[all...] |
| /external/tensorflow/tensorflow/python/saved_model/ |
| signature_def_utils_impl.py | 22 from tensorflow.core.framework import types_pb2
84 if input_tensor_info.dtype != types_pb2.DT_STRING:
89 if output_tensor_info.dtype != types_pb2.DT_FLOAT:
128 if input_tensor_info.dtype != types_pb2.DT_STRING:
135 if classes_tensor_info.dtype != types_pb2.DT_STRING:
141 if scores_tensor_info.dtype != types_pb2.DT_FLOAT:
218 types_pb2.DT_STRING):
225 types_pb2.DT_FLOAT):
240 types_pb2.DT_STRING):
253 != types_pb2.DT_STRING)
[all...] |
| signature_def_utils_test.py | 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
[all...] |
| utils_test.py | 21 from tensorflow.core.framework import types_pb2
36 self.assertEqual(types_pb2.DT_FLOAT, x_tensor_info.dtype)
49 self.assertEqual(types_pb2.DT_FLOAT, x_tensor_info.dtype)
|
| builder_impl.py | 25 from tensorflow.core.framework import types_pb2
213 if tensor_info.dtype is types_pb2.DT_INVALID:
|
| /external/tensorflow/tensorflow/contrib/framework/python/framework/ |
| graph_util_test.py | 23 from tensorflow.core.framework import types_pb2
50 graph_def, ['A'], ['D'], [types_pb2.DT_FLOAT], True, 'FusedOp', 'Op2')
58 [types_pb2.DT_FLOAT])
72 [types_pb2.DT_FLOAT], True, 'FusedOp',
82 [types_pb2.DT_FLOAT])
|
| /external/tensorflow/tensorflow/python/estimator/export/ |
| export_output_test.py | 22 from tensorflow.core.framework import types_pb2
79 dtype_float = types_pb2.DataType.Value("DT_FLOAT")
80 dtype_string = types_pb2.DataType.Value("DT_STRING")
110 dtype_string = types_pb2.DataType.Value("DT_STRING")
145 dtype_float = types_pb2.DataType.Value("DT_FLOAT")
146 dtype_string = types_pb2.DataType.Value("DT_STRING")
185 dtype_float = types_pb2.DataType.Value("DT_FLOAT")
186 dtype_string = types_pb2.DataType.Value("DT_STRING")
|
| /external/tensorflow/tensorflow/contrib/lite/toco/python/ |
| toco_from_protos_test.py | 25 from tensorflow.contrib.lite.toco import types_pb2
51 toco_flags.inference_input_type = types_pb2.FLOAT
52 toco_flags.inference_type = types_pb2.FLOAT
|
| /external/tensorflow/tensorflow/contrib/predictor/ |
| saved_model_predictor_test.py | 26 from tensorflow.core.framework import types_pb2
94 dtype=types_pb2.DT_FLOAT,
98 dtype=types_pb2.DT_FLOAT,
103 dtype=types_pb2.DT_FLOAT,
|
| /external/tensorflow/tensorflow/contrib/learn/python/learn/utils/ |
| saved_model_export_utils_test.py | 32 from tensorflow.core.framework import types_pb2
91 dtype_float = types_pb2.DataType.Value("DT_FLOAT")
92 dtype_string = types_pb2.DataType.Value("DT_STRING")
122 dtype_string = types_pb2.DataType.Value("DT_STRING")
162 dtype_float = types_pb2.DataType.Value("DT_FLOAT")
163 dtype_string = types_pb2.DataType.Value("DT_STRING")
208 dtype_float = types_pb2.DataType.Value("DT_FLOAT")
209 dtype_string = types_pb2.DataType.Value("DT_STRING")
251 dtype_float = types_pb2.DataType.Value("DT_FLOAT")
252 dtype_string = types_pb2.DataType.Value("DT_STRING"
[all...] |
| /external/tensorflow/tensorflow/contrib/all_reduce/python/ |
| all_reduce_test.py | 26 from tensorflow.core.framework import types_pb2
85 types_pb2.DT_FLOAT)
129 var = state_ops.variable_op(shape, types_pb2.DT_FLOAT)
149 x, constant_op.constant(num_devices, dtype=types_pb2.DT_FLOAT))
|
| /external/tensorflow/tensorflow/compiler/tests/ |
| xla_test.py | 28 from tensorflow.core.framework import types_pb2
57 dtypes.as_dtype(types_pb2.DataType.Value(name))
112 dtypes.as_dtype(types_pb2.DataType.Value(name)).as_numpy_dtype
|
| /external/tensorflow/tensorflow/contrib/cloud/python/ops/ |
| bigquery_reader_ops_test.py | 32 from tensorflow.core.framework import types_pb2
174 queue = data_flow_ops.FIFOQueue(8, [types_pb2.DT_STRING], shapes=())
|
| /external/tensorflow/tensorflow/python/training/ |
| sync_replicas_optimizer.py | 21 from tensorflow.core.framework import types_pb2
312 types_pb2.DT_INT32,
|
| /external/tensorflow/tensorflow/contrib/lite/python/ |
| lite.py | 38 from tensorflow.contrib.lite.toco import types_pb2 as _types_pb2
|
| /external/tensorflow/tensorflow/contrib/meta_graph_transform/ |
| meta_graph_transform_test.py | 26 from tensorflow.core.framework import types_pb2
172 types = [types_pb2.DT_INT32, types_pb2.DT_FLOAT, types_pb2.DT_INT32]
|
| /external/tensorflow/tensorflow/python/debug/cli/ |
| tensor_format_test.py | 26 from tensorflow.core.framework import types_pb2
343 dtype=types_pb2.DataType.Value("DT_FLOAT"),
355 dtype=types_pb2.DataType.Value("DT_RESOURCE"),
627 dtype=types_pb2.DataType.Value("DT_FLOAT"),
|
| /external/tensorflow/tensorflow/python/tools/ |
| saved_model_cli.py | 38 from tensorflow.core.framework import types_pb2
159 for (key, value) in types_pb2.DataType.items()}[tensor_info.dtype])
|
| saved_model_cli_test.py | 31 from tensorflow.core.framework import types_pb2
207 ref_tensor_info.dtype = types_pb2.DT_FLOAT_REF
|
| /external/tensorflow/tensorflow/python/debug/lib/ |
| debug_data.py | 31 from tensorflow.core.framework import types_pb2
128 if tensor_proto.dtype == types_pb2.DT_RESOURCE:
[all...] |
