Home
last modified time | relevance | path

Searched refs:getIndexType (Results 1 – 25 of 127) sorted by relevance

123456

/external/tensorflow/tensorflow/compiler/mlir/hlo/lib/Dialect/mhlo/transforms/
Dlhlo_legalize_to_gpu.cc85 loc, rewriter.getIndexType(), in matchAndRewrite()
86 rewriter.getIntegerAttr(rewriter.getIndexType(), size)); in matchAndRewrite()
88 loc, rewriter.getIndexType(), in matchAndRewrite()
89 rewriter.getIntegerAttr(rewriter.getIndexType(), 1)); in matchAndRewrite()
107 loc, rewriter.getIndexType(), in matchAndRewrite()
108 rewriter.getIntegerAttr(rewriter.getIndexType(), 0)); in matchAndRewrite()
111 loc, rewriter.getIndexType(), in matchAndRewrite()
112 rewriter.getIntegerAttr(rewriter.getIndexType(), reduce_dim_size)); in matchAndRewrite()
114 loc, rewriter.getIndexType(), in matchAndRewrite()
115 rewriter.getIntegerAttr(rewriter.getIndexType(), 1)); in matchAndRewrite()
Dtransform_unranked_hlo.cc106 Type indexTy = rewriter.getIndexType(); in matchAndRewrite()
341 {RankedTensorType::kDynamicSize}, builder.getIndexType()); in extendToBroadcastShape()
355 {RankedTensorType::kDynamicSize}, builder.getIndexType()); in createBroadcastToKnownRank()
357 RankedTensorType::get({targeted_rank}, builder.getIndexType()); in createBroadcastToKnownRank()
416 rewriter.getIndexType()); in HandleBroadcastAndOp()
422 rewriter.create<shape::RankOp>(loc, rewriter.getIndexType(), lhs_shape); in HandleBroadcastAndOp()
424 rewriter.create<shape::RankOp>(loc, rewriter.getIndexType(), rhs_shape); in HandleBroadcastAndOp()
/external/angle/src/libANGLE/renderer/d3d/
DIndexBuffer.cpp53 gl::DrawElementsType IndexBufferInterface::getIndexType() const in getIndexType() function in rx::IndexBufferInterface
55 return mIndexBuffer->getIndexType(); in getIndexType()
179 ASSERT(curSize >= size && indexType == getIndexType()); in reserveBufferSpace()
DIndexBuffer.h45 virtual gl::DrawElementsType getIndexType() const = 0;
71 gl::DrawElementsType getIndexType() const;
/external/llvm-project/mlir/test/lib/Transforms/
DTestVectorTransforms.cpp257 op.getLoc(), builder.getIndexType(), in runOnFunction()
258 builder.getIntegerAttr(builder.getIndexType(), 0)); in runOnFunction()
260 op.getLoc(), builder.getIndexType(), in runOnFunction()
261 builder.getIntegerAttr(builder.getIndexType(), 1)); in runOnFunction()
263 op.getLoc(), builder.getIndexType(), in runOnFunction()
264 builder.getIntegerAttr(builder.getIndexType(), multiplicity)); in runOnFunction()
/external/llvm-project/flang/lib/Lower/
DDoLoopHelper.cpp33 builder.createIntegerConstant(loc, builder.getIndexType(), 1), in createLoop()
39 auto indexType = builder.getIndexType(); in createLoop()
DCharacterExpr.cpp97 auto indexType = builder.getIndexType(); in toExtendedValue()
220 auto zero = builder.createIntegerConstant(loc, builder.getIndexType(), 0); in createLengthOneAssign()
285 builder.createConvert(loc, builder.getIndexType(), lhs.getLen()); in createConcatenate()
316 auto idxType = builder.getIndexType(); in createSubstring()
346 auto indexType = builder.getIndexType(); in createLenTrim()
DFIRBuilder.cpp43 auto indexType = getIndexType(); in createNullConstant()
77 auto idxTy = getIndexType(); in allocateLocal()
/external/llvm-project/mlir/lib/Conversion/ShapeToStandard/
DShapeToStandard.cpp97 Type indexTy = rewriter.getIndexType(); in matchAndRewrite()
184 Type indexTy = rewriter.getIndexType(); in matchAndRewrite()
240 Type indexTy = rewriter.getIndexType(); in matchAndRewrite()
319 rewriter.replaceOpWithNewOp<ExtractElementOp>(op, rewriter.getIndexType(), in matchAndRewrite()
372 Type indexTy = rewriter.getIndexType(); in matchAndRewrite()
451 Type indexTy = rewriter.getIndexType(); in matchAndRewrite()
528 loc, rewriter.getIndexType(), extentValues); in matchAndRewrite()
/external/swiftshader/third_party/llvm-10.0/llvm/lib/Transforms/Coroutines/
DCoroInternal.h144 IntegerType *getIndexType() const { in getIndexType() function
150 return ConstantInt::get(getIndexType(), Value); in getIndex()
/external/llvm-project/llvm/lib/Transforms/Coroutines/
DCoroInternal.h181 IntegerType *getIndexType() const { in getIndexType() function
187 return ConstantInt::get(getIndexType(), Value); in getIndex()
/external/llvm-project/mlir/lib/Conversion/StandardToLLVM/
DStandardToLLVM.cpp173 LLVM::LLVMType LLVMTypeConverter::getIndexType() { in getIndexType() function in LLVMTypeConverter
182 return getIndexType(); in convertIndexType()
328 auto indexTy = getIndexType(); in getMemRefDescriptorFields()
364 return {getIndexType(), LLVM::LLVMType::getInt8PtrTy(&getContext())}; in getUnrankedMemRefDescriptorFields()
586 loc, resultType, builder.getIntegerAttr(builder.getIndexType(), value)); in createIndexAttrConstant()
828 LLVM::LLVMType indexType = typeConverter.getIndexType(); in computeSizes()
892 createIndexAttrConstant(builder, loc, typeConverter.getIndexType(), 1); in alignedPtr()
907 createIndexAttrConstant(builder, loc, typeConverter.getIndexType(), 1); in setAlignedPtr()
921 createIndexAttrConstant(builder, loc, typeConverter.getIndexType(), 2); in offset()
925 loc, typeConverter.getIndexType().getPointerTo(), offsetGep); in offset()
[all …]
/external/angle/src/libANGLE/renderer/d3d/d3d9/
DIndexBuffer9.h35 gl::DrawElementsType getIndexType() const override;
/external/angle/src/libANGLE/renderer/d3d/d3d11/
DIndexBuffer11.h36 gl::DrawElementsType getIndexType() const override;
/external/llvm-project/llvm/lib/DebugInfo/PDB/Native/
DNativeTypeArray.cpp46 return Session.getSymbolCache().findSymbolByTypeIndex(Record.getIndexType()); in getArrayIndexTypeId()
/external/swiftshader/third_party/llvm-10.0/llvm/lib/DebugInfo/PDB/Native/
DNativeTypeArray.cpp46 return Session.getSymbolCache().findSymbolByTypeIndex(Record.getIndexType()); in getArrayIndexTypeId()
/external/llvm-project/mlir/lib/Dialect/Affine/IR/
DAffineOps.cpp321 auto indexTy = parser.getBuilder().getIndexType(); in parseDimAndSymbolList()
359 auto indexTy = builder.getIndexType(); in parseAffineApplyOp()
1021 auto indexType = parser.getBuilder().getIndexType(); in parse()
1161 auto indexType = parser.getBuilder().getIndexType(); in parse()
1227 builder.getIntegerAttr(builder.getIndexType(), step)); in build()
1243 Value inductionVar = bodyBlock.addArgument(builder.getIndexType()); in build()
1334 if (p.resolveOperand(boundOpInfos.front(), builder.getIndexType(), in parseBound()
1350 if (p.parseAttribute(boundAttr, builder.getIndexType(), boundAttrName, in parseBound()
1419 builder.getIntegerAttr(builder.getIndexType(), /*value=*/1)); in parseAffineForOp()
1423 if (parser.parseAttribute(stepAttr, builder.getIndexType(), in parseAffineForOp()
[all …]
/external/tensorflow/tensorflow/compiler/mlir/tools/kernel_gen/transforms/
Dtf_framework_legalize_to_llvm.cc136 /*size_t num_elements*/ getIndexType(), in GetFuncType()
137 /*size_t element_size*/ getIndexType(), in GetFuncType()
309 builder.getIntegerAttr(builder.getIndexType(), 0)); in GenerateErrorMessageConstant()
/external/llvm-project/mlir/include/mlir/Dialect/SPIRV/
DSPIRVLowering.h51 static Type getIndexType(MLIRContext *context);
/external/tensorflow/tensorflow/compiler/mlir/hlo/lib/utils/
Dbroadcast_utils.cc71 loc, RankedTensorType::get({result_rank}, builder.getIndexType()), in ComputeBinaryElementwiseBroadcastingResultExtents()
/external/deqp/framework/referencerenderer/
DrrRenderer.hpp99 inline IndexType getIndexType (void) const { return m_indexType; } in getIndexType() function in rr::PrimitiveList
/external/swiftshader/third_party/llvm-10.0/llvm/include/llvm/Analysis/Utils/
DLocal.h32 Type *IntIdxTy = DL.getIndexType(GEP->getType());
/external/llvm-project/mlir/lib/IR/
DBuilders.cpp53 IndexType Builder::getIndexType() { return IndexType::get(context); } in getIndexType() function in Builder
99 return IntegerAttr::get(getIndexType(), APInt(64, value)); in getIndexAttr()
141 getIndexType()), in getIndexTensorAttr()
/external/llvm-project/mlir/lib/Dialect/Shape/IR/
DShape.cpp664 builder.create<ConstantOp>(loc, builder.getIndexType(), dimAttr); in build()
665 build(builder, result, builder.getIndexType(), shape, dim); in build()
749 auto type = builder.getIndexType(); in build()
886 builder.getIndexType()); in fold()
902 bodyBlock.addArgument(builder.getIndexType()); in build()
/external/llvm-project/llvm/include/llvm/Analysis/Utils/
DLocal.h32 Type *IntIdxTy = DL.getIndexType(GEP->getType());

123456