Home
last modified time | relevance | path

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

1234

/external/tensorflow/tensorflow/python/framework/
Ddtypes_test.py51 dt = dtypes.as_dtype(datatype_enum)
58 dtype = dtypes.as_dtype(datatype_enum)
64 dtypes.as_dtype(datatype_enum).base_dtype,
65 dtypes.as_dtype(numpy_dtype))
71 dtypes.as_dtype(types_pb2.DT_INVALID)
74 self.assertIs(dtypes.float32, dtypes.as_dtype(np.float32))
75 self.assertIs(dtypes.float64, dtypes.as_dtype(np.float64))
76 self.assertIs(dtypes.int32, dtypes.as_dtype(np.int32))
77 self.assertIs(dtypes.int64, dtypes.as_dtype(np.int64))
78 self.assertIs(dtypes.uint8, dtypes.as_dtype(np.uint8))
[all …]
Dop_def_library.py59 (param_name, dtypes.as_dtype(dtype).name,
60 ", ".join(dtypes.as_dtype(x).name for x in allowed_list)))
174 v = dtypes.as_dtype(v).base_dtype
402 default_type_attr_map[key] = dtypes.as_dtype(
465 dtype = dtypes.as_dtype(dtype).base_dtype
535 (dtypes.as_dtype(dtype).name, input_arg.name, op_type_name,
550 (prefix, dtypes.as_dtype(input_arg.type).name))
562 (prefix, dtypes.as_dtype(attrs[input_arg.type_attr]).name,
638 ", ".join(dtypes.as_dtype(x).name for x in attr_value),
639 ", ".join(dtypes.as_dtype(x).name
Dtensor_spec.py57 self._dtype = dtypes.as_dtype(dtype)
191 dtype = dtypes.as_dtype(spec.dtype)
Ddtypes.py262 other = as_dtype(other)
271 dtype = as_dtype(other).as_datatype_enum
295 return as_dtype, (self.name,)
698 def as_dtype(type_value): function
/external/tensorflow/tensorflow/python/ops/
Dinit_ops.py110 self.dtype = dtypes.as_dtype(dtype)
131 self.dtype = dtypes.as_dtype(dtype)
238 self.dtype = dtypes.as_dtype(dtype)
282 self.dtype = dtypes.as_dtype(dtype)
324 self.dtype = _assert_float_dtype(dtypes.as_dtype(dtype))
372 self.dtype = _assert_float_dtype(dtypes.as_dtype(dtype))
434 self.dtype = _assert_float_dtype(dtypes.as_dtype(dtype))
518 self.dtype = _assert_float_dtype(dtypes.as_dtype(dtype))
594 self.dtype = _assert_float_dtype(dtypes.as_dtype(dtype))
658 self.dtype = _assert_float_dtype(dtypes.as_dtype(dtype))
[all …]
Dinit_ops_v2.py97 dtype = dtypes.as_dtype(dtype)
116 dtype = dtypes.as_dtype(dtype)
210 dtype = dtypes.as_dtype(dtype)
249 dtype = dtypes.as_dtype(dtype)
764 dtype = dtypes.as_dtype(dtype)
Dgradients_util.py236 "real or integer" % (dtypes.as_dtype(grad_y.dtype).name, y,
237 dtypes.as_dtype(y.dtype).name))
242 "tensor %s with type %s must be real" % (dtypes.as_dtype(
243 grad_y.dtype).name, y, dtypes.as_dtype(y.dtype).name))
248 "tensor %s with type %s must be variant" % (dtypes.as_dtype(
249 grad_y.dtype).name, y, dtypes.as_dtype(y.dtype).name))
259 "to obtain a default gradient" % (y, dtypes.as_dtype(y.dtype).name))
288 dtype = dtypes.as_dtype(dtype)
297 dtype = dtypes.as_dtype(tensor.dtype)
Dstateful_random_ops.py226 dtype = dtypes.as_dtype(dtype)
258 dtype = dtypes.as_dtype(dtype)
Dlookup_ops.py115 self._key_dtype = dtypes.as_dtype(key_dtype)
116 self._value_dtype = dtypes.as_dtype(value_dtype)
340 self._key_dtype = dtypes.as_dtype(key_dtype)
341 self._value_dtype = dtypes.as_dtype(value_dtype)
530 key_dtype = dtypes.as_dtype(key_dtype)
531 value_dtype = dtypes.as_dtype(value_dtype)
Dgradient_checker.py220 t = dtypes.as_dtype(x.dtype)
224 t2 = dtypes.as_dtype(y.dtype)
/external/tensorflow/tensorflow/contrib/remote_fused_graph/pylib/python/ops/
Dremote_fused_graph_ops.py53 type_shape_proto.dtype = dtypes.as_dtype(type_shape[0]).as_datatype_enum
59 type_shape_proto.dtype = dtypes.as_dtype(type_shape[0]).as_datatype_enum
/external/tensorflow/tensorflow/compiler/tests/
Dxla_test.py72 dtypes.as_dtype(types_pb2.DataType.Value(name)).as_numpy_dtype
86 dtypes.as_dtype(types_pb2.DataType.Value(name))
149 tf_types = set([dtypes.as_dtype(t)
171 tf_types = set([dtypes.as_dtype(t)
Dternary_ops_test.py36 pa = array_ops.placeholder(dtypes.as_dtype(a.dtype), a.shape, name="a")
37 pb = array_ops.placeholder(dtypes.as_dtype(b.dtype), b.shape, name="b")
38 pc = array_ops.placeholder(dtypes.as_dtype(c.dtype), c.shape, name="c")
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))
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
Dsegment_reduction_ops_test.py175 minval = dtypes.as_dtype(dtype).min
176 maxval = dtypes.as_dtype(dtype).max
Dargminmax_test.py46 dtypes.as_dtype(op_input.dtype), op_input.shape, name="a")
Dpermute_test.py35 placeholder = array_ops.placeholder(dtypes.as_dtype(x.dtype), x.shape)
Ddynamic_slice_ops_test.py36 array_ops.placeholder(dtypes.as_dtype(arg.dtype), arg.shape)
/external/tensorflow/tensorflow/python/eager/
Dbackprop.py76 return dtypes.as_dtype(value)
78 return [dtypes.as_dtype(v) for v in value]
617 if dtypes.as_dtype(dtype).is_bool:
627 as_dtype = dtypes.as_dtype(dtype)
628 if as_dtype == dtypes.string:
634 if as_dtype.is_bool:
/external/tensorflow/tensorflow/python/saved_model/model_utils/
Dexport_output.py141 and dtypes.as_dtype(classes.dtype) == dtypes.string)):
163 if dtypes.as_dtype(examples.dtype) != dtypes.string:
196 if dtypes.as_dtype(examples.dtype) != dtypes.string:
/external/tensorflow/tensorflow/python/ops/linalg/
Dlinear_operator_test_util.py78 dtype = dtypes.as_dtype(x.dtype)
588 dtype = dtypes.as_dtype(dtype)
656 dtype = dtypes.as_dtype(dtype)
694 dtype = dtypes.as_dtype(dtype)
737 dtype = dtypes.as_dtype(dtype)
792 dtype = dtypes.as_dtype(dtype)
/external/tensorflow/tensorflow/python/ops/ragged/
Dconvert_to_tensor_or_ragged_tensor_op_test.py112 self.assertEqual(dtypes.as_dtype(expected_dtype), converted.dtype)
199 self.assertEqual(dtypes.as_dtype(expected_dtype), converted.dtype)
/external/tensorflow/tensorflow/python/kernel_tests/
Dmatrix_solve_op_test.py61 a_ph = array_ops.placeholder(dtypes.as_dtype(np_type))
62 b_ph = array_ops.placeholder(dtypes.as_dtype(np_type))
/external/tensorflow/tensorflow/contrib/opt/python/training/
Dmultitask_optimizer_wrapper.py136 lambda: array_ops.zeros([], dtype=dtypes.as_dtype(grad.dtype)),

1234