Searched refs:DType (Results 1 - 14 of 14) sorted by relevance

/external/tensorflow/tensorflow/python/framework/
H A Ddtypes.py29 @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 Ddtypes_test.py45 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 Dfunction.py95 *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 Dops_test.py427 # 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 Dpartitioned_variables.py116 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 Dio_ops.py153 @tc.accepts(dtypes.DType, tc.Collection(tc.Union(string_types, core.AxisLike)),
H A Dops.py1084 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 Dcore.py493 tc.register_type_abbreviation(dtypes.DType, 'tensorflow.DType')
/external/tensorflow/tensorflow/python/util/
H A Dpy_checkpoint_reader.i145 return {name: dtypes.DType(type_enum)
/external/tensorflow/tensorflow/contrib/data/python/ops/
H A Dbatching.py229 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 Dsignature_def_utils_impl.py280 """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 DRegTests.pm820 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 Dfunction.py416 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 Dresource_variable_ops_test.py383 self.assertIsInstance(w.dtype, dtypes.DType)

Completed in 969 milliseconds