Searched refs:shape_ty (Results 1 – 5 of 5) sorted by relevance
| /external/tensorflow/tensorflow/compiler/mlir/lite/ |
| D | flatbuffer_import.cc | 731 auto shape_ty = in ConvertOp() local
733 if (shape_ty != nullptr && shape_ty.hasRank() && shape_ty.getRank() > 1) { in ConvertOp()
|
| /external/tensorflow/tensorflow/compiler/mlir/lite/ir/ |
| D | tfl_ops.cc | 1356 auto shape_ty = shape.getType().dyn_cast<RankedTensorType>(); in GetReshapeOutputType() local
1357 if (!shape_ty) return success(); in GetReshapeOutputType()
1358 if (shape_ty.getRank() != 1) in GetReshapeOutputType()
1360 "requires 'shape' to be rank 1, but got {0}", shape_ty.getRank())); in GetReshapeOutputType()
1366 if (shape_ty.hasStaticShape()) { in GetReshapeOutputType()
1367 llvm::SmallVector<int64_t, 8> dynamic_shape(shape_ty.getDimSize(0), in GetReshapeOutputType()
|
| /external/tensorflow/tensorflow/compiler/mlir/tensorflow/ir/ |
| D | tf_ops_n_z.cc | 584 auto shape_ty = shape.getType().dyn_cast<RankedTensorType>(); in GetReshapeOutputType() local
585 if (!shape_ty) return success(); in GetReshapeOutputType()
586 if (shape_ty.getRank() != 1) in GetReshapeOutputType()
588 "requires 'shape' to be rank 1, but got {0}", shape_ty.getRank())); in GetReshapeOutputType()
594 if (shape_ty.hasStaticShape()) { in GetReshapeOutputType()
595 llvm::SmallVector<int64_t, 8> dynamic_shape(shape_ty.getDimSize(0), in GetReshapeOutputType()
|
| /external/tensorflow/tensorflow/compiler/mlir/hlo/lib/Dialect/mhlo/transforms/ |
| D | chlo_legalize_to_hlo.cc | 71 Type shape_ty = in matchAndRewrite() local
74 rewriter.create<tensor::CastOp>(loc, shape_ty, uncasted_shape); in matchAndRewrite()
|
| /external/tensorflow/tensorflow/compiler/mlir/xla/transforms/ |
| D | legalize_tf.cc | 5503 auto shape_ty = shape.getType().cast<ShapedType>(); in matchAndRewrite() local
5506 if (!result_ty.hasRank() || !tensor_ty.hasRank() || !shape_ty.hasRank()) { in matchAndRewrite()
|