Searched refs:linearizeIndex (Results 1 – 4 of 4) sorted by relevance
| /llvm-project-15.0.7/mlir/lib/Bindings/Python/ |
| H A D | PybindUtils.h | 233 intptr_t linearizeIndex(intptr_t index) {
253 static_cast<Derived *>(this)->getRawElement(linearizeIndex(index)));
284 return static_cast<Derived *>(this)->getRawElement(linearizeIndex(index));
|
| /llvm-project-15.0.7/mlir/include/mlir/Dialect/SPIRV/Transforms/ |
| H A D | SPIRVConversion.h | 153 Value linearizeIndex(ValueRange indices, ArrayRef<int64_t> strides,
|
| /llvm-project-15.0.7/mlir/lib/Conversion/TensorToSPIRV/ |
| H A D | TensorToSPIRV.cpp | 83 Value index = spirv::linearizeIndex(adaptor.getIndices(), strides, in matchAndRewrite()
|
| /llvm-project-15.0.7/mlir/lib/Dialect/SPIRV/Transforms/ |
| H A D | SPIRVConversion.cpp | 760 Value mlir::spirv::linearizeIndex(ValueRange indices, ArrayRef<int64_t> strides, in linearizeIndex() function in mlir::spirv
809 linearizeIndex(indices, strides, offset, indexType, loc, builder)); in getElementPtr()
|