Searched refs:shaped_value (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() argument
81 ? b->create<DimOp>(loc, shaped_value, dim_index).getResult() in GetStaticOrDynamicDim()
139 scf::ParallelOp MakeLoopOverShape(Location loc, Value shaped_value, in MakeLoopOverShape() argument
145 shaped_value.getType().cast<ShapedType>().getShape(); in MakeLoopOverShape()
149 GetStaticOrDynamicDim(loc, shaped_value, dim.index(), dim.value(), b)); in MakeLoopOverShape()
|