Home
last modified time | relevance | path

Searched defs:lhsBatchingDims (Results 1 – 5 of 5) sorted by relevance

/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/mlir_hlo/lib/Dialect/mhlo/transforms/
H A Dlegalize_einsum_to_dot_general.cc96 SmallVector<int64_t> lhsContractingDims, lhsBatchingDims, in matchAndRewrite() local
H A Dlegalize_to_linalg.cc1468 auto lhsBatchingDims = dimNumbers.getLhsBatchingDimensions(); in matchAndRewrite() local
3159 auto lhsBatchingDims = dimNumbers.getLhsBatchingDimensions(); in matchAndRewrite() local
/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/mlir_hlo/python/
H A DMlirHloModule.cc192 const std::vector<int64_t> &rhsContractingDims, MlirContext ctx) { in PYBIND11_MODULE()
/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/mlir_hlo/stablehlo/dialect/
H A DStablehloOps.cpp1048 ArrayRef<int64_t> lhsBatchingDims = dimNumbers.getLhsBatchingDimensions(); in verify() local
/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/mlir_hlo/lib/Dialect/mhlo/IR/
H A Dhlo_ops.cc1046 ArrayRef<int64_t> lhsBatchingDims = dimNumbers.getLhsBatchingDimensions(); in verify() local