| /external/tensorflow/tensorflow/compiler/mlir/hlo/lib/Dialect/mhlo/transforms/ |
| D | lhlo_legalize_to_parallel_loops.cc | 113 loc, b->getI1Type(), b->getIntegerAttr(b->getI1Type(), 1)); in MapWindowIvsToInput()
551 loc, b->getI1Type(), b->getIntegerAttr(b->getI1Type(), 0))); in InsertWindowLoops()
646 loc, b->getI1Type(), b->getIntegerAttr(b->getI1Type(), 1)); in SelectOrInitialize()
|
| D | transform_unranked_hlo.cc | 316 return rewriter.create<CmpIOp>(loc, rewriter.getI1Type(), CmpIPredicate::eq, in IsSingleElementShape()
|
| /external/llvm-project/flang/lib/Lower/ |
| D | OpenACC.cpp | 335 firOpBuilder.getI1Type(), cond); in genACCParallelOp()
346 firOpBuilder.getI1Type(), cond); in genACCParallelOp()
457 firOpBuilder.getI1Type(), cond); in genACCDataOp()
561 firOpBuilder.getI1Type(), cond); in genACCEnterDataOp()
661 firOpBuilder.getI1Type(), cond); in genACCExitDataOp()
752 firOpBuilder.getI1Type(), cond); in genACCInitShutdownOp()
815 firOpBuilder.getI1Type(), cond); in genACCUpdateOp()
966 firOpBuilder.getI1Type(), cond); in genACC()
|
| D | CharacterExpr.cpp | 351 auto trueVal = builder.createIntegerConstant(loc, builder.getI1Type(), 1); in createLenTrim()
|
| D | IO.cpp | 181 ? builder.create<fir::WhereOp>(loc, builder.getI1Type(), ok, true) in makeNextConditionalOn()
402 ok = builder.createIntegerConstant(loc, builder.getI1Type(), 1); in genIoLoop()
411 auto falseValue = builder.createIntegerConstant(loc, builder.getI1Type(), 0); in genIoLoop()
|
| /external/tensorflow/tensorflow/compiler/mlir/lite/utils/ |
| D | convert_type.cc | 53 return builder.getI1Type(); in ConvertElementType()
|
| /external/llvm-project/mlir/lib/Conversion/ShapeToStandard/ |
| D | ConvertShapeConstraints.cpp | 81 b.create<OrOp>(loc, b.getI1Type(), extentsAgree, in matchAndRewrite()
|
| D | ShapeToStandard.cpp | 257 Type i1Ty = rewriter.getI1Type(); in matchAndRewrite()
457 Type i1Ty = rewriter.getI1Type(); in matchAndRewrite()
|
| /external/tensorflow/tensorflow/compiler/mlir/hlo/lib/Dialect/mhlo/IR/ |
| D | chlo_ops.cc | 223 builder.getI1Type(), broadcast_dimensions); in build()
251 b.getI1Type()); in getIsInfLikeReturnType()
|
| /external/llvm-project/mlir/lib/IR/ |
| D | Builders.cpp | 55 IntegerType Builder::getI1Type() { return IntegerType::get(1, context); } in getI1Type() function in Builder
108 VectorType::get(static_cast<int64_t>(values.size()), getI1Type()), in getBoolVectorAttr()
|
| /external/tensorflow/tensorflow/compiler/mlir/python/mlir_wrapper/ |
| D | ops.cc | 101 .create<mlir::TF::AnyOp>(loc, opb.getI1Type(), input, in init_ops()
|
| /external/llvm-project/mlir/lib/Conversion/SCFToSPIRV/ |
| D | SCFToSPIRV.cpp | 176 loc, rewriter.getI1Type(), newIndVar, forOperands.upperBound()); in matchAndRewrite()
|
| /external/tensorflow/tensorflow/compiler/mlir/tensorflow/ir/ |
| D | tf_ops_helpers.inc | 80 return UnrankedTensorType::get(builder->getI1Type());
85 if (!ranked_type) return UnrankedTensorType::get(builder->getI1Type());
87 return RankedTensorType::get(ranked_type.getShape(), builder->getI1Type());
|
| D | tf_op_base.td | 602 resultType = UnrankedTensorType::get($_builder.getI1Type());
612 resultType = RankedTensorType::get(resultShape, $_builder.getI1Type());
|
| D | tf_executor.cc | 425 Type i1_type = parser.getBuilder().getI1Type(); in ParseSwitchOp()
|
| /external/llvm-project/mlir/lib/Dialect/GPU/Transforms/ |
| D | AllReduceLowering.cpp | 318 std::array<Type, 2> shuffleType = {valueType, rewriter.getI1Type()}; in createSubgroupReduce()
|
| /external/llvm-project/mlir/include/mlir/IR/ |
| D | Builders.h | 72 IntegerType getI1Type();
|
| /external/tensorflow/tensorflow/compiler/mlir/tensorflow/transforms/ |
| D | collection_ops_util.cc | 233 if (getElementTypeOrSelf(a.getType()) == builder.getI1Type()) { in AccumulateBuffers()
|
| /external/tensorflow/tensorflow/compiler/mlir/tfr/passes/ |
| D | raise_to_tf.cc | 143 result_elt_type = rewriter.getI1Type(); in CastToNonDerivedType()
|
| /external/llvm-project/flang/lib/Optimizer/Dialect/ |
| D | FIROps.cpp | 233 result.types.push_back(builder.getI1Type()); in buildCmpFOp()
280 mlir::Type i1Type = builder.getI1Type(); in parseCmpOp()
300 result.types.push_back(builder.getI1Type()); in buildCmpCOp()
|
| /external/llvm-project/mlir/lib/Dialect/OpenACC/IR/ |
| D | OpenACC.cpp | 202 Type i1Type = builder.getI1Type(); in parseParallelOp()
|
| /external/tensorflow/tensorflow/compiler/mlir/xla/ |
| D | hlo_utils.cc | 225 return builder.getI1Type(); in ConvertPrimitiveTypeToMLIRType()
|
| /external/llvm-project/mlir/lib/Dialect/GPU/IR/ |
| D | GPUDialect.cpp | 212 Type int1Type = parser.getBuilder().getI1Type(); in parseShuffleOp()
|
| /external/llvm-project/mlir/lib/Dialect/SPIRV/ |
| D | SPIRVOps.cpp | 891 Type boolType = builder.getI1Type(); in buildLogicalBinaryOp()
1196 Type boolTy = builder.getI1Type(); in parseBranchConditionalOp()
2176 build(builder, state, builder.getI1Type(), scope); in build()
|
| /external/llvm-project/mlir/lib/Dialect/SPIRV/Serialization/ |
| D | Deserializer.cpp | 1168 typeMap[operands[0]] = opBuilder.getI1Type(); in processType()
1637 constantMap.try_emplace(resultID, attr, opBuilder.getI1Type()); in processConstantBool()
|