Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/compiler/mlir/hlo/lib/Dialect/mhlo/transforms/
Dlegalize_to_linalg.cc73 ShapedType GetHloOpResultType(Operation* op) { in GetHloOpResultType() function
374 auto result_type = GetHloOpResultType<isLHLO>(op); in matchAndRewrite()
417 unsigned nloops = GetHloOpResultType<isLHLO>(broadcast_op).getRank(); in getIndexingMaps()
451 auto result_type = GetHloOpResultType<false>(broadcast_op); in getIndexingMaps()
699 GetHloOpResultType<isLHLO>(op).template cast<ShapedType>(); in getIndexingMaps()
728 ShapedType result_type = GetHloOpResultType<isLHLO>(reshape_op); in matchAndRewrite()
841 ShapedType result_shaped_type = GetHloOpResultType<isLHLO>(iota_op); in matchAndRewrite()
1020 GetHloOpResultType<isLHLO>(op).template cast<ShapedType>(); in getIndexingMaps()