Home
last modified time | relevance | path

Searched defs:broadcastDimensions (Results 1 – 7 of 7) sorted by relevance

/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/mlir_hlo/stablehlo/dialect/
H A DChloOps.cpp103 auto broadcastDimensions = broadcastDimensionsAttr.getValues<APInt>(); in getBroadcastType() local
137 DenseIntElementsAttr broadcastDimensions = in InferBroadcastBinaryOpReturnTypeComponents() local
162 auto broadcastDimensions = op->getAttr("broadcast_dimensions") in ReifyBroadcastBinaryOpReturnTypeShapes() local
214 DenseIntElementsAttr broadcastDimensions, in build()
/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/mlir_hlo/lib/Dialect/mhlo/IR/
H A Dchlo_ops.cc130 auto broadcastDimensions = broadcastDimensionsAttr.getValues<APInt>(); in getBroadcastType() local
164 DenseIntElementsAttr broadcastDimensions = in InferBroadcastBinaryOpReturnTypeComponents() local
189 auto broadcastDimensions = op->getAttr("broadcast_dimensions") in ReifyBroadcastBinaryOpReturnTypeShapes() local
241 DenseIntElementsAttr broadcastDimensions, in build()
H A Dhlo_ops.cc7287 DenseIntElementsAttr broadcastDimensions = in eliminateBroadcastInDimTranspose() local
/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/mlir_hlo/lib/utils/
H A Dhlo_utils.cc52 SmallVector<int64_t, 4> broadcastDimensions(minRank); in getBroadcastDimensionsAttr() local
/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/mlir_hlo/lib/Dialect/lhlo/transforms/
H A Dlhlo_elemental_utils.cc157 auto broadcastDimensions = in elementalLowerImplForBroadcastInDimOps() local
/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/mlir_hlo/lib/Dialect/mhlo/transforms/
H A Dbroadcast_propagation.cc50 Attribute broadcastDimensions; member
H A Dchlo_legalize_to_hlo.cc1606 auto broadcastDimensions = op.broadcast_dimensions(); in matchAndRewrite() local