Home
last modified time | relevance | path

Searched refs:getOutput (Results 1 – 11 of 11) sorted by relevance

/llvm-project-15.0.7/mlir/lib/Dialect/Tosa/Transforms/
H A DTosaDecomposeDepthwise.cpp33 ShapedType resultType = op.getOutput().getType().cast<ShapedType>(); in matchAndRewrite()
98 auto outputShape = op.getOutput().getType().cast<ShapedType>().getShape(); in matchAndRewrite()
/llvm-project-15.0.7/llvm/lib/Analysis/
H A DTFUtils.cpp125 std::vector<TF_Tensor *> &getOutput() { return Output; } in getOutput() function in llvm::EvaluationResultImpl
379 Impl->evaluate(Ret->getOutput().data(), Status.get()); in evaluate()
419 return TF_TensorData(Impl->getOutput()[Index]); in getUntypedTensorValue()
424 return TF_TensorData(Impl->getOutput()[Index]); in getUntypedTensorValue()
/llvm-project-15.0.7/mlir/lib/Dialect/SparseTensor/IR/
H A DSparseTensorDialect.cpp297 Type outputType = getOutput().getType(); in verify()
337 Type outputType = getOutput().getType(); in verify()
/llvm-project-15.0.7/mlir/lib/Dialect/Bufferization/IR/
H A DBufferizationOps.cpp427 effects.emplace_back(MemoryEffects::Write::get(), getOutput(), in getEffects()
429 effects.emplace_back(MemoryEffects::Allocate::get(), getOutput(), in getEffects()
457 memref::findDealloc(cloneOp.getOutput()); in matchAndRewrite()
/llvm-project-15.0.7/mlir/include/mlir/Support/
H A DTiming.h377 raw_ostream &getOutput() const;
/llvm-project-15.0.7/mlir/lib/Support/
H A DTiming.cpp474 raw_ostream &DefaultTimingManager::getOutput() const { in getOutput() function in DefaultTimingManager
/llvm-project-15.0.7/mlir/lib/Conversion/TosaToLinalg/
H A DTosaToLinalg.cpp1157 auto outputTy = op.getOutput().getType().cast<ShapedType>(); in matchAndRewrite()
1166 checkHasDynamicBatchDims(rewriter, op, {input, op.getOutput()}); in matchAndRewrite()
1358 checkHasDynamicBatchDims(rewriter, op, {input, op.getOutput()}); in matchAndRewrite()
1931 auto resultTy = argmaxOp.getOutput().getType().cast<ShapedType>(); in matchAndRewrite()
2053 rewriter, op, {input, indices, op.getOutput()}); in matchAndRewrite()
H A DTosaToLinalgNamed.cpp696 checkHasDynamicBatchDims(rewriter, op, {input, op.getOutput()}); in matchAndRewrite()
773 checkHasDynamicBatchDims(rewriter, op, {input, op.getOutput()}); in matchAndRewrite()
/llvm-project-15.0.7/mlir/lib/Dialect/Tosa/IR/
H A DTosaCanonicalizations.cpp302 Value output = op.getOutput(); in matchAndRewrite()
/llvm-project-15.0.7/mlir/include/mlir/Dialect/Bufferization/IR/
H A DBufferizationOps.td181 Value getTarget() { return getOutput(); }
/llvm-project-15.0.7/mlir/test/lib/Dialect/Test/
H A DTestOps.td261 DerivedTypeAttr<"return getElementTypeOrSelf(getOutput().getType());">;
263 "return getOutput().getType().cast<ShapedType>().getSizeInBits();",