HomeSort by relevance Sort by last modified time
    Searched full:as_dtype (Results 1 - 25 of 68) sorted by null

1 2 3

  /external/tensorflow/tensorflow/python/framework/
dtypes_test.py 51 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...]
op_def_library.py 59 (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
386 default_type_attr_map[key] = dtypes.as_dtype(
449 dtype = dtypes.as_dtype(dtype).base_dtype
518 (dtypes.as_dtype(dtype).name, input_arg.name, op_type_name,
533 (prefix, dtypes.as_dtype(input_arg.type).name))
545 (prefix, dtypes.as_dtype(attrs[input_arg.type_attr]).name,
621 ", ".join(dtypes.as_dtype(x).name for x in attr_value),
622 ", ".join(dtypes.as_dtype(x).nam
    [all...]
tensor_spec.py 52 self._dtype = dtypes.as_dtype(dtype)
168 dtype = dtypes.as_dtype(spec.dtype)
dtypes.py 62 The `tf.as_dtype()` function converts numpy types and string type
71 `tf.as_dtype()` function.
265 other = as_dtype(other)
274 dtype = as_dtype(other).as_datatype_enum
650 @tf_export("as_dtype")
651 def as_dtype(type_value): function
framework_lib.py 28 @@as_dtype
  /external/tensorflow/tensorflow/docs_src/api_guides/python/
framework.md 15 * @{tf.as_dtype}
  /external/tensorflow/tensorflow/contrib/remote_fused_graph/pylib/python/ops/
remote_fused_graph_ops.py 53 type_shape_proto.dtype = int(dtypes.as_dtype(type_shape[0]))
59 type_shape_proto.dtype = int(dtypes.as_dtype(type_shape[0]))
  /external/tensorflow/tensorflow/compiler/tests/
dynamic_stitch_test.py 35 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
reverse_sequence_op_test.py 39 p = array_ops.placeholder(dtypes.as_dtype(x.dtype))
40 lengths = array_ops.placeholder(dtypes.as_dtype(seq_lengths.dtype))
ternary_ops_test.py 35 pa = array_ops.placeholder(dtypes.as_dtype(a.dtype), a.shape, name="a")
36 pb = array_ops.placeholder(dtypes.as_dtype(b.dtype), b.shape, name="b")
37 pc = array_ops.placeholder(dtypes.as_dtype(c.dtype), c.shape, name="c")
xla_test.py 57 dtypes.as_dtype(types_pb2.DataType.Value(name))
112 dtypes.as_dtype(types_pb2.DataType.Value(name)).as_numpy_dtype
119 tf_types = set([dtypes.as_dtype(t)
136 tf_types = set([dtypes.as_dtype(t)
argminmax_test.py 43 dtypes.as_dtype(inp.dtype), inp.shape, name="a")
fft_test.py 76 dtypes.as_dtype(data.dtype), shape=data.shape)
98 dtypes.as_dtype(data.dtype), shape=data.shape)
cholesky_op_test.py 54 dtypes.as_dtype(x.dtype), shape=x.shape)