Home
last modified time | relevance | path

Searched refs:getFunctionType (Results 1 – 25 of 419) sorted by relevance

12345678910>>...17

/external/tensorflow/tensorflow/compiler/mlir/lite/utils/
Dperception_ops_utils.cc103 func_.getLoc(), func_.getFunctionType().getResults(), in RewriteFunc()
118 if (func_.getFunctionType().getNumResults() != 1) { in VerifySignature()
121 << func_.getFunctionType().getNumResults(); in VerifySignature()
199 func_.getFunctionType().getResults(), in RewriteFunc()
214 if (func_.getFunctionType().getNumResults() != 1) { in VerifySignature()
217 << func_.getFunctionType().getNumResults(); in VerifySignature()
222 func_.getFunctionType().getInput(0).dyn_cast_or_null<RankedTensorType>(); in VerifySignature()
229 func_.getFunctionType().getInput(1).dyn_cast_or_null<RankedTensorType>(); in VerifySignature()
236 func_.getFunctionType().getResult(0).dyn_cast_or_null<RankedTensorType>(); in VerifySignature()
Dnms_utils.cc49 auto output_type0 = func_.getFunctionType().getResult(0); in RewriteFunc()
50 auto output_type1 = func_.getFunctionType().getResult(1); in RewriteFunc()
69 if (func_.getFunctionType().getNumResults() != 2) { in VerifySignature()
72 << func_.getFunctionType().getNumResults(); in VerifySignature()
77 func_.getFunctionType().getInput(0).dyn_cast<RankedTensorType>(); in VerifySignature()
99 func_.getLoc(), func_.getFunctionType().getResults(), in RewriteFunc()
188 if (func_.getFunctionType().getNumResults() != 4) { in VerifySignature()
191 << ": " << func_.getFunctionType().getNumResults(); in VerifySignature()
Dtftext_utils.cc64 return func.getFunctionType().getInput(idx).dyn_cast_or_null<TensorType>(); in GetInputType()
68 return func.getFunctionType().getResult(idx).dyn_cast_or_null<TensorType>(); in GetResultType()
138 func.getLoc(), func.getFunctionType().getResults(), func.getArguments(), in ConvertWhitespaceTokenizer()
151 if (func.getFunctionType().getInputs().size() != in VerifyNgrams()
152 func.getFunctionType().getResults().size()) { in VerifyNgrams()
156 int row_splits = func.getFunctionType().getInputs().size() - kRowSplits; in VerifyNgrams()
267 func.getLoc(), func.getFunctionType().getResults(), func.getArguments(), in ConvertNgrams()
274 if (func.getFunctionType().getNumInputs() != 2 || in VerifySgnnProjection()
275 func.getFunctionType().getNumResults() != 1) { in VerifySgnnProjection()
348 func.getLoc(), func.getFunctionType().getResults(), func.getArguments(), in ConvertSgnnProjection()
Dlstm_utils_test.cc65 auto func_type = builder->getFunctionType(input_types, output_type); in createLstmCompositeFunc()
132 EXPECT_EQ(fused_lstm_func_.getFunctionType().getNumResults(), 1); in TEST_F()
183 EXPECT_EQ(fused_lstm_func_.getFunctionType().getNumResults(), 1); in TEST_F()
184 auto output_types = fused_lstm_func_.getFunctionType().getResults(); in TEST_F()
232 EXPECT_EQ(fused_ln_lstm_func_.getFunctionType().getNumResults(), 1); in TEST_F()
254 EXPECT_EQ(fused_ln_lstm_func_.getFunctionType().getNumResults(), 1); in TEST_F()
255 auto output_types = fused_ln_lstm_func_.getFunctionType().getResults(); in TEST_F()
/external/swiftshader/third_party/llvm-16.0/llvm/lib/Target/WebAssembly/
DWebAssemblyFixFunctionBitcasts.cpp78 if (CB->getFunctionType() == F.getValueType()) in findUses()
121 FunctionType::param_iterator PI = F->getFunctionType()->param_begin(); in createWrapper()
122 FunctionType::param_iterator PE = F->getFunctionType()->param_end(); in createWrapper()
126 Type *ExpectedRtnType = F->getFunctionType()->getReturnType(); in createWrapper()
129 if ((F->getFunctionType()->getNumParams() != Ty->getNumParams()) || in createWrapper()
130 (F->getFunctionType()->isVarArg() != Ty->isVarArg()) || in createWrapper()
170 Type *ExpectedRtnType = F->getFunctionType()->getReturnType(); in createWrapper()
254 if (shouldFixMainFunction(F.getFunctionType(), MainTy)) { in runOnModule()
256 << *F.getFunctionType() << "\n"); in runOnModule()
272 FunctionType *Ty = CB->getFunctionType(); in runOnModule()
DWebAssemblyAddMissingPrototypes.cpp81 unsigned NumParams = F.getFunctionType()->getNumParams(); in runOnModule()
109 FunctionType *DestType = CB->getFunctionType(); in runOnModule()
132 NewType = FunctionType::get(F.getFunctionType()->getReturnType(), false); in runOnModule()
/external/swiftshader/third_party/llvm-10.0/llvm/lib/Target/WebAssembly/
DWebAssemblyFixFunctionBitcasts.cpp130 FunctionType::param_iterator PI = F->getFunctionType()->param_begin(); in createWrapper()
131 FunctionType::param_iterator PE = F->getFunctionType()->param_end(); in createWrapper()
135 Type *ExpectedRtnType = F->getFunctionType()->getReturnType(); in createWrapper()
138 if ((F->getFunctionType()->getNumParams() != Ty->getNumParams()) || in createWrapper()
139 (F->getFunctionType()->isVarArg() != Ty->isVarArg()) || in createWrapper()
179 Type *ExpectedRtnType = F->getFunctionType()->getReturnType(); in createWrapper()
260 if (shouldFixMainFunction(F.getFunctionType(), MainTy)) { in runOnModule()
262 << *F.getFunctionType() << "\n"); in runOnModule()
/external/tensorflow/tensorflow/compiler/mlir/xla/transforms/
Doutline_with_xla_framework.cc76 SmallVector<Location> locs(func.getFunctionType().getNumInputs(), loc); in matchAndRewrite()
89 SmallVector<Type> operands(func.getFunctionType().getNumInputs(), in matchAndRewrite()
91 SmallVector<Type> result_array(func.getFunctionType().getNumResults(), in matchAndRewrite()
111 for (const auto &t : llvm::enumerate(func.getFunctionType().getInputs())) { in matchAndRewrite()
117 loc, func.getSymName(), func.getFunctionType().getResults(), args); in matchAndRewrite()
/external/tensorflow/tensorflow/compiler/mlir/tensorflow/ir/
Dtf_remaining_ops.cc107 if (op->getNumOperands() != func.getFunctionType().getNumInputs()) in verify()
109 << "'host_func' has " << func.getFunctionType().getNumInputs() in verify()
113 if (op->getNumResults() != func.getFunctionType().getNumResults()) in verify()
115 << func.getFunctionType().getNumResults() in verify()
/external/swiftshader/third_party/llvm-16.0/llvm/lib/Target/AArch64/
DSMEABIPass.cpp76 Builder.CreateCall(WriteIntr->getFunctionType(), WriteIntr, in emitTPIDR2Save()
99 auto *TPIDR2 = Builder.CreateCall(TPIDR2Intr->getFunctionType(), TPIDR2Intr, in updateNewZAFunctions()
113 Builder.CreateCall(EnableZAIntr->getFunctionType(), EnableZAIntr); in updateNewZAFunctions()
123 Builder.CreateCall(DisableZAIntr->getFunctionType(), DisableZAIntr); in updateNewZAFunctions()
/external/swiftshader/third_party/llvm-16.0/llvm/lib/Transforms/Coroutines/
DCoroutines.cpp376 Prototype->getFunctionType()->dump(); in buildFrom()
385 Prototype->getFunctionType()->dump(); in buildFrom()
404 Prototype->getFunctionType()->dump(); in buildFrom()
412 Prototype->getFunctionType()->dump(); in buildFrom()
463 Alloc->getFunctionType()->getParamType(0), in emitAlloc()
486 Dealloc->getFunctionType()->getParamType(0)); in emitDealloc()
518 auto FT = F->getFunctionType(); in checkWFRetconPrototype()
536 I->getFunction()->getFunctionType()->getReturnType()) in checkWFRetconPrototype()
554 auto FT = F->getFunctionType(); in checkWFAlloc()
569 auto FT = F->getFunctionType(); in checkWFDealloc()
[all …]
/external/swiftshader/third_party/llvm-10.0/llvm/lib/Transforms/Utils/
DCallPromotionUtils.cpp337 unsigned NumParams = Callee->getFunctionType()->getNumParams(); in isLegalToPromote()
351 Type *FormalTy = Callee->getFunctionType()->getFunctionParamType(I); in isLegalToPromote()
381 if (CS.getFunctionType() == Callee->getFunctionType()) in promoteCall()
389 CS.mutateFunctionType(Callee->getFunctionType()); in promoteCall()
394 auto CalleeType = Callee->getFunctionType(); in promoteCall()
/external/swiftshader/third_party/llvm-10.0/llvm/lib/Transforms/Coroutines/
DCoroutines.cpp422 Prototype->getFunctionType()->dump(); in buildFrom()
431 Prototype->getFunctionType()->dump(); in buildFrom()
450 Prototype->getFunctionType()->dump(); in buildFrom()
458 Prototype->getFunctionType()->dump(); in buildFrom()
509 Alloc->getFunctionType()->getParamType(0), in emitAlloc()
530 Dealloc->getFunctionType()->getParamType(0)); in emitDealloc()
560 auto FT = F->getFunctionType(); in checkWFRetconPrototype()
578 I->getFunction()->getFunctionType()->getReturnType()) in checkWFRetconPrototype()
596 auto FT = F->getFunctionType(); in checkWFAlloc()
611 auto FT = F->getFunctionType(); in checkWFDealloc()
DCoroInternal.h167 return RetconLowering.ResumePrototype->getFunctionType(); in getResumeFunctionType()
175 auto FTy = CoroBegin->getFunction()->getFunctionType(); in getRetconResultTypes()
190 auto FTy = RetconLowering.ResumePrototype->getFunctionType(); in getRetconResumeTypes()
/external/swiftshader/third_party/llvm-10.0/llvm/lib/Target/Mips/
DMips16HardFloat.cpp111 TypeID ArgTypeID = F.getFunctionType()->getParamType(0)->getTypeID(); in whichFPParamVariantNeeded()
122 TypeID ArgTypeID0 = F.getFunctionType()->getParamType(0)->getTypeID(); in whichFPParamVariantNeeded()
123 TypeID ArgTypeID1 = F.getFunctionType()->getParamType(1)->getTypeID(); in whichFPParamVariantNeeded()
158 Type *ArgType = F.getFunctionType()->getParamType(0); in needsFPStubFromParams()
271 FStub = Function::Create(F.getFunctionType(), in assureFPCallStub()
420 FunctionType *FT = CI->getFunctionType(); in fixupFPReturnAndCall()
456 (F->getFunctionType(), in createFPFnStub()
/external/llvm/lib/Target/Mips/
DMips16HardFloat.cpp111 TypeID ArgTypeID = F.getFunctionType()->getParamType(0)->getTypeID(); in whichFPParamVariantNeeded()
122 TypeID ArgTypeID0 = F.getFunctionType()->getParamType(0)->getTypeID(); in whichFPParamVariantNeeded()
123 TypeID ArgTypeID1 = F.getFunctionType()->getParamType(1)->getTypeID(); in whichFPParamVariantNeeded()
159 Type *ArgType = F.getFunctionType()->getParamType(0); in needsFPStubFromParams()
276 FStub = Function::Create(F.getFunctionType(), in assureFPCallStub()
428 FunctionType *FT = CI->getFunctionType(); in fixupFPReturnAndCall()
464 (F->getFunctionType(), in createFPFnStub()
/external/swiftshader/third_party/llvm-16.0/llvm/lib/Target/Mips/
DMips16HardFloat.cpp112 TypeID ArgTypeID = F.getFunctionType()->getParamType(0)->getTypeID(); in whichFPParamVariantNeeded()
123 TypeID ArgTypeID0 = F.getFunctionType()->getParamType(0)->getTypeID(); in whichFPParamVariantNeeded()
124 TypeID ArgTypeID1 = F.getFunctionType()->getParamType(1)->getTypeID(); in whichFPParamVariantNeeded()
159 Type *ArgType = F.getFunctionType()->getParamType(0); in needsFPStubFromParams()
272 FStub = Function::Create(F.getFunctionType(), in assureFPCallStub()
419 FunctionType *FT = CI->getFunctionType(); in fixupFPReturnAndCall()
455 (F->getFunctionType(), in createFPFnStub()
/external/swiftshader/third_party/llvm-16.0/llvm/lib/Transforms/Utils/
DCallPromotionUtils.cpp400 unsigned NumParams = Callee->getFunctionType()->getNumParams(); in isLegalToPromote()
433 Type *FormalTy = Callee->getFunctionType()->getFunctionParamType(I); in isLegalToPromote()
484 if (CB.getFunctionType() == Callee->getFunctionType()) in promoteCall()
492 CB.mutateFunctionType(Callee->getFunctionType()); in promoteCall()
497 auto CalleeType = Callee->getFunctionType(); in promoteCall()
/external/swiftshader/third_party/llvm-16.0/llvm/lib/Target/SPIRV/
DSPIRVPrepareFunctions.cpp85 FunctionType::get(RetType, ArgTypes, F->getFunctionType()->isVarArg()); in processFunctionSignature()
116 CI->mutateFunctionType(NewF->getFunctionType()); in processFunctionSignature()
136 if (F && F->getFunctionType() == FT) in getOrCreateFunction()
165 M->getOrInsertFunction(FuncName, Intrinsic->getFunctionType()); in lowerIntrinsicToFunction()
213 FunctionType *FSHFuncTy = FSHIntrinsic->getFunctionType(); in lowerFunnelShifts()
295 FunctionType *UMulFuncTy = UMulIntrinsic->getFunctionType(); in lowerUMulWithOverflow()
/external/swiftshader/third_party/llvm-16.0/llvm/lib/Transforms/IPO/
DDeadArgumentElimination.cpp123 assert(F.getFunctionType()->isVarArg() && "Function isn't varargs!"); in deleteDeadVarargs()
159 FunctionType *FTy = F.getFunctionType(); in deleteDeadVarargs()
276 !F.getFunctionType()->isVarArg()) in removeDeadArgumentsFromCallers()
311 CB->getFunctionType() != F.getFunctionType()) in removeDeadArgumentsFromCallers()
445 if (ArgNo >= F->getFunctionType()->getNumParams()) in surveyUse()
552 CB->getFunctionType() != F.getFunctionType()) { in surveyFunction()
617 if (F.getFunctionType()->isVarArg() || HasMustTailCallers || in surveyFunction()
729 FunctionType *FTy = F->getFunctionType(); in removeDeadStuffFromFunction()
1094 if (F.getFunctionType()->isVarArg()) in run()
/external/swiftshader/third_party/llvm-10.0/llvm/lib/Target/AMDGPU/
DAMDGPULibFunc.h347 virtual FunctionType *getFunctionType(Module &M) const = 0;
391 FunctionType *getFunctionType(Module &M) const { in getFunctionType() function
392 return Impl->getFunctionType(M); in getFunctionType()
416 FunctionType *getFunctionType(Module &M) const override;
446 FunctionType *getFunctionType(Module &M) const override { return FuncTy; } in getFunctionType() function
/external/swiftshader/third_party/llvm-16.0/llvm/lib/Target/AMDGPU/
DAMDGPULibFunc.h349 virtual FunctionType *getFunctionType(Module &M) const = 0;
393 FunctionType *getFunctionType(Module &M) const { in getFunctionType() function
394 return Impl->getFunctionType(M); in getFunctionType()
418 FunctionType *getFunctionType(Module &M) const override;
448 FunctionType *getFunctionType(Module &M) const override { return FuncTy; } in getFunctionType() function
/external/tensorflow/tensorflow/compiler/mlir/tensorflow/transforms/
Dcluster_outlining.cc64 auto func_type = builder->getFunctionType(operand_types, op.getResultTypes()); in BuildFunction()
117 cluster_op.getLoc(), outlined_func.getFunctionType().getResults(), in OutlineCluster()
140 launch_op.getLoc(), outlined_func.getFunctionType().getResults(), in OutlineLaunch()
/external/tensorflow/tensorflow/dtensor/mlir/
Dspmd_expansion.cc65 auto func_type = function.getFunctionType(); in UpdateFunctionInputShape()
73 .setType(function.getFunctionType().getInput(argument_index)); in UpdateFunctionInputShape()
206 function.getFunctionType().getInput(argument_index)); in UpdateFunctionArgsUsingLayout()
217 function.getFunctionType() in UpdateFunctionArgsUsingLayout()
273 function.getContext(), function.getFunctionType().getInputs(), in UpdateReturnValueShapes()
/external/swiftshader/third_party/llvm-16.0/llvm/lib/CodeGen/
DReplaceWithVeclib.cpp50 TLIFunc = Function::Create(OldFunc->getFunctionType(), in replaceWithTLIFunction()
77 assert(OldFunc->getFunctionType() == TLIFunc->getFunctionType() && in replaceWithTLIFunction()

12345678910>>...17