Searched refs:DType (Results 1 - 14 of 14) sorted by relevance
/external/tensorflow/tensorflow/python/framework/ |
H A D | dtypes.py | 29 @tf_export("DType") 30 class DType(object): class in inherits:object 33 The following `DType` objects are defined: 63 names to a `DType` object. 91 """Returns `True` if this `DType` represents a reference type.""" 96 """Returns a reference `DType` based on this `DType`.""" 104 """Returns a non-reference `DType` based on this `DType`.""" 131 """Returns a `numpy.dtype` based on this `DType` [all...] |
H A D | dtypes_test.py | 45 dtypes.DType(datatype_enum).as_datatype_enum) 69 dtypes.DType(types_pb2.DT_INVALID) 287 dtype = dtypes.DType(enum) 291 self.assertEquals(type(dtype2), dtypes.DType)
|
H A D | function.py | 95 *input_types: A list of `tf.DType` 596 dtypes.DType(_.type) 775 output_types = [dtypes.DType(x.type) for x in sig.output_arg]
|
H A D | ops_test.py | 427 # First check that type_val is a DType, because the assertEquals will work 428 # no matter what since DType overrides __eq__ 429 self.assertIsInstance(type_val, dtypes.DType) 433 self.assertTrue(all(isinstance(x, dtypes.DType) for x in type_list_val))
|
/external/tensorflow/tensorflow/python/ops/ |
H A D | partitioned_variables.py | 116 dtype: A `DType`. 123 a `DType`. 129 if not isinstance(dtype, dtypes.DType): 130 raise ValueError("dtype is not a DType: %s" % dtype)
|
/external/tensorflow/tensorflow/contrib/labeled_tensor/python/ops/ |
H A D | io_ops.py | 153 @tc.accepts(dtypes.DType, tc.Collection(tc.Union(string_types, core.AxisLike)),
|
H A D | ops.py | 1084 tc.Optional(dtypes.DType), 1129 tc.Optional(dtypes.DType), tc.Optional(string_types)) 1149 tc.Optional(dtypes.DType), tc.Optional(string_types)) 1169 tc.Optional(dtypes.DType), tc.Optional(string_types))
|
H A D | core.py | 493 tc.register_type_abbreviation(dtypes.DType, 'tensorflow.DType')
|
/external/tensorflow/tensorflow/python/util/ |
H A D | py_checkpoint_reader.i | 145 return {name: dtypes.DType(type_enum)
|
/external/tensorflow/tensorflow/contrib/data/python/ops/ |
H A D | batching.py | 229 if not isinstance(input_dataset.output_types, dtypes.DType): 280 output_types: A nested structure of `tf.DType` objects.
|
/external/tensorflow/tensorflow/python/saved_model/ |
H A D | signature_def_utils_impl.py | 280 """Returns a map of keys to DType objects. 286 Map with corresponding DType objects as values. 289 key: dtypes.DType(tensor_info.dtype) 313 Map from string to DType objects. 337 Map from string to DType objects.
|
/external/abi-compliance-checker/modules/Internals/ |
H A D | RegTests.pm | 820 struct DType 825 $DECL_SPEC int parameterTypeFormat_Safe(struct DType param);"; 827 int parameterTypeFormat_Safe(struct DType param) { return 0; }"; 830 class DType 835 $DECL_SPEC int parameterTypeFormat_Safe(class DType param);"; 837 int parameterTypeFormat_Safe(class DType param) { return 0; }"; 3371 struct DType 3376 $DECL_SPEC int parameterTypeFormat2(struct DType param);"; 3378 int parameterTypeFormat2(struct DType param) { return 0; }"; 3381 union DType [all...] |
/external/tensorflow/tensorflow/python/eager/ |
H A D | function.py | 416 tuple(dtypes_module.DType(x.type) for x in signature.output_arg), 510 tuple(dtypes_module.DType(x.type) for x in signature.output_arg),
|
/external/tensorflow/tensorflow/python/kernel_tests/ |
H A D | resource_variable_ops_test.py | 383 self.assertIsInstance(w.dtype, dtypes.DType)
|
Completed in 969 milliseconds