Home
last modified time | relevance | path

Searched refs:refined_element_ty (Results 1 – 1 of 1) sorted by relevance

/external/tensorflow/tensorflow/compiler/mlir/tensorflow/ir/
Dtf_types.cc326 mlir::Type refined_element_ty = GetCastCompatibleType( in GetCastCompatibleType() local
328 if (!refined_element_ty) return nullptr; in GetCastCompatibleType()
331 return mlir::UnrankedTensorType::get(refined_element_ty); in GetCastCompatibleType()
334 return mlir::RankedTensorType::get(b_tt.getShape(), refined_element_ty); in GetCastCompatibleType()
337 return mlir::RankedTensorType::get(a_tt.getShape(), refined_element_ty); in GetCastCompatibleType()
344 return mlir::RankedTensorType::get(refined_shape, refined_element_ty); in GetCastCompatibleType()