Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/compiler/mlir/hlo/lib/Dialect/mhlo/transforms/
Dlhlo_legalize_roots_to_loops.cc50 const ShapeConstraintAnalysis* shape_constraint_analysis) { in elemwiseLowerHelper() argument
56 if (shape_constraint_analysis) { in elemwiseLowerHelper()
58 shape_constraint_analysis->GetLeaderValueWithSameShape(result_memref); in elemwiseLowerHelper()
80 const ShapeConstraintAnalysis* shape_constraint_analysis) { in miscLowerHelper() argument
85 if (shape_constraint_analysis) { in miscLowerHelper()
87 shape_constraint_analysis->GetLeaderValueWithSameShape(result_memref); in miscLowerHelper()
103 const ShapeConstraintAnalysis* shape_constraint_analysis) { in elemwiseLowerHelperOr() argument
105 shape_constraint_analysis); in elemwiseLowerHelperOr()
111 const ShapeConstraintAnalysis* shape_constraint_analysis) { in elemwiseLowerHelperOr() argument
114 shape_constraint_analysis)) || in elemwiseLowerHelperOr()
[all …]