Home
last modified time | relevance | path

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

/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/mlir_hlo/lib/Dialect/gml_st/transforms/
H A Dfusion_interface_impl.cc60 auto operandDimSize = std::get<1>(e); in operandMaterializesToPoint() local
/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/mlir_hlo/lib/Dialect/lhlo/IR/
H A Dlhlo_ops.cc310 int64_t lowPad, highPad, interiorPad, operandDimSize, outputDimSize; in verify() local
/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/mlir_hlo/lib/Dialect/mhlo/transforms/
H A Dhlo_legalize_to_memref.cc246 Value operandDimSize = in insertDynamicMemrefCastOp() local
H A Dlegalize_to_linalg.cc890 Value operandDimSize = in matchAndRewrite() local
3108 Value operandDimSize = in matchAndRewrite() local
/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/mlir_hlo/stablehlo/dialect/
H A DStablehloOps.cpp1479 auto operandDimSize = operandShape.getDimSize(it.index()); in verifyStaticGather() local
/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/mlir_hlo/lib/Dialect/mhlo/IR/
H A Dhlo_ops.cc1591 auto operandDimSize = operandShape.getDimSize(it.index()); in verifyStaticGather() local