Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/core/ir/types/
Ddialect.h58 if (auto tensor_ty = type.dyn_cast<TensorType>()) in IsValidTFTensorType() local
Ddialect.cc101 TensorType tensor_ty; in ParseTypeWithSubtype() local
/external/tensorflow/tensorflow/compiler/mlir/tensorflow/ir/
Dtf_ops_n_z.cc711 auto tensor_ty = tensor.getType().cast<TensorType>(); in GetReshapeOutputType() local
810 auto tensor_ty = op.tensor().getType().cast<TensorType>(); in Verify() local
1984 auto tensor_ty = shape_op.input().getType().dyn_cast<RankedTensorType>(); in fold() local
2285 auto tensor_ty = op.tensor().getType().dyn_cast<RankedTensorType>(); in Verify() local
/external/tensorflow/tensorflow/compiler/mlir/tfr/ir/
Dtfr_ops.cc910 if (auto tensor_ty = type.dyn_cast<TFRTensorType>()) { in printType() local
/external/tensorflow/tensorflow/compiler/mlir/hlo/lib/Dialect/mhlo/transforms/
Drank_specialization.cc304 auto tensor_ty = ty.dyn_cast<TensorType>(); in DeriveRankedTensorTypes() local
/external/tensorflow/tensorflow/compiler/mlir/xla/transforms/
Dlegalize_tf.cc4649 auto tensor_ty = in matchAndRewrite() local