Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/compiler/xla/python_api/
Dtypes.py43 numpy_dtype=dtypes.bfloat16.as_numpy_dtype,
49 numpy_dtype=_np.float16,
55 numpy_dtype=_np.float32,
61 numpy_dtype=_np.float64,
67 numpy_dtype=_np.int8,
73 numpy_dtype=_np.int16,
79 numpy_dtype=_np.int32,
85 numpy_dtype=_np.int64,
91 numpy_dtype=_np.uint8,
97 numpy_dtype=_np.uint16,
[all …]
Dxla_literal.py36 type_record.numpy_dtype)
56 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.py475 numpy_dtype = dtypes.as_dtype(nparray.dtype)
476 if numpy_dtype is None:
482 numpy_dtype = dtype
485 dtype.base_dtype != numpy_dtype.base_dtype):
516 dtype=numpy_dtype.as_datatype_enum,
519 if is_same_size and numpy_dtype in _TENSOR_CONTENT_TYPES and shape_size > 1:
531 if numpy_dtype == dtypes.string and not isinstance(values, np.ndarray):
/external/flatbuffers/python/flatbuffers/
Dtable.py113 numpy_dtype = N.to_numpy_type(flags)
114 return encode.GetVectorAsNumpy(numpy_dtype, self.Bytes, length, offset)
122 numpy_dtype = N.to_numpy_type(flags)
123 return encode.GetVectorAsNumpy(numpy_dtype, self.Bytes, length, off)
/external/pytorch/test/xpu/
Dtest_gemm.py26 numpy_dtype = dtype
28 numpy_dtype = torch.float
46 m.to(numpy_dtype).cpu().numpy()
47 v.to(numpy_dtype).cpu().numpy()
49 m.to(numpy_dtype).cpu().numpy() @ v.to(numpy_dtype).cpu().numpy()
52 res3 += (beta * t).to(numpy_dtype).cpu().numpy()
59 res3 = res3_t.to(numpy_dtype).cpu().numpy()
292 numpy_dtype = dtype if dtype != torch.bfloat16 else torch.float32
346 b1.to(numpy_dtype).cpu().numpy() @ b2.to(numpy_dtype).cpu().numpy()
431 numpy_dtype = dtype if dtype != torch.bfloat16 else torch.float32
[all …]
/external/tensorflow/tensorflow/python/kernel_tests/array_ops/
Dconstant_op_eager_test.py408 numpy_dtype = np.string_
410 numpy_dtype = dtype.as_numpy_dtype
411 d = constant_op.constant(np.ones((2, 3), dtype=numpy_dtype), dtype=dtype)
521 numpy_dtype = dtype.as_numpy_dtype
523 d = constant_op.constant(np.ones((2, 3), dtype=numpy_dtype), dtype=dtype)
Dconstant_op_test.py497 numpy_dtype = np.string_
499 numpy_dtype = dtype.as_numpy_dtype
502 np.ones((2, 3), dtype=numpy_dtype), dtype=dtype)
516 feed_dict[d] = np.ones((2, 3), dtype=numpy_dtype)
681 numpy_dtype = dtype.as_numpy_dtype
686 (2, 3), dtype=numpy_dtype), dtype=dtype)
/external/tensorflow/tensorflow/python/client/
Dsession.py138 def _convert_to_numpy_obj(numpy_dtype, obj): argument
140 return numpy_dtype(obj) if numpy_dtype is not object else str(obj)
/external/tensorflow/tensorflow/python/ops/
Dnn_batchnorm_test.py301 numpy_dtype = dtype.as_numpy_dtype
303 x_val = np.random.random_sample(x_shape).astype(numpy_dtype)
/external/pytorch/test/
Dtest_linalg.py5958 numpy_dtype = dtype
5960 numpy_dtype = torch.float
5978 res3 = alpha * (m.to(numpy_dtype).cpu().numpy() @ v.to(numpy_dtype).cpu().numpy())
5980 res3 += (beta * t).to(numpy_dtype).cpu().numpy()
5987 res3 = res3_t.to(numpy_dtype).cpu().numpy()
6787 numpy_dtype = dtype if dtype != torch.bfloat16 else torch.float32
6835 …b1.to(numpy_dtype).cpu().numpy() @ b2.to(numpy_dtype).cpu().numpy()).to(device=device, dtype=dtype)
6921 numpy_dtype = dtype if dtype != torch.bfloat16 else torch.float32
6930 b1.to(numpy_dtype).cpu().numpy() @ b2.to(numpy_dtype).cpu().numpy()
6941 b1.to(numpy_dtype).cpu().numpy() @ b2.to(numpy_dtype).cpu().numpy()
[all …]
Dtest_sparse_csr.py101 numpy_dtype = dtype
103 numpy_dtype = torch.float
132 res3 = alpha * (m.to(numpy_dtype).cpu().numpy() @ v.to(numpy_dtype).cpu().numpy())
134 res3 += (beta * t).to(numpy_dtype).cpu().numpy()
Dtest_tensor_creation_ops.py136 numpy_dtype = torch_to_numpy_dtype_dict[dtype]
138 np_x = np.array(x, dtype=numpy_dtype)
3661 numpy_dtype = torch_to_numpy_dtype_dict[dtype]
3671 numpy_frombuffer = np.frombuffer(original, dtype=numpy_dtype, count=count, offset=offset)
Dtest_mps.py7001 numpy_dtype = {
7013 ref = (X.to(numpy_dtype).cpu().detach().numpy())
9166 numpy_dtype = dtype
9174 res3 = alpha * (m.to(numpy_dtype).cpu().numpy() @ v.to(numpy_dtype).cpu().numpy())
9176 res3 += (torch.mul(t, beta)).to(numpy_dtype).cpu().numpy()
9207 numpy_dtype = dtype
9211 res2 = alpha * np.outer(m.to(numpy_dtype).cpu().numpy(), v.to(numpy_dtype).cpu().numpy())
9213 res2 += (torch.mul(t, beta)).to(numpy_dtype).cpu().numpy()
Dtest_torch.py10461 numpy_dtype = dtype
10498 …b1.to(numpy_dtype).cpu().numpy() @ b2.to(numpy_dtype).cpu().numpy()).to(device=device, dtype=dtype)
/external/tensorflow/tensorflow/compiler/xla/python/xla_extension/
D__init__.pyi85 def numpy_dtype(self) -> np.dtype: ...