Home
last modified time | relevance | path

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

/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/mlir_hlo/tosa/lib/Transforms/
H A Dlegalize_mhlo.cc98 auto rhsShape = rhsType.getShape(); in matchAndRewrite() local
/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/mlir_hlo/lib/Dialect/mhlo/transforms/
H A Dlegalize_to_linalg.cc324 auto rhsShape = rhs.getType().dyn_cast<RankedTensorType>().getShape(); in extractDynamicEinsumSizes() local
1371 ArrayRef<int64_t> rhsShape = in getDotOperationType() local
H A Dchlo_legalize_to_hlo.cc1627 Value rhsShape = rewriter.create<shape::ShapeOfOp>(loc, rhs); in matchAndRewrite() local
/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/mlir_hlo/lib/Dialect/mhlo/IR/
H A Dhlo_ops.cc1127 auto rhsShape = rhsType.getShape(); in verify() local
2436 llvm::SmallVector<int64_t> rhsShape; in matchAndRewrite() local
/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/mlir_hlo/stablehlo/dialect/
H A DStablehloOps.cpp1129 auto rhsShape = rhsType.getShape(); in verify() local