Home
last modified time | relevance | path

Searched defs:tensor_ty (Results 1 – 5 of 5) sorted by relevance

/aosp_15_r20/external/tensorflow/tensorflow/core/ir/types/
H A Ddialect.h60 if (auto tensor_ty = type.dyn_cast<TensorType>()) in IsValidTFTensorType() local
H A Ddialect.cc84 TensorType tensor_ty; in ParseTypeWithSubtype() local
/aosp_15_r20/external/tensorflow/tensorflow/compiler/mlir/tensorflow/ir/
H A Dtf_ops_n_z.cc750 auto tensor_ty = tensor.getType().cast<TensorType>(); in GetReshapeOutputType() local
850 auto tensor_ty = op.tensor().getType().cast<TensorType>(); in verify() local
2069 auto tensor_ty = shape_op.input().getType().dyn_cast<RankedTensorType>(); in fold() local
2410 auto tensor_ty = op.tensor().getType().dyn_cast<RankedTensorType>(); in verify() local
/aosp_15_r20/external/tensorflow/tensorflow/compiler/mlir/tfr/ir/
H A Dtfr_ops.cc978 if (auto tensor_ty = type.dyn_cast<TFRTensorType>()) { in printType() local
/aosp_15_r20/external/tensorflow/tensorflow/compiler/mlir/xla/transforms/
H A Dlegalize_tf.cc4450 auto tensor_ty = in matchAndRewrite() local
5380 if (auto tensor_ty = ty.dyn_cast<RankedTensorType>()) { in matchAndRewrite() local