Home
last modified time | relevance | path

Searched refs:numpy_dtype (Results 1 – 9 of 9) sorted by relevance

/external/tensorflow/tensorflow/compiler/xla/python_api/
Dtypes.py47 numpy_dtype=dtypes.bfloat16.as_numpy_dtype,
53 numpy_dtype=_np.float16,
59 numpy_dtype=_np.float32,
65 numpy_dtype=_np.float64,
71 numpy_dtype=_np.int8,
77 numpy_dtype=_np.int16,
83 numpy_dtype=_np.int32,
89 numpy_dtype=_np.int64,
95 numpy_dtype=_np.uint8,
101 numpy_dtype=_np.uint16,
[all …]
Dxla_literal.py40 type_record.numpy_dtype)
60 dtype=type_record.numpy_dtype)
/external/tensorflow/tensorflow/python/framework/
Ddtypes_test.py60 numpy_dtype = dtype.as_numpy_dtype
61 _ = np.empty((1, 1, 1, 1), dtype=numpy_dtype)
66 dtypes.as_dtype(numpy_dtype))
250 numpy_dtype = dtype.as_numpy_dtype
263 if numpy_dtype == np.bool_:
266 if numpy_dtype == np.int8:
269 if numpy_dtype == np.int16:
272 if numpy_dtype == np.int32:
275 if numpy_dtype == np.int64:
278 if numpy_dtype == np.uint8:
[all …]
Dtensor_util.py480 numpy_dtype = dtypes.as_dtype(nparray.dtype)
481 if numpy_dtype is None:
487 numpy_dtype = dtype
490 dtype.base_dtype != numpy_dtype.base_dtype):
522 dtype=numpy_dtype.as_datatype_enum,
525 if is_same_size and numpy_dtype in _TENSOR_CONTENT_TYPES and shape_size > 1:
537 if numpy_dtype == dtypes.string and not isinstance(values, np.ndarray):
563 "Element type not supported in TensorProto: %s" % numpy_dtype.name)
/external/flatbuffers/python/flatbuffers/
Dtable.py113 numpy_dtype = N.to_numpy_type(flags)
114 return encode.GetVectorAsNumpy(numpy_dtype, self.Bytes, length, offset)
/external/tensorflow/tensorflow/python/kernel_tests/
Dconstant_op_eager_test.py414 numpy_dtype = np.string_
416 numpy_dtype = dtype.as_numpy_dtype
417 d = constant_op.constant(np.ones((2, 3), dtype=numpy_dtype), dtype=dtype)
527 numpy_dtype = dtype.as_numpy_dtype
529 d = constant_op.constant(np.ones((2, 3), dtype=numpy_dtype), dtype=dtype)
Dconstant_op_test.py503 numpy_dtype = np.string_
505 numpy_dtype = dtype.as_numpy_dtype
508 np.ones((2, 3), dtype=numpy_dtype), dtype=dtype)
522 feed_dict[d] = np.ones((2, 3), dtype=numpy_dtype)
687 numpy_dtype = dtype.as_numpy_dtype
692 (2, 3), dtype=numpy_dtype), dtype=dtype)
/external/tensorflow/tensorflow/python/client/
Dsession.py141 def _convert_to_numpy_obj(numpy_dtype, obj): argument
143 return numpy_dtype(obj) if numpy_dtype is not object else str(obj)
/external/tensorflow/tensorflow/python/ops/
Dnn_batchnorm_test.py306 numpy_dtype = dtype.as_numpy_dtype
308 x_val = np.random.random_sample(x_shape).astype(numpy_dtype)