| /external/llvm/unittests/ADT/ |
| D | APIntTest.cpp | 987 EXPECT_FALSE(A.isSplat(1)); in TEST()
988 EXPECT_FALSE(A.isSplat(2)); in TEST()
989 EXPECT_FALSE(A.isSplat(4)); in TEST()
990 EXPECT_TRUE(A.isSplat(8)); in TEST()
991 EXPECT_TRUE(A.isSplat(16)); in TEST()
992 EXPECT_TRUE(A.isSplat(32)); in TEST()
995 EXPECT_FALSE(B.isSplat(1)); in TEST()
996 EXPECT_TRUE(B.isSplat(2)); in TEST()
997 EXPECT_TRUE(B.isSplat(4)); in TEST()
998 EXPECT_TRUE(B.isSplat(8)); in TEST()
[all …]
|
| /external/tensorflow/tensorflow/compiler/mlir/tensorflow/transforms/ |
| D | rewrite_util.h | 56 if (matchPattern(value, m_Constant(&float_attr)) && float_attr.isSplat() && in IsConstantValueOf()
61 if (matchPattern(value, m_Constant(&int_attr)) && int_attr.isSplat() && in IsConstantValueOf()
|
| D | legalize_hlo.cc | 93 if (!matchPattern(value, m_Constant(&attr)) || !attr.isSplat()) { in GetConstantSplatValue()
173 if (!lhs_dilation.isSplat() || lhs_dilation.getSplatValue<int64_t>() != 1) in IsSupportedConvOp()
443 (conv_op.padding().getValue().isSplat() && in matchAndRewrite()
750 (conv_op.padding().getValue().isSplat() && in matchAndRewrite()
833 if (lhs_dilation.isSplat() && lhs_dilation.getSplatValue<int64_t>() == 1) in IsSupportedConvOp()
938 if (lhs_dilation.getNumElements() != 2 || !rhs_dilation.isSplat() || in MatchResizeOp()
2014 !init_attr.isSplat() || !init_attr.getSplatValue<BoolAttr>().getValue()) in MatchInitValue()
2030 !init_attr.isSplat() || init_attr.getSplatValue<BoolAttr>().getValue()) in MatchInitValue()
2329 !(rw.window_dilations()->isSplat() && in IsSpatialPoolingWithoutDilation()
2334 !(rw.base_dilations()->isSplat() && in IsSpatialPoolingWithoutDilation()
[all …]
|
| /external/tensorflow/tensorflow/compiler/mlir/tensorflow/utils/ |
| D | convert_tensor.cc | 317 if (attr.isSplat()) { in ConvertElementsAttr()
330 if (attr.isSplat()) { in ConvertFloatElementsAttr()
342 if (attr.isSplat()) { in ConvertHalfElementsAttr()
359 if (attr.isSplat()) { in ConvertIntElementsAttr()
373 if (attr.isSplat()) { in ConvertUIntElementsAttr()
383 if (attr.isSplat()) { in ConvertBfloat16ElementsAttr()
|
| D | convert_tensor_test.cc | 178 return attr.cast<mlir::DenseElementsAttr>().isSplat(); in IsSplat()
|
| /external/tensorflow/tensorflow/core/ir/importexport/ |
| D | convert_tensor.cc | 317 if (attr.isSplat()) { in ConvertElementsAttr()
329 if (attr.isSplat()) { in ConvertFloatElementsAttr()
344 if (attr.isSplat()) { in ConvertHalfElementsAttr()
360 if (attr.isSplat()) { in ConvertIntElementsAttr()
373 if (attr.isSplat()) { in ConvertUIntElementsAttr()
383 if (attr.isSplat()) { in ConvertBfloat16ElementsAttr()
|
| /external/tensorflow/tensorflow/compiler/mlir/tensorflow/ir/ |
| D | tf_arith_ops_folder.h | 111 if (rhs_attr.isSplat() && in IdentityArithmeticOpFolder()
119 if (lhs_attr.isSplat() && in IdentityArithmeticOpFolder()
|
| /external/tensorflow/tensorflow/compiler/mlir/quantization/tensorflow/passes/ |
| D | utils.h | 96 !value_attr.isSplat()) { in GetSplatValue()
104 if (!matchPattern(value, m_Constant(&value_attr)) || !value_attr.isSplat()) { in GetSplatValue()
|
| /external/tensorflow/tensorflow/compiler/mlir/lite/quantization/ir/ |
| D | QuantizeUtils.cc | 52 if (realFPElementsAttr.isSplat()) { in convertDenseFPElementsAttr()
|
| /external/tensorflow/tensorflow/dtensor/mlir/expansions/ |
| D | nullary_spmd_expander.cc | 57 if (dense.isSplat()) { in ExpandOp()
|
| /external/tensorflow/tensorflow/compiler/mlir/lite/quantization/tensorflow/ |
| D | fallback_to_flex_ops.cc | 255 if (fp_attr.isSplat()) { in FloatValueEquals()
|
| /external/tensorflow/tensorflow/compiler/mlir/xla/transforms/ |
| D | legalize_tf_collective.cc | 299 if (!group_size_attr.isSplat() || group_size_attr.size() != 1) { in matchAndRewrite()
|
| /external/llvm/lib/Transforms/InstCombine/ |
| D | InstCombineVectorOps.cpp | 1187 bool isSplat = true; in visitShuffleVectorInst() local
1243 isSplat = false; in visitShuffleVectorInst()
1252 if (isSplat || newMask == LHSMask || newMask == RHSMask || newMask == Mask) { in visitShuffleVectorInst()
|
| /external/swiftshader/third_party/llvm-10.0/llvm/lib/IR/ |
| D | Constants.cpp | 47 if (CV->getElementType()->isFloatingPointTy() && CV->isSplat()) in isNegativeZeroValue()
73 if (CV->getElementType()->isFloatingPointTy() && CV->isSplat()) in isZeroValue()
117 if (CV->isSplat()) { in isAllOnesValue()
143 if (CV->isSplat()) { in isOneValue()
193 if (CV->isSplat()) { in isMinSignedValue()
2866 bool ConstantDataVector::isSplat() const { in isSplat() function in ConstantDataVector
2880 return isSplat() ? getElementAsConstant(0) : nullptr; in getSplatValue()
|
| /external/tensorflow/tensorflow/compiler/mlir/lite/transforms/ |
| D | legalize_tf.cc | 455 !padding_attr.isSplat() || in ConvertTFMatrixDiagV2orV3()
462 !padding_attr.isSplat() || in ConvertTFMatrixDiagV2orV3()
|
| D | optimize.cc | 389 if (fp_attr.isSplat()) { in FloatValueEquals()
1220 return elements_attr->isSplat(); in IsScalarizableSplatConstant()
|
| /external/tensorflow/tensorflow/compiler/xla/mlir_hlo/lib/Dialect/mhlo/IR/ |
| D | hlo_ops.cc | 224 if (attr.isSplat() && newType.getElementType().isInteger(/*width=*/1)) { in reshape()
5365 if (!cstAttr.isSplat()) { in matchAndRewrite()
5551 if (predicate.isSplat()) { in fold()
5636 if (!size || !size.isSplat()) return {}; in fold()
6706 if (lhsVal && lhsVal.isSplat()) { in fold()
6716 if (rhsVal && rhsVal.isSplat()) { in fold()
6736 if (lhsVal && lhsVal.isSplat()) { in fold()
6746 if (rhsVal && rhsVal.isSplat()) { in fold()
6771 if (lhsVal && lhsVal.isSplat()) { in fold()
6777 if (rhsVal && rhsVal.isSplat()) { in fold()
[all …]
|
| /external/tensorflow/tensorflow/compiler/xla/mlir_hlo/lib/Dialect/gml_st/transforms/ |
| D | bufferize_tiled_loop.cc | 226 if (sizeAttr.isSplat()) return operands.slice(*sizeIt * index, *sizeIt); in getVariadicOperands()
|
| /external/tensorflow/tensorflow/dtensor/mlir/ |
| D | dtensor_allreduce_sum_optimization.cc | 59 if (!attr || !attr.isSplat()) return false; in IsZeroConstant()
|
| /external/tensorflow/tensorflow/compiler/xla/mlir_hlo/lib/Transforms/ |
| D | bufferize.cc | 73 bool allSameElems = elementsAttr.isSplat(); in matchAndRewrite()
|
| /external/swiftshader/third_party/llvm-10.0/llvm/lib/Transforms/InstCombine/ |
| D | InstCombineVectorOps.cpp | 2161 bool isSplat = true; in visitShuffleVectorInst() local
2217 isSplat = false; in visitShuffleVectorInst()
2226 if (isSplat || newMask == LHSMask || newMask == RHSMask || newMask == Mask) { in visitShuffleVectorInst()
|
| /external/llvm/include/llvm/CodeGen/ |
| D | SelectionDAGNodes.h | 1248 bool isSplat() const { return isSplatMask(Mask, getValueType(0)); }
1250 assert(isSplat() && "Cannot get splat index for non-splat!");
|
| /external/llvm/include/llvm/ADT/ |
| D | APInt.h | 415 bool isSplat(unsigned SplatSizeInBits) const;
|
| /external/swiftshader/third_party/llvm-subzero/include/llvm/ADT/ |
| D | APInt.h | 419 bool isSplat(unsigned SplatSizeInBits) const;
|
| /external/swiftshader/third_party/llvm-10.0/llvm/include/llvm/CodeGen/ |
| D | SelectionDAGNodes.h | 1528 bool isSplat() const { return isSplatMask(Mask, getValueType(0)); }
1531 assert(isSplat() && "Cannot get splat index for non-splat!");
|