Searched defs:dimensionNumbers (Results 1 – 6 of 6) sorted by relevance
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/mlir_hlo/lib/Dialect/mhlo/transforms/ |
| H A D | legalize_gather_to_torch_index_select.cc | 51 auto dimensionNumbers = gather.dimension_numbers(); in matchAndRewrite() local
|
| H A D | optimize_mhlo.cc | 57 auto dimensionNumbers = gather.dimension_numbers(); in matchAndRewrite() local
|
| H A D | hlo_legalize_to_lhlo.cc | 177 auto dimensionNumbers = mhlo::DotDimensionNumbersAttr::get( in matchAndRewrite() local
|
| H A D | legalize_to_linalg.cc | 142 mhlo::ConvDimensionNumbersAttr dimensionNumbers) { in hasCanonicalDimensionNumbers()
1993 auto dimensionNumbers = op.dimension_numbers(); in matchAndRewrite() local
2267 const mhlo::ConvDimensionNumbersAttr& dimensionNumbers = in matchAndRewrite() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/mlir_hlo/stablehlo/dialect/ |
| H A D | StablehloOps.cpp | 1346 ShapeAdaptor sliceSizesShape, GatherDimensionNumbersAttr dimensionNumbers, in verifyGather()
1455 GatherDimensionNumbersAttr dimensionNumbers, in verifyStaticGather()
1494 GatherDimensionNumbersAttr dimensionNumbers, in inferGatherShape()
1556 GatherDimensionNumbersAttr dimensionNumbers, in inferGatherReturnTypeComponents()
1667 GatherDimensionNumbersAttr dimensionNumbers = adaptor.dimension_numbers(); in inferReturnTypeComponents() local
1714 GatherDimensionNumbersAttr dimensionNumbers = adaptor.dimension_numbers(); in inferReturnTypeComponents() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/mlir_hlo/lib/Dialect/mhlo/IR/ |
| H A D | hlo_ops.cc | 1458 ShapeAdaptor sliceSizesShape, GatherDimensionNumbersAttr dimensionNumbers, in verifyGather()
1567 GatherDimensionNumbersAttr dimensionNumbers, in verifyStaticGather()
1606 GatherDimensionNumbersAttr dimensionNumbers, in inferGatherShape()
1668 GatherDimensionNumbersAttr dimensionNumbers, in inferGatherReturnTypeComponents()
1779 GatherDimensionNumbersAttr dimensionNumbers = adaptor.dimension_numbers(); in inferReturnTypeComponents() local
1826 GatherDimensionNumbersAttr dimensionNumbers = adaptor.dimension_numbers(); in inferReturnTypeComponents() local
|