Home
last modified time | relevance | path

Searched defs:lhsContractingDims (Results 1 – 6 of 6) sorted by relevance

/external/tensorflow/tensorflow/compiler/xla/mlir_hlo/lib/Dialect/mhlo/transforms/
Dlegalize_einsum_to_dot_general.cc96 SmallVector<int64_t> lhsContractingDims, lhsBatchingDims, in matchAndRewrite() local
Dlower_general_dot.cc196 auto lhsContractingDims = dotNumbers.getLhsContractingDimensions(); in matchAndRewrite() local
Dlegalize_to_linalg.cc1470 auto lhsContractingDims = dimNumbers.getLhsContractingDimensions(); in matchAndRewrite() local
3161 auto lhsContractingDims = dimNumbers.getLhsContractingDimensions(); in matchAndRewrite() local
/external/tensorflow/tensorflow/compiler/xla/mlir_hlo/python/
DMlirHloModule.cc192 const std::vector<int64_t> &rhsContractingDims, MlirContext ctx) { in PYBIND11_MODULE()
/external/tensorflow/tensorflow/compiler/xla/mlir_hlo/stablehlo/dialect/
DStablehloOps.cpp1050 ArrayRef<int64_t> lhsContractingDims = in verify() local
/external/tensorflow/tensorflow/compiler/xla/mlir_hlo/lib/Dialect/mhlo/IR/
Dhlo_ops.cc1048 ArrayRef<int64_t> lhsContractingDims = in verify() local