Searched refs:reshape_type (Results 1 – 5 of 5) sorted by relevance
| /external/tensorflow/tensorflow/compiler/mlir/tensorflow/transforms/ |
| D | optimize.cc | 59 auto reshape_type = reshape_op.output().getType().cast<ShapedType>(); in matchAndRewrite() local
61 if (!reshape_type.hasStaticShape()) return failure(); in matchAndRewrite()
62 ArrayRef<int64_t> reshape_shape = reshape_type.getShape(); in matchAndRewrite()
|
| D | tpu_space_to_depth_pass.cc | 243 auto reshape_type = RankedTensorType::get( in GetReshapeOpForConv2DFilter() local
245 auto reshape_sizes = DenseIntElementsAttr::get(reshape_type, new_shape); in GetReshapeOpForConv2DFilter()
|
| D | legalize_hlo.cc | 745 auto reshape_type = RankedTensorType::get(reshape_shape, element_type); in matchAndRewrite() local
748 result = rewriter.create<TF::ReshapeOp>(iota_op.getLoc(), reshape_type, in matchAndRewrite()
|
| /external/tensorflow/tensorflow/compiler/mlir/hlo/lib/Dialect/mhlo/IR/ |
| D | hlo_ops.cc | 250 auto reshape_type = RankedTensorType::get(reshape_shape, element_type); in matchAndRewrite() local
252 rewriter.create<ReshapeOp>(gather.getLoc(), reshape_type, result); in matchAndRewrite()
|
| /external/tensorflow/tensorflow/compiler/mlir/tosa/transforms/ |
| D | legalize_tfl.cc | 1476 RankedTensorType reshape_type = RankedTensorType::get( in matchAndRewrite() local
1479 op->getLoc(), reshape_type, tfl_fc_op.input(), in matchAndRewrite()
|