Home
last modified time | relevance | path

Searched defs:ranked_ty (Results 1 – 7 of 7) sorted by relevance

/external/tensorflow/tensorflow/compiler/mlir/xla/transforms/
Dlegalize_tf_with_tf2xla.cc539 auto ranked_ty = ty.dyn_cast<ShapedType>(); in LegalizeOp() local
Dlegalize_tf.cc230 RankedTensorType ranked_ty = ty.dyn_cast<RankedTensorType>(); in GetDimSize() local
/external/tensorflow/tensorflow/compiler/mlir/hlo/lib/Dialect/mhlo/transforms/
Drank_specialization.cc295 auto ranked_ty = ty.dyn_cast<RankedTensorType>(); in IsScalarTensorType() local
311 if (auto ranked_ty = ty.dyn_cast<RankedTensorType>()) in DeriveUnrankedTensorTypes() local
/external/tensorflow/tensorflow/core/ir/types/
Ddialect.cc781 if (auto ranked_ty = ty.dyn_cast<RankedTensorType>()) { in DropTypeHelper() local
/external/tensorflow/tensorflow/compiler/mlir/hlo/lib/Dialect/mhlo/IR/
Dhlo_ops.cc749 if (auto ranked_ty = operand_ty.dyn_cast<RankedTensorType>()) { in build() local
2319 auto ranked_ty = operand_ty.dyn_cast<RankedTensorType>(); in GetReduceResultType() local
3315 RankedTensorType ranked_ty = ty.dyn_cast<RankedTensorType>(); in inferReturnTypes() local
/external/tensorflow/tensorflow/compiler/mlir/tensorflow/transforms/
Dlower_tf.cc163 auto ranked_ty = ty.dyn_cast<RankedTensorType>(); in InferExpandDimsType() local
/external/tensorflow/tensorflow/compiler/mlir/tensorflow/ir/
Dtf_ops_n_z.cc1016 auto ranked_ty = input_ty.dyn_cast<RankedTensorType>(); in ConvertShapeToAttr() local