Home
last modified time | relevance | path

Searched refs:as_dtype (Results 1 – 25 of 122) sorted by relevance

12345

/external/tensorflow/tensorflow/python/framework/
Ddtypes_test.py52 dt = dtypes.as_dtype(datatype_enum)
59 dtype = dtypes.as_dtype(datatype_enum)
65 dtypes.as_dtype(datatype_enum).base_dtype,
66 dtypes.as_dtype(numpy_dtype))
73 self.assertEqual(dtypes.as_dtype(datatype_enum), dtype)
79 dtypes.as_dtype(types_pb2.DT_INVALID)
82 self.assertIs(dtypes.float32, dtypes.as_dtype(np.float32))
83 self.assertIs(dtypes.float64, dtypes.as_dtype(np.float64))
84 self.assertIs(dtypes.int32, dtypes.as_dtype(np.int32))
85 self.assertIs(dtypes.int64, dtypes.as_dtype(np.int64))
[all …]
Dop_def_library.py62 (param_name, dtypes.as_dtype(dtype).name,
63 ", ".join(dtypes.as_dtype(x).name for x in allowed_list)))
198 v = dtypes.as_dtype(v).base_dtype
367 default_type_attr_map[key] = dtypes.as_dtype(
432 dtype = dtypes.as_dtype(dtype).base_dtype
530 (dtypes.as_dtype(dtype).name, input_arg.name, op_type_name,
545 (prefix, dtypes.as_dtype(input_arg.type).name))
557 (prefix, dtypes.as_dtype(attrs[input_arg.type_attr]).name,
623 (input_name, op_type_name, dtypes.as_dtype(attr_value).name,
624 dtypes.as_dtype(attrs[input_arg.type_attr]).name,
[all …]
Dindexed_slices.py231 self._values_dtype = dtypes.as_dtype(dtype)
232 self._indices_dtype = dtypes.as_dtype(indices_dtype)
236 self._dense_shape_dtype = dtypes.as_dtype(dense_shape_dtype)
324 if dtype and not dtypes.as_dtype(dtype).is_compatible_with(value.dtype):
327 (dtypes.as_dtype(dtype).name, value.dtype.name, str(value)))
Ddtypes.py188 other = as_dtype(other)
199 other = as_dtype(other)
216 return as_dtype, (self.name,)
607 def as_dtype(type_value): function
/external/tensorflow/tensorflow/compiler/tests/
Deinsum_op_test.py38 dtypes.as_dtype(inp.dtype), inp.shape, name='a')
49 pa = array_ops.placeholder(dtypes.as_dtype(a.dtype), a.shape, name='a')
50 pb = array_ops.placeholder(dtypes.as_dtype(b.dtype), b.shape, name='b')
Dreverse_sequence_op_test.py39 p = array_ops.placeholder(dtypes.as_dtype(x.dtype))
40 lengths = array_ops.placeholder(dtypes.as_dtype(seq_lengths.dtype))
Dxla_test.py75 dtypes.as_dtype(types_pb2.DataType.Value(name)).as_numpy_dtype
100 dtypes.as_dtype(types_pb2.DataType.Value(name))
163 tf_types = set([dtypes.as_dtype(t)
185 tf_types = set([dtypes.as_dtype(t)
Ddynamic_stitch_test.py35 array_ops.placeholder(dtypes.as_dtype(arg.dtype)) for arg in indices
38 array_ops.placeholder(dtypes.as_dtype(arg.dtype)) for arg in data
Dcomplex_div_test.py41 pa = array_ops.placeholder(dtypes.as_dtype(a.dtype), a.shape, name="a")
42 pb = array_ops.placeholder(dtypes.as_dtype(b.dtype), b.shape, name="b")
Ddata_format_ops_test.py37 placeholder = array_ops.placeholder(dtypes.as_dtype(x.dtype), x.shape)
88 placeholder = array_ops.placeholder(dtypes.as_dtype(x.dtype), x.shape)
Dternary_ops_test.py39 pa = array_ops.placeholder(dtypes.as_dtype(a.dtype), a.shape, name="a")
40 pb = array_ops.placeholder(dtypes.as_dtype(b.dtype), b.shape, name="b")
41 pc = array_ops.placeholder(dtypes.as_dtype(c.dtype), c.shape, name="c")
Dsegment_reduction_ops_test.py176 minval = dtypes.as_dtype(dtype).min
177 maxval = dtypes.as_dtype(dtype).max
/external/tensorflow/tensorflow/python/ops/
Dinit_ops.py109 self.dtype = dtypes.as_dtype(dtype)
129 self.dtype = dtypes.as_dtype(dtype)
221 self.dtype = dtypes.as_dtype(dtype)
263 self.dtype = dtypes.as_dtype(dtype)
303 self.dtype = _assert_float_dtype(dtypes.as_dtype(dtype))
349 self.dtype = _assert_float_dtype(dtypes.as_dtype(dtype))
409 self.dtype = _assert_float_dtype(dtypes.as_dtype(dtype))
492 self.dtype = _assert_float_dtype(dtypes.as_dtype(dtype))
564 self.dtype = _assert_float_dtype(dtypes.as_dtype(dtype))
627 self.dtype = _assert_float_dtype(dtypes.as_dtype(dtype))
[all …]
Dinit_ops_v2.py149 dtype = dtypes.as_dtype(dtype)
195 dtype = dtypes.as_dtype(dtype)
289 dtype = dtypes.as_dtype(dtype)
348 dtype = dtypes.as_dtype(dtype)
1049 dtype = dtypes.as_dtype(dtype)
Dgradients_util.py184 "real or integer" % (dtypes.as_dtype(grad_y.dtype).name, y,
185 dtypes.as_dtype(y.dtype).name))
190 "tensor %s with type %s must be real" % (dtypes.as_dtype(
191 grad_y.dtype).name, y, dtypes.as_dtype(y.dtype).name))
196 "tensor %s with type %s must be variant" % (dtypes.as_dtype(
197 grad_y.dtype).name, y, dtypes.as_dtype(y.dtype).name))
207 "to obtain a default gradient" % (y, dtypes.as_dtype(y.dtype).name))
234 dtype = dtypes.as_dtype(tensor.dtype)
Dparsing_config.py203 dtype = dtypes.as_dtype(dtype)
207 row_splits_dtype = dtypes.as_dtype(row_splits_dtype)
417 self.sparse_types = [dtypes.as_dtype(t) for t in sparse_types]
419 self.dense_types = [dtypes.as_dtype(t) for t in dense_types]
423 self.ragged_value_types = [dtypes.as_dtype(t) for t in ragged_value_types]
424 self.ragged_split_types = [dtypes.as_dtype(t) for t in ragged_split_types]
/external/tensorflow/tensorflow/lite/tools/signature/
Dsignature_def_utils_test.py45 dtype=tf.as_dtype(tf.float32).as_datatype_enum,
49 dtype=tf.as_dtype(tf.float32).as_datatype_enum,
/external/tensorflow/tensorflow/python/keras/mixed_precision/
Dpolicy.py245 dtype = dtypes.as_dtype(name).name
519 not dtypes.as_dtype(policy.compute_dtype).is_floating):
549 dtypes.as_dtype(dtype)
/external/tensorflow/tensorflow/python/ops/ragged/
Dsegment_id_ops.py69 out_type = dtypes.as_dtype(out_type)
111 out_type = dtypes.as_dtype(out_type)
Dconvert_to_tensor_or_ragged_tensor_op_test.py127 self.assertEqual(dtypes.as_dtype(expected_dtype), converted.dtype)
214 self.assertEqual(dtypes.as_dtype(expected_dtype), converted.dtype)
/external/tensorflow/tensorflow/python/eager/
Dbackprop.py91 return dtypes.as_dtype(value)
93 return [dtypes.as_dtype(v) for v in value]
685 if dtypes.as_dtype(dtype).is_bool:
695 as_dtype = dtypes.as_dtype(dtype)
696 if as_dtype == dtypes.string:
702 if as_dtype.is_bool:
Dbackprop_util.py58 dtype = dtypes.as_dtype(dtype)
/external/tensorflow/tensorflow/python/saved_model/model_utils/
Dexport_output.py142 and dtypes.as_dtype(classes.dtype) == dtypes.string)):
164 if dtypes.as_dtype(examples.dtype) != dtypes.string:
197 if dtypes.as_dtype(examples.dtype) != dtypes.string:
/external/tensorflow/tensorflow/python/keras/layers/preprocessing/
Dtable_utils.py241 if dtypes.as_dtype(expected_type) != dtypes.as_dtype(values.dtype):
252 np_dtype = dtypes.as_dtype(dtype).as_numpy_dtype
/external/tensorflow/tensorflow/compiler/mlir/tensorflow/tests/tf_saved_model/
Dno_input_shape_v1.py53 name=x.name, dtype=tf.as_dtype(x.dtype).as_datatype_enum)

12345