Searched defs:shapeValues (Results 1 – 2 of 2) sorted by relevance
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/mlir_hlo/stablehlo/dialect/ |
| H A D | StablehloOps.cpp | 1621 SmallVector<Value, 4> shapeValues; in reifyGatherShape() local
2585 SmallVector<int64_t> shapeValues(dimensionAttr.getValues<int64_t>()); in inferReturnTypeComponents() local
2603 SmallVector<Value, 4> shapeValues; in reifyReturnTypeShapes() local
3080 auto& shapeValues = allShapeValues[0]; in reifyReturnTypeShapes() local
3230 SmallVector<Value, 4> shapeValues; in reifyReturnTypeShapes() local
4083 SmallVector<Value, 4> shapeValues; in reifyReturnTypeShapes() local
4486 SmallVector<Value, 4> shapeValues; in reifyReturnTypeShapes() local
4919 SmallVector<Value, 4> shapeValues(permutation.size()); in reifyReturnTypeShapes() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/mlir_hlo/lib/Dialect/mhlo/IR/ |
| H A D | hlo_ops.cc | 1733 SmallVector<Value, 4> shapeValues; in reifyGatherShape() local
3132 SmallVector<int64_t> shapeValues(dimensionAttr.getValues<int64_t>()); in inferReturnTypeComponents() local
3150 SmallVector<Value, 4> shapeValues; in reifyReturnTypeShapes() local
3980 auto& shapeValues = allShapeValues[0]; in reifyReturnTypeShapes() local
4353 SmallVector<Value, 4> shapeValues; in reifyReturnTypeShapes() local
5404 SmallVector<Value, 4> shapeValues; in reifyReturnTypeShapes() local
6059 SmallVector<Value, 4> shapeValues; in reifyReturnTypeShapes() local
7353 SmallVector<Value, 4> shapeValues(permutation.size()); in reifyReturnTypeShapes() local
|