Searched refs:GetStaticOrDynamicDim (Results 1 – 1 of 1) sorted by relevance
| /external/tensorflow/tensorflow/compiler/mlir/hlo/lib/Dialect/mhlo/transforms/ |
| D | lhlo_legalize_to_parallel_loops.cc | 78 Value GetStaticOrDynamicDim(mlir::Location loc, Value shaped_value, in GetStaticOrDynamicDim() function
125 GetStaticOrDynamicDim(loc, operand, i, operand_shape[i], b); in MapWindowIvsToInput()
149 GetStaticOrDynamicDim(loc, shaped_value, dim.index(), dim.value(), b)); in MakeLoopOverShape()
243 Value ub = GetStaticOrDynamicDim(loc, operand, dim.index(), dim.value(), in CreateReduceOpInNestedParallelLoops()
|