Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/core/ir/types/
Ddialect.h60 if (auto tensor_ty = type.dyn_cast<TensorType>()) in IsValidTFTensorType() local
Ddialect.cc84 TensorType tensor_ty; in ParseTypeWithSubtype() local
/external/tensorflow/tensorflow/compiler/mlir/tensorflow/ir/
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
/external/tensorflow/tensorflow/compiler/mlir/tfr/ir/
Dtfr_ops.cc978 if (auto tensor_ty = type.dyn_cast<TFRTensorType>()) { in printType() local
/external/tensorflow/tensorflow/compiler/mlir/xla/transforms/
Dlegalize_tf.cc4450 auto tensor_ty = in matchAndRewrite() local
5380 if (auto tensor_ty = ty.dyn_cast<RankedTensorType>()) { in matchAndRewrite() local