Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/compiler/mlir/xla/transforms/
Dlegalize_tf_with_tf2xla.cc406 auto ranked_ty = ty.dyn_cast<ShapedType>(); in LegalizeOp() local
Dlegalize_tf.cc253 RankedTensorType ranked_ty = ty.dyn_cast<RankedTensorType>(); in GetDimSize() local
/external/tensorflow/tensorflow/compiler/mlir/tensorflow/ir/
Dtf_types.cc382 if (auto ranked_ty = ty.dyn_cast<RankedTensorType>()) { in DropTypeHelper() local
Dtf_ops_n_z.cc890 auto ranked_ty = input_ty.dyn_cast<RankedTensorType>(); in ConvertShapeToAttr() local
/external/tensorflow/tensorflow/compiler/mlir/hlo/lib/Dialect/mhlo/IR/
Dhlo_ops.cc518 if (auto ranked_ty = operand_ty.dyn_cast<RankedTensorType>()) { in build() local
1754 auto ranked_ty = operand_ty.dyn_cast<RankedTensorType>(); in GetReduceResultType() local
2370 RankedTensorType ranked_ty = ty.dyn_cast<RankedTensorType>(); in inferReturnTypes() local
/external/tensorflow/tensorflow/compiler/mlir/tensorflow/transforms/
Dlower_tf.cc117 auto ranked_ty = ty.dyn_cast<RankedTensorType>(); in InferExpandDimsType() local