Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/compiler/xla/mlir_hlo/lib/Dialect/mhlo/transforms/
Dlegalize_einsum_to_dot_general.cc97 rhsContractingDims, rhsBatchingDims; in matchAndRewrite() local
Dlower_general_dot.cc197 auto rhsContractingDims = dotNumbers.getRhsContractingDimensions(); in matchAndRewrite() local
Dlegalize_to_linalg.cc1471 auto rhsContractingDims = dimNumbers.getRhsContractingDimensions(); in matchAndRewrite() local
3162 auto rhsContractingDims = dimNumbers.getRhsContractingDimensions(); 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.cpp1052 ArrayRef<int64_t> rhsContractingDims = in verify() local
/external/tensorflow/tensorflow/compiler/xla/mlir_hlo/lib/Dialect/mhlo/IR/
Dhlo_ops.cc1050 ArrayRef<int64_t> rhsContractingDims = in verify() local