Home
last modified time | relevance | path

Searched defs:dimNumbers (Results 1 – 4 of 4) sorted by relevance

/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/mlir_hlo/lib/Dialect/mhlo/transforms/
H A Dlegalize_einsum_to_dot_general.cc149 auto dimNumbers = mhlo::DotDimensionNumbersAttr::get( in matchAndRewrite() local
H A Dlegalize_to_linalg.cc1467 mhlo::DotDimensionNumbersAttr dimNumbers = op.dot_dimension_numbers(); in matchAndRewrite() local
3158 mhlo::DotDimensionNumbersAttr dimNumbers = op.dot_dimension_numbers(); in matchAndRewrite() local
/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/mlir_hlo/stablehlo/dialect/
H A DStablehloOps.cpp1046 auto dimNumbers = this->dot_dimension_numbers(); in verify() local
1157 auto dimNumbers = dot_dimension_numbers(); in reifyReturnTypeShapes() local
5243 ScatterDimensionNumbersAttr dimNumbers, Location loc) { in validateScatterDimensionNumbers()
/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/mlir_hlo/lib/Dialect/mhlo/IR/
H A Dhlo_ops.cc1044 auto dimNumbers = this->dot_dimension_numbers(); in verify() local
1194 auto dimNumbers = dot_dimension_numbers(); in reifyReturnTypeShapes() local
7825 ScatterDimensionNumbersAttr dimNumbers, Location loc) { in validateScatterDimensionNumbers()