| /external/tensorflow/tensorflow/compiler/mlir/lite/utils/ |
| D | arithmetic_count_util.h | 31 if (!output_type || !output_type.hasStaticShape()) return false; in GetFirstOutputCount()
42 if (!input_type || !input_type.hasStaticShape()) { in GetInputTensorTotalSize()
58 if (weight_type == nullptr || !weight_type.hasStaticShape()) return false; in GetArithmeticCountForConvAndFullyconnectedOp()
63 if (output_type == nullptr || !output_type.hasStaticShape()) return false; in GetArithmeticCountForConvAndFullyconnectedOp()
77 if (bias_type && bias_type.hasStaticShape()) { in GetArithmeticCountForConvAndFullyconnectedOp()
|
| /external/tensorflow/tensorflow/compiler/mlir/lite/ir/ |
| D | tfl_ops.cc | 173 if (!shaped_type.hasStaticShape()) { in VerifyOperandsHaveSameShapesOrBroadcastableShape()
643 if (!result_shape_type.hasStaticShape()) return {}; in ConstFoldUnaryOp()
825 if (!output_type.hasStaticShape()) return false; in IsConcatenationOpConstFoldable()
900 if (shaped_type.hasStaticShape() && shaped_type.getNumElements() == 0) { in fold()
992 if (!input_type.hasStaticShape() || !output_type.hasStaticShape() || in matchAndRewrite()
1027 if (!input_type.hasStaticShape() || !filter_type.hasStaticShape()) { in verify()
1045 if (!output_type.hasStaticShape()) { in verify()
1103 if (!input_type.hasStaticShape() || !weights_type.hasStaticShape() || in fold()
1104 (has_bias && !bias_type.hasStaticShape())) { in fold()
1419 if (!indices_type.hasStaticShape() || !updates_type.hasStaticShape()) { in verify()
[all …]
|
| /external/tensorflow/tensorflow/compiler/xla/mlir_hlo/lib/Dialect/mhlo/transforms/ |
| D | restrict_max_rank.cc | 68 if (!operandTy.hasStaticShape() || !resultTy.hasStaticShape()) in matchAndRewrite()
93 if (!inputTy.hasStaticShape() || !outputTy.hasStaticShape()) in matchAndRewrite()
|
| D | unfuse_batch_norm.cc | 48 assert(resultType.hasStaticShape()); in broadcastToFeatureDim()
91 if (broadcastToType.hasStaticShape()) { in materializeEpsilon()
136 if (!inputType.hasStaticShape()) { in matchAndRewrite()
205 if (!operandType.hasStaticShape()) { in calculateReduceSize()
317 if (!operandType.hasStaticShape()) { in matchAndRewrite()
|
| D | optimize_mhlo.cc | 61 !gather.operand().getType().cast<ShapedType>().hasStaticShape() || in matchAndRewrite()
63 !gather.start_indices().getType().cast<ShapedType>().hasStaticShape()) { in matchAndRewrite()
|
| D | materialize_broadcasts.cc | 45 if (!operandType.hasStaticShape()) return failure(); in matchAndRewrite()
|
| /external/tensorflow/tensorflow/compiler/mlir/lite/experimental/tac/transforms/ |
| D | device_transform_patterns.cc | 292 if (input_type == nullptr || !input_type.hasStaticShape()) return failure(); in matchAndRewrite()
373 if (!input_type || !input_type.hasStaticShape()) return failure(); in matchAndRewrite()
379 if (!begin_type || !begin_type.hasStaticShape()) return failure(); in matchAndRewrite()
383 if (!size_type || !size_type.hasStaticShape()) return failure(); in matchAndRewrite()
386 if (!output_type || !output_type.hasStaticShape()) return failure(); in matchAndRewrite()
474 if (!input_type || !input_type.hasStaticShape()) return failure(); in matchAndRewrite()
479 if (!weight_type || !weight_type.hasStaticShape()) return failure(); in matchAndRewrite()
484 if (!output_type || !output_type.hasStaticShape()) return failure(); in matchAndRewrite()
534 if (!input_type || !input_type.hasStaticShape()) return failure(); in matchAndRewrite()
540 if (!output_type || !output_type.hasStaticShape()) return failure(); in matchAndRewrite()
|
| D | cost_model.cc | 63 if (input_type == nullptr || !input_type.hasStaticShape()) continue; in GetTransferredTensorBytes()
83 if (input_type == nullptr || !input_type.hasStaticShape()) continue; in GetTransferredElementCount()
|
| /external/tensorflow/tensorflow/compiler/mlir/hlo/lib/Transforms/ |
| D | buffer_reuse.cc | 125 if (shapedA.hasStaticShape() && shapedB.hasStaticShape()) in checkReuseCompatibility()
129 if (shapedA.hasStaticShape() != shapedB.hasStaticShape()) return false; in checkReuseCompatibility()
|
| /external/tensorflow/tensorflow/compiler/xla/mlir_hlo/lib/Transforms/ |
| D | buffer_reuse.cc | 127 if (shapedA.hasStaticShape() && shapedB.hasStaticShape()) in checkReuseCompatibility()
131 if (shapedA.hasStaticShape() != shapedB.hasStaticShape()) return false; in checkReuseCompatibility()
|
| D | lower_index_cast_pass.cc | 44 resultTy.hasStaticShape() ? ValueRange{} in matchAndRewrite()
|
| /external/tensorflow/tensorflow/core/transforms/shape_inference/ |
| D | pass.cc | 55 return !shape_type.hasStaticShape(); in CanBeRefined()
93 return (!shape.hasStaticShape() || GetTensorValue(value) != nullptr); in TryToCacheResultsTensorValue()
123 if (!operand_shape.hasStaticShape()) return; in TryToCacheResultsTensorValue()
142 if (!operand_shape.hasStaticShape()) continue; in TryToCacheResultsTensorValue()
177 if (!rt || rt.getRank() != 1 || !rt.hasStaticShape()) return {}; in runOnOperation()
|
| /external/tensorflow/tensorflow/compiler/mlir/tensorflow/transforms/ |
| D | fold_broadcast.cc | 145 if (!result_type || !result_type.hasStaticShape()) return failure(); in RewriteOp()
157 if (!broadcast_arg_type || !broadcast_arg_type.hasStaticShape()) continue; in RewriteOp()
164 if (!argument_type || !argument_type.hasStaticShape()) continue; in RewriteOp()
|
| D | constant_fold.cc | 56 if (!tensor_type.hasStaticShape()) { in ShouldBeFolded()
115 return shaped_ty.hasStaticShape() && shaped_ty.getNumElements() == 0 && in ConstantFoldFallbackHook()
|
| D | ensure_static_shapes_pass.cc | 33 return !shaped_type.hasStaticShape(); in runOnFunction()
|
| D | set_tpu_infeed_layout.cc | 38 if (!ty.hasStaticShape()) return mlir::WalkResult::interrupt(); in SetTPUInfeedLayout()
|
| D | legalize_hlo.cc | 164 if (!conv_op.lhs().getType().cast<ShapedType>().hasStaticShape() || in IsSupportedConvOp()
165 !conv_op.rhs().getType().cast<ShapedType>().hasStaticShape() || in IsSupportedConvOp()
166 !conv_op.getType().cast<ShapedType>().hasStaticShape()) in IsSupportedConvOp()
810 if (!conv_op.lhs().getType().cast<ShapedType>().hasStaticShape() || in IsSupportedConvOp()
811 !conv_op.rhs().getType().cast<ShapedType>().hasStaticShape() || in IsSupportedConvOp()
812 !conv_op.getType().cast<ShapedType>().hasStaticShape()) in IsSupportedConvOp()
1044 if (!input_type.hasStaticShape()) return failure(); in matchAndRewrite()
1207 if (!operand_type.hasStaticShape() || !update_type.hasStaticShape()) in matchAndRewrite()
1320 if (!operand_type || !operand_type.hasStaticShape()) return false; in MatchReshapedIota()
1581 if (!keys_ty || !keys_ty.hasStaticShape() || in matchAndRewrite()
[all …]
|
| /external/tensorflow/tensorflow/compiler/mlir/lite/transforms/ |
| D | optimize.cc | 461 if (!val_type.hasStaticShape()) return false; in IsLastDimensionEqualOne()
673 if (!addend_shape.hasStaticShape()) return failure(); in matchAndRewrite()
743 if (!multiplier_shape.hasStaticShape()) return failure(); in matchAndRewrite()
1176 if (!non_splat_operand_type.hasStaticShape() || in matchAndRewrite()
1229 if (!type.hasStaticShape()) { in IsScalar()
1291 if (!input_type.hasStaticShape() || !output_type.hasStaticShape()) in matchAndRewrite()
1368 if (!input_ty.hasStaticShape() || in matchAndRewrite()
1370 fully_connected_op.keep_num_dims() || !output_ty.hasStaticShape() || in matchAndRewrite()
1381 if (!reshape_input_ty.hasStaticShape() || input_ty.getRank() == 0 || in matchAndRewrite()
1425 if (!input_shape.hasStaticShape() || !output_shape.hasStaticShape() || in matchAndRewrite()
[all …]
|
| D | legalize_tf.cc | 101 if (!shaped_type || !shaped_type.hasStaticShape()) { in HasSameStaticShapes()
133 if (shaped_type.hasStaticShape()) { in GetShape()
692 if (!lhs.getType().cast<ShapedType>().hasStaticShape() || in matchAndRewrite()
693 !rhs.getType().cast<ShapedType>().hasStaticShape()) { in matchAndRewrite()
832 if (!lhs.getType().cast<ShapedType>().hasStaticShape() || in matchAndRewrite()
833 !rhs.getType().cast<ShapedType>().hasStaticShape() || in matchAndRewrite()
834 !cond.getType().cast<ShapedType>().hasStaticShape()) { in matchAndRewrite()
|
| /external/tensorflow/tensorflow/compiler/mlir/tensorflow/ir/ |
| D | tf_ops_n_z.cc | 345 if (!input_ty.hasStaticShape() || !output_ty.hasStaticShape()) { in matchAndRewrite()
510 if (y_value.isZero() && output_type.hasStaticShape()) { in fold()
711 if (!output_type || !output_type.hasStaticShape()) return {}; in fold()
764 if (shape_ty.hasStaticShape()) { in GetReshapeOutputType()
802 if (!tensor_ty.hasStaticShape()) { in GetReshapeOutputType()
851 if (output_ty.hasStaticShape() && tensor_ty.hasStaticShape()) { in verify()
894 if (result_ty && result_ty.hasStaticShape() && in fold()
1024 if (result_ranked_type.hasStaticShape() && in VerifyShapeOperandAndResult()
1031 } else if (result_ranked_type.hasStaticShape()) { in VerifyShapeOperandAndResult()
1065 if (!ranked_ty || !ranked_ty.hasStaticShape()) return {}; in ConvertShapeToAttr()
[all …]
|
| D | tf_arith_ops_folder.h | 78 return operand_ty == result_ty && identity_ty.hasStaticShape() && in IdentityArithmeticOpFolder()
79 result_ty.hasStaticShape() && in IdentityArithmeticOpFolder()
|
| /external/tensorflow/tensorflow/core/transforms/constant_folding/ |
| D | pass.cc | 308 if (!tensor_type.hasStaticShape()) return failure(); in ReplaceOperationWithBroadcastTo()
558 if (!shape || !shape.hasStaticShape()) continue; in IsFoldableUncached()
567 if (!shape || !shape.hasStaticShape()) continue; in IsFoldableUncached()
602 if (!filter_shape.hasStaticShape() || !const_shape.hasStaticShape()) in IsValidConstShapeForMulConvPushDown()
797 if (!input_shape.hasStaticShape()) return failure(); in matchAndRewrite()
836 if (!input_shape.hasStaticShape()) return failure(); in matchAndRewrite()
949 if (!operand_shape.hasStaticShape()) continue; in matchAndRewrite()
1127 bool full_reduction = output_rank == 0 || (indices_shape.hasStaticShape() && in matchAndRewrite()
1141 if (!shape.hasStaticShape() || shape.getNumElements() != 1) in matchAndRewrite()
1188 if (!output_type.hasStaticShape()) return failure(); in matchAndRewrite()
[all …]
|
| /external/tensorflow/tensorflow/compiler/mlir/quantization/tensorflow/passes/ |
| D | utils.cc | 47 return shaped_type.hasStaticShape(); in HasStaticShape()
121 return shaped_ty.hasStaticShape() && shaped_ty.getNumElements() == 0 && in FoldOperation()
|
| /external/tensorflow/tensorflow/compiler/xla/mlir/utils/runtime/ |
| D | constraints.cc | 80 if (constraint == ArgumentConstraint::kShape && shaped.hasStaticShape()) in ResolveArgumentConstraint()
|
| /external/tensorflow/tensorflow/compiler/mlir/tools/kernel_gen/transforms/ |
| D | lower_index_cast_pass.cc | 44 result_ty.hasStaticShape() ? ValueRange{} in matchAndRewrite()
|