| /external/tensorflow/tensorflow/compiler/mlir/hlo/lib/Dialect/mhlo/transforms/ |
| D | optimize_mhlo.cc | 84 auto result_ty = gather.getResult().getType().dyn_cast<RankedTensorType>(); in matchAndRewrite() local
87 if (!result_ty) { in matchAndRewrite()
91 result_ty.getRank()) { in matchAndRewrite()
108 result_ty.getShape().size() + 1) { in matchAndRewrite()
112 for (auto it : llvm::enumerate(result_ty.getShape())) { in matchAndRewrite()
169 RankedTensorType::get(sliceShape, result_ty.getElementType()); in matchAndRewrite()
|
| D | legalize_gather_to_torch_index_select.cc | 71 auto result_ty = gather.getResult().getType().dyn_cast<RankedTensorType>(); in matchAndRewrite() local
72 if (!result_ty) { in matchAndRewrite()
78 result_ty.getRank() - index_vector_dim) { in matchAndRewrite()
|
| D | chlo_legalize_to_hlo.cc | 52 auto result_ty = op.getType().cast<ShapedType>(); in matchAndRewrite() local
55 if (!result_ty.hasRank()) return failure(); in matchAndRewrite()
58 if (result_ty.hasStaticShape()) { in matchAndRewrite()
60 op, DenseElementsAttr::get(result_ty, op.value())); in matchAndRewrite()
72 RankedTensorType::get({result_ty.getRank()}, rewriter.getIndexType()); in matchAndRewrite()
76 op, result_ty, constant, shape, rewriter.getI64TensorAttr({})); in matchAndRewrite()
|
| /external/tensorflow/tensorflow/compiler/mlir/hlo/lib/Dialect/mhlo/IR/ |
| D | hlo_ops.cc | 309 auto result_ty = iota.getType().cast<ShapedType>(); in matchAndRewrite() local
310 if (!result_ty.hasRank() || result_ty.getRank() < 2) { in matchAndRewrite()
317 {result_ty.getDimSize(iota_dimension)}, result_ty.getElementType()); in matchAndRewrite()
325 rewriter.replaceOpWithNewOp<BroadcastInDimOp>(iota, result_ty, new_iota, in matchAndRewrite()
338 auto result_ty = getResult().getType().cast<ShapedType>(); in fold() local
339 if (result_ty.hasRank() && result_ty.getDimSize(dimension) == 1) { in fold()
341 return builder.getZeroAttr(result_ty); in fold()
358 auto result_ty = iota.getType().cast<ShapedType>(); in matchAndRewrite() local
359 if (!result_ty.hasStaticShape()) { in matchAndRewrite()
363 rewriter.replaceOpWithNewOp<IotaOp>(iota, result_ty, iota.iota_dimension()); in matchAndRewrite()
[all …]
|
| /external/tensorflow/tensorflow/compiler/mlir/lite/transforms/ |
| D | dilated_conv.h | 99 auto result_ty = op.getResult().getType().template cast<TensorType>(); in matchAndRewrite() local
100 if (!result_ty.hasRank() || result_ty.getRank() != 4 || in matchAndRewrite()
101 result_ty.isDynamicDim(0) || result_ty.isDynamicDim(3)) in matchAndRewrite()
|
| D | lower_static_tensor_list.cc | 213 auto result_ty = in matchAndRewrite() local
240 op, result_ty, values, /*axis=*/rewriter.getI64IntegerAttr(0)); in matchAndRewrite()
|
| /external/tensorflow/tensorflow/compiler/mlir/tensorflow/transforms/ |
| D | contraction_fusion.cc | 83 Type result_ty = fuse_into->getResult(0).getType(); in matchAndRewrite() local
126 Value fused = rewriter.create<FusedOp>(loc, result_ty, operands, attrs); in matchAndRewrite()
|
| D | lower_tf.cc | 1299 auto result_ty = op.getType().cast<ShapedType>(); in matchAndRewrite() local
1501 rewriter.create<ReshapeOp>(loc, result_ty, gathered_values, out_shape); in matchAndRewrite()
|
| /external/tensorflow/tensorflow/compiler/mlir/xla/ir/ |
| D | mlir_hlo_builder.cc | 445 mlir::Type result_ty, in TriangularSolveInternal()
448 loc_, result_ty, GetValue(a), GetValue(b), in TriangularSolveInternal()
460 mlir::Type result_ty, in CholeskyInternal()
463 loc_, result_ty, GetValue(a), builder_.getBoolAttr(lower)); in CholeskyInternal()
521 mlir::Type result_ty, in DynamicSliceInternal()
524 loc_, result_ty, GetValue(operand), GetValues(start_indices), in DynamicSliceInternal()
532 mlir::Type result_ty, in DynamicUpdateSliceInternal()
535 loc_, result_ty, GetValue(operand), GetValue(update), in DynamicUpdateSliceInternal()
|
| /external/tensorflow/tensorflow/compiler/mlir/tensorflow/ir/ |
| D | tf_ops_helpers.inc | 464 ShapedType result_ty) -> bool {
468 if (scalar_identity) return operand_ty == result_ty;
475 return operand_ty == result_ty && identity_ty == result_ty &&
476 result_ty.hasStaticShape();
|
| D | tf_ops_n_z.cc | 722 auto result_ty = getType().dyn_cast<ShapedType>(); in fold() local
723 if (result_ty && result_ty.hasStaticShape() && in fold()
724 result_ty == tensor.getType()) { in fold()
1552 auto result_ty = getType().template dyn_cast<RankedTensorType>(); in fold() local
1553 if (!result_ty) return {}; in fold()
1557 if (!keep_dims() && input_ty == result_ty) { in fold()
|
| D | tf_ops_a_m.cc | 575 auto result_ty = getType().dyn_cast<ShapedType>(); in fold() local
576 if (result_ty && result_ty.hasStaticShape() && result_ty == input.getType()) { in fold()
|
| /external/tensorflow/tensorflow/compiler/mlir/xla/transforms/ |
| D | legalize_tf.cc | 1174 auto result_ty = op.getType().template dyn_cast<RankedTensorType>(); in matchAndRewrite() local
1178 for (RankedTensorType ty : {input_ty, filter_ty, result_ty}) in matchAndRewrite()
3163 RankedTensorType result_ty, in rewriteWithUnknownBegin() argument
3229 int output_elements = result_ty.getNumElements(); in rewriteWithUnknownBegin()
3309 auto result_ty = op.getType().dyn_cast<RankedTensorType>(); in matchAndRewrite() local
3310 if (!result_ty || !result_ty.hasStaticShape()) return failure(); in matchAndRewrite()
3315 return rewriteWithUnknownBegin(op, input_ty, result_ty, rewriter); in matchAndRewrite()
5478 auto result_ty = op.getResult().getType().dyn_cast<RankedTensorType>(); in matchAndRewrite() local
5479 if (!result_ty) { in matchAndRewrite()
5484 RankedTensorType::get(result_ty.getShape(), rewriter.getIndexType()); in matchAndRewrite()
[all …]
|