| /external/tensorflow/tensorflow/compiler/mlir/tensorflow/ir/ |
| D | tf_attributes.cc | 82 if (shaped_type.hasRank()) in get()
89 if (hasRank()) return getShape(); in getValue()
93 bool ShapeAttr::hasRank() const { return !getImpl()->unranked; } in hasRank() function in mlir::TF::ShapeAttr
96 assert(hasRank()); in getRank()
101 assert(hasRank()); in getShape()
106 if (!hasRank()) return false; in hasStaticShape()
|
| D | tf_types.cc | 29 if (shaped_type.hasRank()) return shaped_type.getShape(); in GetShape()
330 if (!a_tt.hasRank() && !b_tt.hasRank()) { in GetCastCompatibleType()
333 if (!a_tt.hasRank()) { in GetCastCompatibleType()
336 if (!b_tt.hasRank()) { in GetCastCompatibleType()
|
| D | tf_attributes.h | 53 bool hasRank() const;
|
| D | tf_ops_n_z.cc | 261 if (!value_ty.hasRank() || value_ty.getRank() != expected_rank) return None; in fold()
289 if (!arg_ty || !arg_ty.hasRank() || arg_ty.getNumDynamicDims() != 1 || in fold()
460 if (!ttype.hasRank()) return success(); in Verify()
759 bool then_has_rank = then_tensor.hasRank(); in Verify()
760 bool else_has_rank = else_tensor.hasRank(); in Verify()
1038 if (!output_type.hasRank()) return {}; in fold()
1174 (broadcasted_ty.hasRank() && broadcasted_ty.getRank() != 2)) in Verify()
1235 if (input_type.hasRank() && input_type.getRank() < 1 + block_rank) { in Verify()
1310 if (!input_type.hasRank()) { in inferReturnTypes()
2457 if (!x_type.hasRank()) in build()
|
| D | tf_ops_a_m.cc | 281 if (input_type.hasRank()) { in Verify()
300 if (crops_type.hasRank()) { in Verify()
335 if (output_type.hasRank()) { in Verify()
1232 return ranked && ranked.hasRank() && ranked.getRank() == 0; in GetHoistParams()
1636 if (input_ty.hasRank()) { in inferConvReturnTypes()
1640 if (filter_ty.hasRank()) { in inferConvReturnTypes()
1647 if (input_ty.hasRank() && filter_ty.hasRank()) { in inferConvReturnTypes()
2604 if (!x_type.hasRank()) return success(); in Verify()
|
| /external/tensorflow/tensorflow/compiler/mlir/hlo/lib/Dialect/mhlo/transforms/ |
| D | legalize_gather_to_torch_index_select.cc | 35 if (!start_indices_ty.hasRank()) { in matchAndRewrite()
41 if (!operand_ty.hasRank()) { in matchAndRewrite()
110 if (!dimension_numbers.collapsed_slice_dims().getType().hasRank() || in matchAndRewrite()
|
| D | optimize_mhlo.cc | 62 if (!gather.operand().getType().cast<ShapedType>().hasRank() || in matchAndRewrite()
64 !gather.start_indices().getType().cast<ShapedType>().hasRank() || in matchAndRewrite()
|
| D | test_infer_shaped_type_pass.cc | 52 if (it.value().hasRank()) { in matchAndRewrite()
|
| /external/llvm-project/mlir/lib/IR/ |
| D | TypeUtilities.cpp | 83 if (!sType1.hasRank() || !sType2.hasRank()) in verifyCompatibleShape()
|
| /external/tensorflow/tensorflow/compiler/mlir/lite/utils/ |
| D | tftext_utils.cc | 74 return type && type.hasRank() && type.getRank() == rank; in RankEquals()
94 !input_type.hasRank()) { in VerifyWhitespaceTokenizer()
170 if (input_values.hasRank() && output_values.hasRank() && in VerifyNgrams()
|
| D | nms_utils.cc | 78 if (!boxes_type.hasRank() || boxes_type.getRank() != 2) { in VerifySignature()
|
| /external/tensorflow/tensorflow/compiler/mlir/tensorflow/transforms/ |
| D | shape_inference.cc | 104 if (shape_type.hasRank()) return true; in CanRefineTypeWith()
348 if (!type.hasRank() || type.getRank() != 1) return failure(); in ComputeInputsRequiredForOutput()
385 if (!type.hasRank() || type.getRank() != 1) return nullptr; in ComputeOutputComponent()
913 if (!operand_type.hasRank()) continue; in RefineShapeForPassThroughOps()
914 if (result_type.hasRank() && in RefineShapeForPassThroughOps()
983 if (!current_type.hasRank()) return true; in CanWhileTypeBeRefinedWith()
984 if (!potential_refined_type.hasRank()) return false; in CanWhileTypeBeRefinedWith()
1016 body_result_type.hasRank() ? body_result_type.getShape() in InferShapeForWhile()
1023 result_type.hasRank() ? result_type.getShape() in InferShapeForWhile()
1123 if (inferred.hasRank()) in InferShapeForSingleOperation()
[all …]
|
| D | optimize.cc | 67 if (!input_type.hasRank() || !output_type.hasRank()) return failure(); in matchAndRewrite()
|
| D | lower_tf.cc | 562 if (original_tensor_type.hasRank()) { in matchAndRewrite()
834 if (!paddings_type.hasRank()) { in matchAndRewrite()
1051 if (!block_shape_ty.hasRank() || block_shape_ty.getRank() != 1) { in matchAndRewrite()
1220 if (tensor_type.hasRank()) { in matchAndRewrite()
1309 if (!input_ty.hasRank() || input_ty.getRank() != 4) { in matchAndRewrite()
1314 if (!out_size_ty.hasRank() || out_size_ty.getRank() != 1 || in matchAndRewrite()
1341 if (!input_ty.hasRank() || input_ty.getRank() != 4) { in matchAndRewrite()
|
| D | layout_optimization.cc | 247 return result_type && operand_type && result_type.hasRank() && in MoveTransposeBefore()
248 operand_type.hasRank() && in MoveTransposeBefore()
|
| /external/llvm-project/mlir/lib/Interfaces/ |
| D | InferTypeOpInterface.cpp | 39 if (shapeAndType.hasRank()) in inferReturnTensorTypes()
|
| /external/tensorflow/tensorflow/compiler/mlir/lite/ir/ |
| D | tfl_ops.cc | 74 if (!shaped_type || !shaped_type.hasRank()) { in VerifyOperandsHaveSameShapesOrBroadcastableShape()
794 if (filter_type.hasRank() && filter_type.getRank() != 2) { in Verify()
972 if (!params_type.hasRank() || !indices_type.hasRank()) in BuildGatherOp()
1067 if (output_type.hasRank()) { in Verify()
1178 if (input_type.hasRank()) { in Verify()
1698 if (!val_type.hasRank()) in BuildTopKOp()
1775 if (!input_type || !input_type.hasRank()) { in inferReturnTypes()
2013 if (!input_type.hasRank() || input_type.getRank() < 2) in Verify()
2052 if (!forget_layer_norm_coefficients.hasRank() || in Verify()
2298 if (input_type.hasRank() && input_type.getRank() != 0) { in fold()
[all …]
|
| /external/tensorflow/tensorflow/compiler/mlir/hlo/lib/Dialect/mhlo/IR/ |
| D | hlo_ops.cc | 290 if (!shape.hasRank()) return success(); in Verify()
310 if (!result_ty.hasRank() || result_ty.getRank() < 2) { in matchAndRewrite()
339 if (result_ty.hasRank() && result_ty.getDimSize(dimension) == 1) { in fold()
376 if (!result_ty.hasRank() || result_ty.getRank() < 2) { in matchAndRewrite()
468 if (operand_ty.hasRank()) { in inferReturnTypes()
1121 if (shaped_type.hasRank()) { in inferReturnTypes()
1128 if (!first_type.hasRank()) { in inferReturnTypes()
1141 if (!shaped_ty.hasRank()) { in inferReturnTypes()
1157 if (!type.hasRank()) { in inferReturnTypes()
1260 if (!second_type.hasRank()) { in Verify()
[all …]
|
| /external/llvm-project/mlir/include/mlir/Interfaces/ |
| D | InferTypeOpInterface.h | 62 bool hasRank() const { return ranked; }; in hasRank() function
|
| /external/tensorflow/tensorflow/compiler/mlir/tensorflow/utils/ |
| D | xla_sharding_util.cc | 68 if (input_type.hasRank()) { in CreateSplitOp()
122 if (input_type.hasRank()) { in CreateConcatOp()
518 if (cluster_func_output_type.hasRank()) { in GetOutputTypesForLogicalDeviceComputation()
|
| D | export_utils.cc | 113 if (attr.hasRank()) { in ConvertAttribute()
418 if (shaped_type.hasRank()) { in SetShapeAttribute()
|
| /external/llvm-project/mlir/lib/Conversion/GPUToVulkan/ |
| D | ConvertGPULaunchFuncToVulkanLaunchFunc.cpp | 59 return memRefType.hasRank() && in isSupportedType()
|
| /external/llvm-project/flang/lib/Lower/ |
| D | SymbolMap.h | 112 bool hasRank() const { in hasRank() function
|
| /external/llvm-project/mlir/examples/toy/Ch7/mlir/ |
| D | LowerToAffineLoops.cpp | 32 assert(type.hasRank() && "expected only ranked shapes"); in convertTensorToMemRef()
|
| /external/llvm-project/mlir/examples/toy/Ch6/mlir/ |
| D | LowerToAffineLoops.cpp | 32 assert(type.hasRank() && "expected only ranked shapes"); in convertTensorToMemRef()
|