Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/lite/experimental/support/java/src/java/org/tensorflow/lite/support/common/ops/
DCastOp.java24 public class CastOp implements TensorOperator { class
41 public CastOp(DataType destinationType) { in CastOp() method in CastOp
/external/tensorflow/tensorflow/compiler/tf2xla/kernels/
Dcast_op.cc34 class CastOp : public XlaOpKernel { class
36 explicit CastOp(OpKernelConstruction* ctx) : XlaOpKernel(ctx) { in CastOp() function in tensorflow::__anon1671a62a0111::CastOp
100 TF_DISALLOW_COPY_AND_ASSIGN(CastOp);
103 REGISTER_XLA_OP(Name("Cast"), CastOp);
/external/llvm/unittests/Analysis/
DValueTrackingTest.cpp55 Instruction::CastOps CastOp; in expectPattern() local
56 SelectPatternResult R = matchSelectPattern(A, LHS, RHS, &CastOp); in expectPattern()
/external/swiftshader/third_party/llvm-7.0/llvm/lib/Transforms/Scalar/
DFloat2Int.cpp199 auto CastOp = (Instruction::CastOps)I->getOpcode(); in walkBackwards() local
200 seen(I, validateRange(Input.castOp(CastOp, MaxIntegerBW+1))); in walkBackwards()
264 auto CastOp = (Instruction::CastOps)I->getOpcode(); in walkForwards() local
265 return Ops[0].castOp(CastOp, MaxIntegerBW+1); in walkForwards()
/external/tensorflow/tensorflow/compiler/mlir/tensorflow/transforms/
Dfunctional_control_flow_to_cfg.cc84 builder->create<TF::CastOp>(loc, expected, val, in CallFn()
108 builder->create<TF::CastOp>(loc, expected, val, in PrepareValsForJump()
142 builder->create<TF::CastOp>(loc, result.getType(), arg, in ReplaceOpResultWithBlockArgs()
Dshape_inference.cc78 if (auto cast_op = dyn_cast_or_null<CastOp>(arg_defining_op)) { in InferShapeForFunctionReturnType()
117 TF::CastOp cast_op; in AddCastBackForUnsupportedNonTFUses()
121 builder.create<TF::CastOp>(op->getLoc(), old_type, result, in AddCastBackForUnsupportedNonTFUses()
203 if (isa<CastOp>(op) && in InferShapeForSingleOperation()
/external/swiftshader/third_party/llvm-7.0/llvm/unittests/Analysis/
DValueTrackingTest.cpp56 Instruction::CastOps CastOp; in expectPattern() local
57 SelectPatternResult R = matchSelectPattern(A, LHS, RHS, &CastOp); in expectPattern()
/external/swiftshader/third_party/llvm-7.0/llvm/include/llvm/Analysis/
DValueTracking.h558 Instruction::CastOps *CastOp = nullptr,
562 Instruction::CastOps *CastOp = nullptr) {
/external/swiftshader/third_party/llvm-7.0/llvm/include/llvm/IR/
DConstantRange.h230 ConstantRange castOp(Instruction::CastOps CastOp,
/external/llvm/lib/Analysis/
DBasicAliasAnalysis.cpp270 Value *CastOp = cast<CastInst>(V)->getOperand(0); in GetLinearExpression() local
272 unsigned SmallWidth = CastOp->getType()->getPrimitiveSizeInBits(); in GetLinearExpression()
275 GetLinearExpression(CastOp, Scale, Offset, ZExtBits, SExtBits, DL, in GetLinearExpression()
295 Result = CastOp; in GetLinearExpression()
308 Result = CastOp; in GetLinearExpression()
DValueTracking.cpp3840 Instruction::CastOps *CastOp) { in lookThroughCast() argument
3845 *CastOp = CI->getOpcode(); in lookThroughCast()
3900 Instruction::CastOps *CastOp) { in matchSelectPattern() argument
3921 if (CastOp && CmpLHS->getType() != TrueVal->getType()) { in matchSelectPattern()
3922 if (Value *C = lookThroughCast(CmpI, TrueVal, FalseVal, CastOp)) in matchSelectPattern()
3926 if (Value *C = lookThroughCast(CmpI, FalseVal, TrueVal, CastOp)) in matchSelectPattern()
DScalarEvolution.cpp4678 Optional<unsigned> CastOp; in getRangeViaFactoring() local
4697 CastOp = SCast->getSCEVType(); in getRangeViaFactoring()
4716 if (CastOp.hasValue()) in getRangeViaFactoring()
4717 switch (*CastOp) { in getRangeViaFactoring()
6642 if (Constant *CastOp = BuildConstantFromSCEV(SS->getOperand())) in BuildConstantFromSCEV() local
6643 return ConstantExpr::getSExt(CastOp, SS->getType()); in BuildConstantFromSCEV()
6648 if (Constant *CastOp = BuildConstantFromSCEV(SZ->getOperand())) in BuildConstantFromSCEV() local
6649 return ConstantExpr::getZExt(CastOp, SZ->getType()); in BuildConstantFromSCEV()
6654 if (Constant *CastOp = BuildConstantFromSCEV(ST->getOperand())) in BuildConstantFromSCEV() local
6655 return ConstantExpr::getTrunc(CastOp, ST->getType()); in BuildConstantFromSCEV()
/external/swiftshader/third_party/llvm-7.0/llvm/lib/Analysis/
DBasicAliasAnalysis.cpp330 Value *CastOp = cast<CastInst>(V)->getOperand(0); in GetLinearExpression() local
332 unsigned SmallWidth = CastOp->getType()->getPrimitiveSizeInBits(); in GetLinearExpression()
335 GetLinearExpression(CastOp, Scale, Offset, ZExtBits, SExtBits, DL, in GetLinearExpression()
355 Result = CastOp; in GetLinearExpression()
368 Result = CastOp; in GetLinearExpression()
DValueTracking.cpp4711 Instruction::CastOps *CastOp) { in lookThroughCast() argument
4716 *CastOp = Cast1->getOpcode(); in lookThroughCast()
4720 if (*CastOp == Cast2->getOpcode() && SrcTy == Cast2->getSrcTy()) in lookThroughCast()
4730 switch (*CastOp) { in lookThroughCast()
4796 ConstantExpr::getCast(*CastOp, CastedTo, C->getType(), true); in lookThroughCast()
4804 Instruction::CastOps *CastOp, in matchSelectPattern() argument
4829 if (CastOp && CmpLHS->getType() != TrueVal->getType()) { in matchSelectPattern()
4830 if (Value *C = lookThroughCast(CmpI, TrueVal, FalseVal, CastOp)) { in matchSelectPattern()
4833 if (*CastOp == Instruction::FPToSI || *CastOp == Instruction::FPToUI) in matchSelectPattern()
4839 if (Value *C = lookThroughCast(CmpI, FalseVal, TrueVal, CastOp)) { in matchSelectPattern()
[all …]
DScalarEvolution.cpp5847 Optional<unsigned> CastOp; in getRangeViaFactoring() local
5866 CastOp = SCast->getSCEVType(); in getRangeViaFactoring()
5885 if (CastOp.hasValue()) in getRangeViaFactoring()
5886 switch (*CastOp) { in getRangeViaFactoring()
7994 if (Constant *CastOp = BuildConstantFromSCEV(SS->getOperand())) in BuildConstantFromSCEV() local
7995 return ConstantExpr::getSExt(CastOp, SS->getType()); in BuildConstantFromSCEV()
8000 if (Constant *CastOp = BuildConstantFromSCEV(SZ->getOperand())) in BuildConstantFromSCEV() local
8001 return ConstantExpr::getZExt(CastOp, SZ->getType()); in BuildConstantFromSCEV()
8006 if (Constant *CastOp = BuildConstantFromSCEV(ST->getOperand())) in BuildConstantFromSCEV() local
8007 return ConstantExpr::getTrunc(CastOp, ST->getType()); in BuildConstantFromSCEV()
/external/llvm/include/llvm/Analysis/
DValueTracking.h447 Instruction::CastOps *CastOp = nullptr);
/external/llvm/lib/Transforms/InstCombine/
DInstCombineSelect.cpp1104 Instruction::CastOps CastOp; in visitSelectInst() local
1105 SelectPatternResult SPR = matchSelectPattern(&SI, LHS, RHS, &CastOp); in visitSelectInst()
1124 Value *NewSI = Builder->CreateCast(CastOp, in visitSelectInst()
/external/swiftshader/third_party/llvm-7.0/llvm/lib/IR/
DConstantRange.cpp573 ConstantRange ConstantRange::castOp(Instruction::CastOps CastOp, in castOp() argument
575 switch (CastOp) { in castOp()
/external/tensorflow/tensorflow/compiler/mlir/lite/transforms/
Dlegalize_tf.cc212 .create<TF::CastOp>(op->getLoc(), new_type, shape, in matchAndRewrite()
/external/swiftshader/third_party/llvm-7.0/llvm/lib/Transforms/InstCombine/
DInstCombineSelect.cpp1763 Instruction::CastOps CastOp; in visitSelectInst() local
1764 SelectPatternResult SPR = matchSelectPattern(&SI, LHS, RHS, &CastOp); in visitSelectInst()
1795 Value *NewCast = Builder.CreateCast(CastOp, NewSI, SelType); in visitSelectInst()
DInstCombineCalls.cpp1627 Optional<Instruction::CastOps> CastOp; in SimplifyNVVMIntrinsic() member
1640 SimplifyAction(Instruction::CastOps CastOp) : CastOp(CastOp) {} in SimplifyNVVMIntrinsic()
1827 if (Action.CastOp) in SimplifyNVVMIntrinsic()
1828 return CastInst::Create(*Action.CastOp, II->getArgOperand(0), II->getType(), in SimplifyNVVMIntrinsic()
/external/tensorflow/tensorflow/compiler/mlir/tensorflow/ir/
Dtf_ops.cc517 OpFoldResult CastOp::fold(ArrayRef<Attribute> operands) { in fold()
2802 return builder.create<TF::CastOp>(conversion_loc, result_type, input, in materializeCallConversion()
/external/swiftshader/third_party/llvm-7.0/llvm/lib/Bitcode/Reader/
DBitcodeReader.cpp3584 auto CastOp = (Instruction::CastOps)Opc; in parseFunctionBody() local
3585 if (!CastInst::castIsValid(CastOp, Op, ResTy)) in parseFunctionBody()
3587 I = CastInst::Create(CastOp, Op, ResTy); in parseFunctionBody()
/external/llvm/lib/Bitcode/Reader/
DBitcodeReader.cpp4572 auto CastOp = (Instruction::CastOps)Opc; in parseFunctionBody() local
4573 if (!CastInst::castIsValid(CastOp, Op, ResTy)) in parseFunctionBody()
4575 I = CastInst::Create(CastOp, Op, ResTy); in parseFunctionBody()
/external/swiftshader/third_party/llvm-7.0/llvm/lib/Transforms/Vectorize/
DLoopVectorize.cpp2926 Instruction::CastOps CastOp = in createVectorizedLoopSkeleton() local
2928 Value *CRD = B.CreateCast(CastOp, CountRoundDown, StepType, "cast.crd"); in createVectorizedLoopSkeleton()