Home
last modified time | relevance | path

Searched refs:getTensor (Results 1 – 16 of 16) sorted by relevance

/llvm-project-15.0.7/llvm/unittests/Analysis/
H A DMLModelRunnerTest.cpp62 NIMR.getTensor<int64_t>(0)[0] = 1; in TEST()
63 std::memcpy(NIMR.getTensor<int64_t>(1), in TEST()
66 std::memcpy(NIMR.getTensor<float>(2), in TEST()
69 ASSERT_EQ(NIMR.getTensor<int64_t>(0)[0], 1); in TEST()
70 ASSERT_EQ(NIMR.getTensor<int64_t>(1)[8], 9); in TEST()
71 ASSERT_EQ(NIMR.getTensor<float>(2)[1], 0.2f); in TEST()
80 *Evaluator->getTensor<int64_t>(0) = 1; in TEST()
81 *Evaluator->getTensor<int64_t>(1) = 2; in TEST()
94 *Evaluator->getTensor<int64_t>(0) = 1; in TEST()
95 *Evaluator->getTensor<int64_t>(1) = 2; in TEST()
[all …]
H A DTFUtilsTest.cpp120 int32_t *V = Evaluator->getTensor<int32_t>(0); in TEST()
125 float *F = Evaluator->getTensor<float>(1); in TEST()
/llvm-project-15.0.7/llvm/lib/Analysis/
H A DMLInlineAdvisor.cpp347 *ModelRunner->getTensor<int64_t>(FeatureIndex::CalleeBasicBlockCount) = in getAdviceImpl()
349 *ModelRunner->getTensor<int64_t>(FeatureIndex::CallSiteHeight) = in getAdviceImpl()
351 *ModelRunner->getTensor<int64_t>(FeatureIndex::NodeCount) = NodeCount; in getAdviceImpl()
353 *ModelRunner->getTensor<int64_t>(FeatureIndex::EdgeCount) = EdgeCount; in getAdviceImpl()
354 *ModelRunner->getTensor<int64_t>(FeatureIndex::CallerUsers) = in getAdviceImpl()
356 *ModelRunner->getTensor<int64_t>( in getAdviceImpl()
359 *ModelRunner->getTensor<int64_t>(FeatureIndex::CallerBasicBlockCount) = in getAdviceImpl()
361 *ModelRunner->getTensor<int64_t>( in getAdviceImpl()
364 *ModelRunner->getTensor<int64_t>(FeatureIndex::CalleeUsers) = in getAdviceImpl()
371 *ModelRunner->getTensor<int64_t>(inlineCostFeatureToMlFeature( in getAdviceImpl()
[all …]
/llvm-project-15.0.7/mlir/lib/Conversion/TensorToSPIRV/
H A DTensorToSPIRV.cpp48 TensorType tensorType = extractOp.getTensor().getType().cast<TensorType>(); in matchAndRewrite()
66 Type varType = spirv::PointerType::get(adaptor.getTensor().getType(), in matchAndRewrite()
70 if (adaptor.getTensor().getDefiningOp<spirv::ConstantOp>()) { in matchAndRewrite()
73 /*initializer=*/adaptor.getTensor()); in matchAndRewrite()
/llvm-project-15.0.7/mlir/lib/Dialect/SparseTensor/IR/
H A DSparseTensorDialect.cpp237 auto e = getSparseTensorEncoding(getTensor().getType()); in verify()
238 if (failed(isInBounds(getDim(), getTensor()))) in verify()
246 auto e = getSparseTensorEncoding(getTensor().getType()); in verify()
247 if (failed(isInBounds(getDim(), getTensor()))) in verify()
255 RankedTensorType ttp = getTensor().getType().cast<RankedTensorType>(); in verify()
/llvm-project-15.0.7/llvm/include/llvm/Analysis/
H A DMLModelRunner.h35 template <typename T, typename I> T *getTensor(I FeatureID) { in getTensor() function
40 template <typename T, typename I> const T *getTensor(I FeatureID) const { in getTensor() function
/llvm-project-15.0.7/mlir/lib/Dialect/Bufferization/IR/
H A DBufferizationOps.cpp90 auto memrefToTensor = toMemref.getTensor().getDefiningOp<ToTensorOp>(); in foldToMemrefToTensorPair()
519 FailureOr<Value> buffer = getBuffer(rewriter, getTensor(), options); in bufferize()
538 return toMemref.getTensor(); in fold()
569 if (auto memrefToTensor = getTensor().getDefiningOp<ToTensorOp>()) in fold()
623 rewriter.replaceOpWithNewOp<tensor::ExtractOp>(load, toMemref.getTensor(), in matchAndRewrite()
/llvm-project-15.0.7/mlir/lib/Dialect/SparseTensor/Transforms/
H A DSparseTensorConversion.cpp879 auto enc = getSparseTensorEncoding(op.getTensor().getType()); in matchAndRewrite()
966 Type elemTp = op.getTensor().getType().cast<ShapedType>().getElementType(); in matchAndRewrite()
983 ShapedType srcType = op.getTensor().getType().cast<ShapedType>(); in matchAndRewrite()
988 rewriter.setInsertionPointAfter(op.getTensor().getDefiningOp()); in matchAndRewrite()
1030 Type elemTp = op.getTensor().getType().cast<ShapedType>().getElementType(); in matchAndRewrite()
1059 ShapedType srcType = op.getTensor().getType().cast<ShapedType>(); in matchAndRewrite()
H A DSparseTensorPasses.cpp134 return converter.isLegal(op.getTensor().getType()); in runOnOperation()
/llvm-project-15.0.7/llvm/lib/CodeGen/
H A DMLRegallocEvictAdvisor.cpp674 Runner->getTensor<float>(FeatureIndex)[Pos] /= Largest[FeatureIndex]; in tryFindEvictionCandidate()
677 *Runner->getTensor<float>(FeatureIDs::progress) = in tryFindEvictionCandidate()
811 Runner->getTensor<TYPE>(FeatureIDs::ID)[Pos] = static_cast<TYPE>(VAL); \ in extractFeatures()
/llvm-project-15.0.7/mlir/test/IR/
H A Dcore-ops.mlir18 // CHECK: %[[T:.*]] = "getTensor"() : () -> tensor<4x4x?xf32>
19 %t = "getTensor"() : () -> tensor<4x4x?xf32>
/llvm-project-15.0.7/mlir/utils/vim/syntax/
H A Dmlir.vim40 syn keyword mlirOps getTensor index_cast load log memref_cast
/llvm-project-15.0.7/mlir/lib/Dialect/Bufferization/Transforms/
H A DBufferize.cpp111 rewriter.replaceOp(op, adaptor.getTensor()); in matchAndRewrite()
/llvm-project-15.0.7/mlir/lib/Dialect/Tensor/IR/
H A DTensorOps.cpp407 if (auto tensorType = getTensor().getType().dyn_cast<RankedTensorType>()) in verify()
430 if (auto fromElementsOp = getTensor().getDefiningOp<FromElementsOp>()) { in fold()
508 auto indexCast = extract.getTensor().getDefiningOp<arith::IndexCastOp>(); in matchAndRewrite()
687 auto tensorFromElements = extract.getTensor().getDefiningOp<GenerateOp>(); in matchAndRewrite()
717 auto tensorCast = extract.getTensor().getDefiningOp<tensor::CastOp>(); in matchAndRewrite()
/llvm-project-15.0.7/mlir/lib/Dialect/Tensor/Transforms/
H A DBufferizableOpInterfaceImpl.cpp331 getBuffer(rewriter, extractOp.getTensor(), options); in bufferize()
764 FailureOr<Value> v = getBuffer(rewriter, rankOp.getTensor(), options); in bufferize()
/llvm-project-15.0.7/mlir/lib/Dialect/SCF/IR/
H A DSCF.cpp1005 tensorToMemref.getTensor()); in matchAndRewrite()