| /external/tensorflow/tensorflow/compiler/mlir/lite/experimental/estimators/ |
| D | arithmetic_count_util.h | 29 if (!output_type || !output_type.hasStaticShape()) return false; in GetArithmeticCountForBroadcastableOp()
40 if (!input_type || !input_type.hasStaticShape()) { in GetInputTensorTotalSize()
56 if (weight_type == nullptr || !weight_type.hasStaticShape()) return false; in GetArithmeticCountForConvAndFullyconnectedOp()
61 if (output_type == nullptr || !output_type.hasStaticShape()) return false; in GetArithmeticCountForConvAndFullyconnectedOp()
75 if (bias_type && bias_type.hasStaticShape()) { in GetArithmeticCountForConvAndFullyconnectedOp()
|
| /external/tensorflow/tensorflow/compiler/mlir/lite/ir/ |
| D | tfl_ops.cc | 80 if (!shaped_type.hasStaticShape()) { in VerifyOperandsHaveSameShapesOrBroadcastableShape()
501 if (!result_shape_type.hasStaticShape()) return {}; in ConstFoldUnaryOp()
693 if (!output_type.hasStaticShape()) return false; in IsConcatenationOpConstFoldable()
748 if (shaped_type.hasStaticShape() && shaped_type.getNumElements() == 0) { in fold()
778 if (!opaque_attr.getType().hasStaticShape()) in Verify()
798 if (!input_type.hasStaticShape() || !filter_type.hasStaticShape()) { in Verify()
816 if (!output_type.hasStaticShape()) { in Verify()
874 if (!input_type.hasStaticShape() || !weights_type.hasStaticShape() || in fold()
875 (has_bias && !bias_type.hasStaticShape())) { in fold()
1037 if (!indices_type.hasStaticShape() || !updates_type.hasStaticShape()) { in Verify()
[all …]
|
| /external/tensorflow/tensorflow/compiler/mlir/tensorflow/transforms/ |
| D | fold_broadcast.cc | 143 if (!result_type || !result_type.hasStaticShape()) return failure(); in RewriteOp()
155 if (!broadcast_arg_type || !broadcast_arg_type.hasStaticShape()) continue; in RewriteOp()
162 if (!argument_type || !argument_type.hasStaticShape()) continue; in RewriteOp()
|
| D | constant_fold.cc | 53 if (!tensor_type.hasStaticShape()) { in ShouldBeFolded()
100 return shaped_ty.hasStaticShape() && shaped_ty.getNumElements() == 0 && in ConstantFoldFallbackHook()
|
| D | legalize_hlo.cc | 198 if (!conv_op.lhs().getType().cast<ShapedType>().hasStaticShape() || in IsSupportedConvOp()
199 !conv_op.rhs().getType().cast<ShapedType>().hasStaticShape() || in IsSupportedConvOp()
200 !conv_op.getType().cast<ShapedType>().hasStaticShape()) in IsSupportedConvOp()
309 if (!input_type.hasStaticShape()) return failure(); in matchAndRewrite()
1022 if (!operand_type.hasStaticShape() || in matchAndRewrite()
1023 !start_indices_type.hasStaticShape() || !result_type.hasStaticShape()) { in matchAndRewrite()
1102 if (!operand_type.hasStaticShape() || !indices_type.hasStaticShape() || in matchAndRewrite()
1103 !updates_type.hasStaticShape()) { in matchAndRewrite()
|
| /external/tensorflow/tensorflow/compiler/mlir/hlo/lib/Dialect/mhlo/transforms/ |
| D | unfuse_batch_norm.cc | 47 assert(result_type.hasStaticShape()); in BroadcastToFeatureDim()
96 if (broadcast_to_type.hasStaticShape()) { in MaterializeEpsilon()
142 if (!input_type.hasStaticShape()) { in matchAndRewrite()
|
| D | optimize_mhlo.cc | 63 !gather.operand().getType().cast<ShapedType>().hasStaticShape() || in matchAndRewrite()
65 !gather.start_indices().getType().cast<ShapedType>().hasStaticShape()) { in matchAndRewrite()
|
| D | lower_general_dot.cc | 161 if (!lhs_shape_type.hasStaticShape() || !rhs_shape_type.hasStaticShape()) in matchAndRewrite()
|
| D | materialize_broadcasts.cc | 45 if (!operand_type.hasStaticShape()) return failure(); in matchAndRewrite()
|
| /external/tensorflow/tensorflow/compiler/mlir/tensorflow/ir/ |
| D | tf_ops_n_z.cc | 437 if (y_value.isZero() && output_type.hasStaticShape()) { in fold()
539 if (!output_type || !output_type.hasStaticShape()) return {}; in fold()
594 if (shape_ty.hasStaticShape()) { in GetReshapeOutputType()
632 if (!tensor_ty.hasStaticShape()) { in GetReshapeOutputType()
680 if (output_ty.hasStaticShape() && tensor_ty.hasStaticShape()) { in Verify()
723 if (result_ty && result_ty.hasStaticShape() && in fold()
857 if (result_ranked_type.hasStaticShape() && in VerifyShapeOperandAndResult()
864 } else if (result_ranked_type.hasStaticShape()) { in VerifyShapeOperandAndResult()
891 if (!ranked_ty || !ranked_ty.hasStaticShape()) return {}; in ConvertShapeToAttr()
1040 if (!input_type.hasStaticShape()) return {}; in fold()
[all …]
|
| D | tf_attributes.h | 64 bool hasStaticShape() const;
|
| D | tf_attributes.cc | 105 bool ShapeAttr::hasStaticShape() const { in hasStaticShape() function in mlir::TF::ShapeAttr
|
| /external/tensorflow/tensorflow/compiler/mlir/lite/transforms/ |
| D | legalize_tf.cc | 106 if (!shaped_type || !shaped_type.hasStaticShape()) { in HasSameStaticShapes()
593 if (!lhs.getType().cast<ShapedType>().hasStaticShape() || in matchAndRewrite()
594 !rhs.getType().cast<ShapedType>().hasStaticShape()) { in matchAndRewrite()
660 if (!lhs.getType().cast<ShapedType>().hasStaticShape() || in matchAndRewrite()
661 !rhs.getType().cast<ShapedType>().hasStaticShape() || in matchAndRewrite()
662 !cond.getType().cast<ShapedType>().hasStaticShape()) { in matchAndRewrite()
|
| D | prepare_composite_functions_tf.cc | 209 if (!input_type.hasStaticShape()) { in CheckFusableKerasLstm()
222 if (!input_type.hasStaticShape()) { in CheckFusableKerasLstm()
|
| /external/tensorflow/tensorflow/compiler/mlir/hlo/lib/Dialect/mhlo/IR/ |
| D | hlo_ops.cc | 359 if (!result_ty.hasStaticShape()) { in matchAndRewrite()
533 if (!result_ty.hasStaticShape()) return {}; in fold()
819 if (!attrs[0] || !type.hasStaticShape()) return {}; in fold()
911 if (!type || !type.hasStaticShape()) { in matchAndRewrite()
1231 if (!type.hasStaticShape()) return {}; in fold()
1297 if (result_type && output_shape_type && output_shape_type.hasStaticShape() && in Verify()
1313 if (!type || !type.hasStaticShape()) { in matchAndRewrite()
1340 if (!type || type.getRank() != 1 || type.hasStaticShape()) { in matchAndRewrite()
1346 operand_type.hasStaticShape()) { in matchAndRewrite()
1993 !return_type.hasStaticShape()) in fold()
[all …]
|
| /external/llvm-project/mlir/lib/Conversion/LinalgToSPIRV/ |
| D | LinalgToSPIRV.cpp | 81 if (!originalInputType.hasStaticShape() || originalInputType.getRank() != 1) in matchAsPerformingReduction()
84 if (!originalOutputType.hasStaticShape() || in matchAsPerformingReduction()
|
| /external/llvm-project/mlir/lib/Conversion/GPUCommon/ |
| D | GPUOpsLowering.h | 40 assert(type && type.hasStaticShape() && "unexpected type in attribution"); in matchAndRewrite()
124 assert(type && type.hasStaticShape() && in matchAndRewrite()
|
| /external/llvm-project/mlir/lib/IR/ |
| D | BuiltinTypes.cpp | 205 assert(hasStaticShape() && "cannot get element count of dynamic shaped type"); in getNumElements()
239 assert(hasStaticShape() && in getSizeInBits()
271 bool ShapedType::hasStaticShape() const { in hasStaticShape() function in ShapedType
275 bool ShapedType::hasStaticShape(ArrayRef<int64_t> shape) const { in hasStaticShape() function in ShapedType
276 return hasStaticShape() && getShape() == shape; in hasStaticShape()
|
| /external/llvm-project/mlir/lib/Dialect/Linalg/Transforms/ |
| D | DropUnitDims.cpp | 428 if (!srcType.hasStaticShape() || !dstType.hasStaticShape() || in matchAndRewrite()
429 !parentSrcType.hasStaticShape() || in matchAndRewrite()
|
| D | Vectorization.cpp | 280 if (!operand.getType().cast<ShapedType>().hasStaticShape()) in vectorizeLinalgOpPrecondition()
283 if (!outputTensorType.cast<ShapedType>().hasStaticShape()) in vectorizeLinalgOpPrecondition()
546 if (!inShapeType.hasStaticShape() || !kShapeType.hasStaticShape()) in matchAndRewrite()
|
| /external/tensorflow/tensorflow/compiler/mlir/hlo/include/mlir-hlo/Dialect/mhlo/IR/ |
| D | infer_fusibility_op_interface.td | 130 if (!first_ty || !first_ty.hasStaticShape() ||
131 !second_ty || !second_ty.hasStaticShape() ||
|
| /external/llvm-project/mlir/lib/Dialect/SPIRV/ |
| D | SPIRVLowering.cpp | 185 if (!memRefType.hasStaticShape() || in getTypeNumBytes()
211 if (!tensorType.hasStaticShape()) { in getTypeNumBytes()
308 if (!type.hasStaticShape()) { in convertTensorType()
376 if (!type.hasStaticShape()) { in convertMemrefType()
|
| /external/tensorflow/tensorflow/compiler/mlir/tools/kernel_gen/transforms/ |
| D | bufferize.cc | 44 if (!result_type || !result_type.hasStaticShape() || result_rank > 1) in matchAndRewrite()
|
| D | buffer_reuse_pass.cc | 77 if (a_ty && b_ty && a_ty.hasStaticShape() && b_ty.hasStaticShape() && in build()
|
| /external/llvm-project/mlir/lib/Dialect/StandardOps/Transforms/ |
| D | ExpandOps.cpp | 82 if (!shapeType.hasStaticShape()) in matchAndRewrite()
227 return !op.shape().getType().cast<MemRefType>().hasStaticShape(); in runOnFunction()
|