| /external/tensorflow/tensorflow/compiler/mlir/tools/kernel_gen/transforms/ |
| D | lower_index_cast_pass.cc | 39 auto result_ty = op.getType().dyn_cast<ShapedType>(); in matchAndRewrite() local
|
| D | tf_framework_legalize_to_llvm.cc | 387 Type result_ty = in matchAndRewrite() local
|
| /external/tensorflow/tensorflow/compiler/mlir/hlo/lib/Dialect/mhlo/transforms/ |
| D | legalize_gather_to_torch_index_select.cc | 72 auto result_ty = gather.getResult().getType().dyn_cast<RankedTensorType>(); in matchAndRewrite() local
|
| D | optimize_mhlo.cc | 84 auto result_ty = gather.getResult().getType().dyn_cast<RankedTensorType>(); in matchAndRewrite() local
|
| D | mhlo_mark_shape_calc.cc | 180 auto result_ty = op->getResult(0).getType().dyn_cast<RankedTensorType>(); in MarkRegardAsShapeCalcOps() local
|
| D | chlo_legalize_to_hlo.cc | 53 auto result_ty = op.getType().cast<ShapedType>(); in matchAndRewrite() local
1029 auto result_ty = x.getType().cast<ShapedType>(); in MaterializeNextAfter() local
1140 auto result_ty = x.getType().cast<ShapedType>(); in MaterializeSinhApproximationForLargeX() local
1454 auto result_ty = op.getType().cast<ShapedType>(); in matchAndRewrite() local
|
| D | legalize_to_linalg.cc | 362 auto result_ty = this->typeConverter in matchAndRewrite() local
548 Optional<ShapedType> result_ty; in matchAndRewrite() local
2632 auto result_ty = this->typeConverter->convertType(op.getResult().getType()) in matchAndRewrite() local
|
| /external/tensorflow/tensorflow/compiler/mlir/hlo/lib/Dialect/mhlo/IR/ |
| D | hlo_ops.cc | 385 auto result_ty = in GatherShapeInferImpl() local
526 auto result_ty = iota.getType().cast<ShapedType>(); in matchAndRewrite() local
555 auto result_ty = getResult().getType().cast<ShapedType>(); in fold() local
575 auto result_ty = iota.getType().cast<ShapedType>(); in matchAndRewrite() local
592 auto result_ty = iota.getType().cast<ShapedType>(); in matchAndRewrite() local
646 ShapedType result_ty = shape::getExtentTensorType( in castToIndexTensor() local
722 Type result_ty; in inferReturnTypes() local
747 Type result_ty; in build() local
759 auto result_ty = getResult().getType().cast<TensorType>(); in fold() local
1275 Type result_ty; in inferReturnTypes() local
[all …]
|
| /external/tensorflow/tensorflow/compiler/mlir/lite/transforms/ |
| D | dilated_conv.h | 108 auto result_ty = op.getResult().getType().template cast<TensorType>(); in matchAndRewrite() local
|
| D | lower_static_tensor_list.cc | 282 auto result_ty = in matchAndRewrite() local
|
| /external/tensorflow/tensorflow/compiler/mlir/tensorflow/ir/ |
| D | tf_ops_n_z.cc | 853 auto result_ty = getType().dyn_cast<ShapedType>(); in fold() local
1607 auto result_ty = getType().template dyn_cast<RankedTensorType>(); in fold() local
3255 TensorType result_ty; in inferReturnTypes() local
|
| D | tf_ops_a_m.cc | 568 auto result_ty = getType().dyn_cast<ShapedType>(); in fold() local
|
| /external/tensorflow/tensorflow/compiler/mlir/xla/transforms/ |
| D | legalize_tf.cc | 1313 auto result_ty = op.getType().template dyn_cast<RankedTensorType>(); in matchAndRewriteDynamicConv() local
1472 auto result_ty = op.getType().template dyn_cast<RankedTensorType>(); in matchAndRewrite() local
3847 RankedTensorType result_ty, in rewriteWithUnknownBegin()
3993 auto result_ty = op.getType().dyn_cast<RankedTensorType>(); in matchAndRewrite() local
6618 auto result_ty = op.getResult().getType().dyn_cast<RankedTensorType>(); in matchAndRewrite() local
6640 auto result_ty = op.getType().cast<ShapedType>(); in matchAndRewrite() local
|
| /external/tensorflow/tensorflow/compiler/mlir/tensorflow/transforms/ |
| D | lower_tf.cc | 1335 auto result_ty = op.getType().cast<ShapedType>(); in matchAndRewrite() local
|
| /external/tensorflow/tensorflow/compiler/mlir/lite/ir/ |
| D | tfl_ops.cc | 1967 auto result_ty = getType().dyn_cast<RankedTensorType>(); in fold() local
|