Home
last modified time | relevance | path

Searched refs:Tensor (Results 1 – 25 of 60) sorted by relevance

123

/llvm-project-15.0.7/mlir/test/Integration/Dialect/SparseTensor/taco/
H A Dunit_test_tensor_core.py278 A = mlir_pytaco.Tensor([I, J])
281 B = mlir_pytaco.Tensor([I, J])
313 B = mlir_pytaco.Tensor([I])
329 A = mlir_pytaco.Tensor([4])
330 B = mlir_pytaco.Tensor([4])
331 C = mlir_pytaco.Tensor([4])
426 B = mlir_pytaco.Tensor([J])
427 C = mlir_pytaco.Tensor([I])
461 C = mlir_pytaco.Tensor([K])
462 D = mlir_pytaco.Tensor([L])
[all …]
H A Dunit_test_tensor_io.py112 a = mlir_pytaco.Tensor([2, 3])
128 a = mlir_pytaco.Tensor([2, 3])
132 b = mlir_pytaco.Tensor([2, 3])
/llvm-project-15.0.7/mlir/test/Integration/Dialect/SparseTensor/taco/tools/
H A Dmlir_pytaco_io.py24 Tensor = mlir_pytaco.Tensor variable
35 dtype: DType = DType(Type.FLOAT32)) -> Tensor:
57 return Tensor.from_file(filename, fmt, dtype)
60 def write(filename: str, tensor: Tensor) -> None: argument
77 if not isinstance(tensor, Tensor):
H A Dmlir_pytaco_api.py22 from_array = mlir_pytaco.Tensor.from_array
33 tensor = mlir_pytaco.Tensor
/llvm-project-15.0.7/mlir/lib/Dialect/Tensor/IR/
H A DCMakeLists.txt13 ${PROJECT_SOURCE_DIR}/include/mlir/Dialect/Tensor
39 ${MLIR_MAIN_INCLUDE_DIR}/mlir/Dialect/Tensor
53 ${MLIR_MAIN_INCLUDE_DIR}/mlir/Dialect/Tensor
/llvm-project-15.0.7/mlir/test/Target/Cpp/
H A Dstdops.mlir17 // CPP-DEFAULT-NEXT: Tensor<int32_t> [[V3:[^ ]*]] = {0};
18 // CPP-DEFAULT-NEXT: Tensor<size_t, 2> [[V4:[^ ]*]] = {0, 1};
19 // CPP-DEFAULT-NEXT: Tensor<float, 2, 2> [[V5:[^ ]*]] = {(float)0.0e+00, (float)1.000000000e+00, (f…
25 // CPP-DECLTOP-NEXT: Tensor<int32_t> [[V3:[^ ]*]];
26 // CPP-DECLTOP-NEXT: Tensor<size_t, 2> [[V4:[^ ]*]];
27 // CPP-DECLTOP-NEXT: Tensor<float, 2, 2> [[V5:[^ ]*]];
H A Dcommon-cpp.mlir60 // CHECK: Tensor<float, 2> mixed_types(Tensor<double, 2> [[V1]])
67 // CHECK: Tensor<uint64_t> mhlo_convert(Tensor<uint32_t> [[V1]])
/llvm-project-15.0.7/mlir/include/mlir/Dialect/SparseTensor/IR/
H A DSparseTensorBase.td31 to tensor algebra by [Kjolstad17,Kjolstad20] in the Sparse Tensor
60 Format Abstraction for Sparse Tensor Algebra Compilers. Proceedings of
66 Lugato, and Saman Amarasinghe. The Tensor Algebra Compiler. Proceedings of
69 and Saman Amarasinghe. Tensor Algebra Compilation with Workspaces,
72 * [Kjolstad20] Fredrik Berg Kjolstad. Sparse Tensor Algebra Compilation.
/llvm-project-15.0.7/mlir/include/mlir/Dialect/Tosa/IR/
H A DTosaTypesBase.td98 // Tensor types
117 // Tensor types with constrained ranks.
195 def Tosa_TensorTypeAttr : TypeAttrBase<"TensorType", "Tensor type attribute">;
197 // Tensor to buffer types.
/llvm-project-15.0.7/mlir/lib/CAPI/Dialect/
H A DTensor.cpp13 MLIR_DEFINE_CAPI_DIALECT_REGISTRATION(Tensor, tensor,
H A DCMakeLists.txt102 Tensor.cpp
/llvm-project-15.0.7/mlir/include/mlir-c/Dialect/
H A DTensor.h19 MLIR_DECLARE_CAPI_DIALECT_REGISTRATION(Tensor, tensor);
/llvm-project-15.0.7/mlir/lib/Dialect/Tensor/Utils/
H A DCMakeLists.txt5 ${MLIR_MAIN_INCLUDE_DIR}/mlir/Dialect/Tensor
/llvm-project-15.0.7/mlir/include/mlir/Dialect/Tensor/Transforms/
H A DCMakeLists.txt2 mlir_tablegen(Passes.h.inc -gen-pass-decls -name Tensor)
/llvm-project-15.0.7/mlir/lib/Dialect/Tensor/Transforms/
H A DCMakeLists.txt8 ${MLIR_MAIN_INCLUDE_DIR}/mlir/Dialect/Tensor/Transforms
/llvm-project-15.0.7/mlir/test/lib/Dialect/
H A DCMakeLists.txt11 add_subdirectory(Tensor)
/llvm-project-15.0.7/mlir/python/mlir/dialects/
H A DTensorOps.td13 include "mlir/Dialect/Tensor/IR/TensorOps.td"
/llvm-project-15.0.7/mlir/include/mlir/Dialect/
H A DCMakeLists.txt30 add_subdirectory(Tensor)
/llvm-project-15.0.7/mlir/lib/Dialect/
H A DCMakeLists.txt30 add_subdirectory(Tensor)
/llvm-project-15.0.7/mlir/include/mlir/IR/
H A DTensorEncoding.td1 //===- TensorEncoding.td - Tensor encoding interfaces ------*- tablegen -*-===//
/llvm-project-15.0.7/mlir/include/mlir/Dialect/Arithmetic/IR/
H A DArithmeticOps.td196 // Tensor element-wise addition.
246 // Tensor element-wise integer division.
275 // Tensor element-wise integer division.
373 // Tensor element-wise integer division remainder.
430 // Tensor element-wise bitwise integer and.
459 // Tensor element-wise bitwise integer or.
488 // Tensor element-wise bitwise integer xor.
585 // Tensor element-wise negation value.
613 // Tensor addition.
644 // Tensor subtraction.
[all …]
/llvm-project-15.0.7/mlir/include/mlir/Dialect/Math/IR/
H A DMathBase.td28 // Tensor elementwise absolute value.
/llvm-project-15.0.7/mlir/lib/Dialect/Shape/IR/
H A DShapeCanonicalization.td3 include "mlir/Dialect/Tensor/IR/TensorOps.td"
/llvm-project-15.0.7/mlir/docs/Dialects/
H A DTOSA.md1 # Tensor Operator Set Architecture (TOSA) Dialect
17 ## TOSA and Tensor Level Expressiveness
/llvm-project-15.0.7/mlir/docs/Rationale/
H A DRationaleLinalgDialect.md22 working group for discussing the [Development of high-level Tensor Compute
27 path to defining these High-Level Tensor Compute Primitives.
127 - [Tensor Comprehensions](https://arxiv.org/abs/1802.04730),
272 ### Lessons from Tensor Comprehensions<a name="lessonstc"></a>
273 [Tensor Comprehensions](https://arxiv.org/abs/1802.04730) is a
288 Tensor Comprehensions showed that, fixing a few predefined strategies
296 The ultimate goal of Tensor Comprehensions was to concretely mix
302 - Halide was never properly used in Tensor Comprehensions beyond shape
309 - Tensor Comprehensions emitted CUDA code which was then JIT compiled
328 [PolyMage](http://mcl.csa.iisc.ac.in/polymage.html) and [Tensor
[all …]

123