Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/compiler/mlir/tensorflow/ir/
Dtf_ops_tensor_helper.cc68 auto ranked_ty = input_ty.dyn_cast<RankedTensorType>(); in InferReductionOpType() local
127 RankedTensorType ranked_ty = ty.dyn_cast<RankedTensorType>(); in VerifyTypesCompatibility() local
Dtf_ops_n_z.cc1064 auto ranked_ty = input_ty.dyn_cast<RankedTensorType>(); in ConvertShapeToAttr() local
/external/tensorflow/tensorflow/compiler/mlir/xla/transforms/
Dlegalize_tf_with_tf2xla.cc546 auto ranked_ty = ty.dyn_cast<ShapedType>(); in LegalizeOp() local
Dlegalize_tf.cc191 RankedTensorType ranked_ty = ty.dyn_cast<RankedTensorType>(); in GetDimSize() local
/external/tensorflow/tensorflow/compiler/mlir/tensorflow/utils/
Dcompile_mlir_util.cc165 auto ranked_ty = ty.dyn_cast_or_null<mlir::RankedTensorType>(); in GetBufferType() local
/external/tensorflow/tensorflow/core/ir/types/
Ddialect.cc844 if (auto ranked_ty = ty.dyn_cast<RankedTensorType>()) { in DropTypeHelper() local
/external/tensorflow/tensorflow/compiler/mlir/tensorflow/transforms/
Dlower_tf.cc153 auto ranked_ty = ty.dyn_cast<RankedTensorType>(); in InferExpandDimsType() local
/external/tensorflow/tensorflow/compiler/mlir/xla/
Dmlir_hlo_to_hlo.cc120 auto ranked_ty = ty.dyn_cast_or_null<mlir::RankedTensorType>(); in IsBoundedOrStatic() local