Home
last modified time | relevance | path

Searched refs:dtype (Results 1 – 25 of 34) sorted by relevance

12

/llvm-project-15.0.7/flang/test/Semantics/
H A Dresolve104.f9021 end type dtype
22 type(dtype) :: v1
23 type(dtype(4, 4)) :: v2
24 type(dtype(99, 4)) :: v3
32 type dtype(dtypeParam) type
37 end type dtype
39 type(dtype) :: v1
40 type(dtype(8)) :: v2
41 type(dtype(343)) :: v3
48 end type dtype
[all …]
H A Dresolve105.f905 type dtype(kindParam) type
41 type(dtype) :: arg
46 type(dtype(*)),intent(inout) :: arg
51 type(dtype(66)) :: arg
56 type(dtype(55)) :: local
H A Dspec-expr.f90161 type :: dtype(param) type
164 end type dtype
178 type :: dtype(param) type
181 end type dtype
/llvm-project-15.0.7/mlir/include/mlir/Dialect/Tosa/Utils/
H A DShapeUtils.h35 ValueKnowledge(bool hasRank, llvm::ArrayRef<int64_t> newSizes, Type dtype) in ValueKnowledge()
36 : hasError(false), hasRank(hasRank), dtype(dtype) { in ValueKnowledge()
54 result.dtype = shapedType.getElementType(); in getKnowledgeFromType()
71 return RankedTensorType::get(llvm::makeArrayRef(sizes), dtype); in getType()
72 return UnrankedTensorType::get(dtype); in getType()
76 return hasRank == rhs.hasRank && sizes == rhs.sizes && dtype == rhs.dtype;
89 if (!lhs || !rhs || lhs.dtype != rhs.dtype) in join()
93 result.dtype = lhs.dtype; in join()
141 if (!rhs || !rhs || lhs.dtype != rhs.dtype) in meet()
145 result.dtype = lhs.dtype; in meet()
[all …]
/llvm-project-15.0.7/mlir/test/python/ir/
H A Darray_attributes.py111 array = np.array([2.0, 1.0, 1.0], dtype=np.float32)
158 zero = np.array(0, dtype=np.uint8)
159 one = np.array(255, dtype=np.uint8)
215 array = np.array([[1, 2, 3], [4, 5, 6]], dtype=np.int16)
228 array = np.array([[1, 2, 3], [4, 5, 6]], dtype=np.uint16)
241 array = np.array([[1, 2, 3], [4, 5, 6]], dtype=np.int16)
254 array = np.array([[1, 2, 3], [4, 5, 6]], dtype=np.uint16)
267 array = np.array([[1, 2, 3], [4, 5, 6]], dtype=np.int32)
293 array = np.array([[1, 2, 3], [4, 5, 6]], dtype=np.int32)
320 array = np.array([[1, 2, 3], [4, 5, 6]], dtype=np.int64)
[all …]
/llvm-project-15.0.7/llvm/test/CodeGen/NVPTX/
H A Dsurf-tex.py262 for geom, vec, dtype, clamp in product(
268 if vec == "v4" and dtype == "b64":
278 dtype=(vec + get_llvm_type(dtype)),
287 dtype=dtype,
613 if vec == "v2" or dtype == "f16":
626 if vec == "v2" and dtype != "f16":
641 dtype=dtype,
653 dtype=dtype,
763 for comp, geom, dtype in product(
784 dtype=dtype),
[all …]
/llvm-project-15.0.7/mlir/python/mlir/runtime/
H A Dnp_to_memref.py28 if dtp is np.dtype(np.complex128):
30 if dtp is np.dtype(np.complex64):
32 if dtp is np.dtype(np.float16):
39 if array.dtype == C128:
41 if array.dtype == C64:
43 if array.dtype == F16:
48 def make_nd_memref_descriptor(rank, dtype): argument
55 ("aligned", ctypes.POINTER(dtype)),
64 def make_zero_d_memref_descriptor(dtype): argument
71 ("aligned", ctypes.POINTER(dtype)),
[all …]
/llvm-project-15.0.7/mlir/test/Integration/Dialect/SparseTensor/taco/tools/
H A Dmlir_pytaco_utils.py23 _SupportFuncLocator = Callable[[np.dtype], Tuple[_SupportFunc, _SupportFunc]]
51 ty: np.dtype, to_func: _SupportFunc, from_func: _SupportFunc, argument
52 ty_to_funcs: Dict[np.dtype, Tuple[_SupportFunc, _SupportFunc]]) -> None: argument
106 def get_support_funcs(ty: np.dtype): argument
116 dtype: np.dtype, argument
138 convert_from = _get_support_func_locator()(dtype)[1]
143 values = ctypes.POINTER(runtime.as_ctype(np.dtype(dtype)))()
184 ctypes.POINTER(runtime.as_ctype(np.dtype(np_values.dtype))))
190 convert_to = _get_support_func_locator()(np_values.dtype.type)[0]
H A Dmlir_pytaco.py1055 dtype = dtype or DType(Type.FLOAT32)
1061 self._dtype = dtype
1195 if array.dtype != np.float32 and array.dtype != np.float64:
1199 dtype=_nptype_to_taco_type(array.dtype.type),
1209 dtype: DType,
1237 tensor = Tensor(shape, fmt, dtype=dtype)
1247 dtype: DType,
1264 tensor = Tensor(shape.tolist(), fmt, dtype=dtype)
1754 return self.a.dtype()
1815 return self.a.dtype()
[all …]
H A Dmlir_pytaco_io.py35 dtype: DType = DType(Type.FLOAT32)) -> Tensor:
57 return Tensor.from_file(filename, fmt, dtype)
H A Dmlir_pytaco_api.py27 dtype = mlir_pytaco.DType variable
/llvm-project-15.0.7/llvm/unittests/Analysis/Inputs/ir2native_x86_64_model/
H A Dsaved_model.pbtxt9 type_attr: "dtype"
16 name: "dtype"
30 name: "dtype"
54 name: "dtype"
156 key: "dtype"
201 key: "dtype"
220 key: "dtype"
259 key: "dtype"
278 key: "dtype"
323 key: "dtype"
[all …]
/llvm-project-15.0.7/mlir/test/Integration/Dialect/SparseTensor/python/
H A Dtest_stress.py51 def dtype_to_irtype(self, dtype: np.dtype) -> ir.Type: argument
54 return self.sctype_to_irtype(dtype.type)
65 def irtype_to_dtype(self, tp: ir.Type) -> np.dtype:
67 return np.dtype(self.irtype_to_sctype(tp))
81 self.dtype_to_irtype(nparray.dtype))
158 np_out = np.zeros(np_arg0.shape, dtype=np_arg0.dtype)
235 np_arg0 = np.arange(size, dtype=tyconv.irtype_to_dtype(f64)).reshape(*shape)
/llvm-project-15.0.7/mlir/test/Integration/Dialect/SparseTensor/taco/
H A Dtest_true_dense_tensor_algebra.py13 A = pt.from_array(np.full([2,3], 1, dtype=np.float64))
14 B = pt.from_array(np.full([2,3], 2, dtype=np.float64))
17 C = pt.tensor([2, 3], dtype=pt.float64, is_dense=True)
H A Dtest_tensor_complex.py16 A = pt.tensor([2, 3], dtype=t)
17 B = pt.tensor([2, 3], dtype=t)
18 C = pt.tensor([2, 3], compressed, dtype=t)
H A Dtest_tensor_types.py20 A = pt.tensor([2, 3], dtype=t)
21 B = pt.tensor([2, 3], dtype=t)
22 C = pt.tensor([2, 3], compressed, dtype=t)
H A Dtest_SpMV.py34 x = pt.from_array(np.full((A.shape[1],), 1, dtype=np.float32))
35 z = pt.from_array(np.full((A.shape[0],), 2, dtype=np.float32))
H A Dtest_MTTKRP.py33 C = pt.from_array(np.full((B.shape[1], 25), 1, dtype=np.float32))
34 D = pt.from_array(np.full((B.shape[2], 25), 2, dtype=np.float32))
H A Dtest_SDDMM.py18 A = pt.from_array(np.full((8, 8), 2.0, dtype=np.float32))
19 B = pt.from_array(np.full((8, 8), 3.0, dtype=np.float32))
/llvm-project-15.0.7/llvm/lib/Analysis/models/
H A Dgen-inline-oz-test-model.py37 tf.TensorSpec(dtype=tf.int64, shape=(), name=key) for key in [
79 tf.TensorSpec(dtype=tf.float32, shape=(), name=key)
85 tf.TensorSpec(dtype=tf.int32, shape=(), name=key)
H A Dgen-regalloc-eviction-test-model.py32 (key, tf.TensorSpec(dtype=tf.int64, shape=(NUM_REGISTERS), name=key))
46 module.var = tf.Variable(0, dtype=tf.int64)
/llvm-project-15.0.7/mlir/test/Integration/Dialect/SparseTensor/python/tools/
H A Dnp_to_sparse_tensor.py39 def sparse_tensor_to_coo_tensor(support_lib, sparse, dtype): argument
65 values = ctypes.POINTER(np.ctypeslib.as_ctypes_type(dtype))()
/llvm-project-15.0.7/mlir/test/python/dialects/
H A Dshape.py26 DenseElementsAttr.get(np.array([10, 20], dtype=np.int64), type=IndexType.get()))
/llvm-project-15.0.7/mlir/test/python/
H A Dexecution_engine.py177 inp_arr_1 = np.array([5, 6, 7], dtype=np.float32)
401 np.dtype(np.complex64))
/llvm-project-15.0.7/mlir/benchmark/python/
H A Dbenchmark_sparse.py97 np_timers_ns = np.array([0], dtype=np.int64)

12