Home
last modified time | relevance | path

Searched defs:dtype (Results 1 – 10 of 10) sorted by relevance

/llvm-project-15.0.7/mlir/test/Integration/Dialect/SparseTensor/taco/tools/
H A Dmlir_pytaco_utils.py51 ty: np.dtype, to_func: _SupportFunc, from_func: _SupportFunc,
52 ty_to_funcs: Dict[np.dtype, Tuple[_SupportFunc, _SupportFunc]]) -> None:
106 def get_support_funcs(ty: np.dtype):
116 dtype: np.dtype, argument
H A Dmlir_pytaco_api.py27 dtype = mlir_pytaco.DType variable
H A Dmlir_pytaco.py144 def _nptype_to_taco_type(ty: np.dtype) -> DType:
520 def dtype(self) -> DType: member in IndexExpr
810 def dtype(self) -> DType: member in IndexVar
1296 def dtype(self) -> DType: member in Tensor
1675 def dtype(self) -> DType: member in Access
1752 def dtype(self) -> DType: member in _UnaryExpr
1813 def dtype(self) -> DType: member in _BinaryExpr
/llvm-project-15.0.7/flang/test/Semantics/
H A Dresolve104.f9011 type dtype(goodDefaultKind, badDefaultKind) type
32 type dtype(dtypeParam) type
45 type dtype(goodDefaultLen, badDefaultLen) type
H A Dresolve105.f905 type dtype(kindParam) type
H A Dspec-expr.f90161 type :: dtype(param) type
178 type :: dtype(param) type
/llvm-project-15.0.7/mlir/python/mlir/runtime/
H A Dnp_to_memref.py48 def make_nd_memref_descriptor(rank, dtype): argument
64 def make_zero_d_memref_descriptor(dtype): argument
/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
/llvm-project-15.0.7/mlir/include/mlir/Dialect/Tosa/Utils/
H A DShapeUtils.h178 Type dtype; member
/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