Home
last modified time | relevance | path

Searched defs:featureGroupCount (Results 1 – 3 of 3) sorted by relevance

/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/mlir_hlo/lib/Dialect/mhlo/transforms/
H A Dlegalize_to_linalg.cc2009 auto featureGroupCount = op.feature_group_count(); in matchAndRewrite() local
/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/mlir_hlo/lib/Dialect/mhlo/IR/
H A Dhlo_ops.cc2258 const int64_t featureGroupCount = op.feature_group_count(); in verifyConvolutionAttributes() local
2429 int64_t featureGroupCount = op.feature_group_count(); in matchAndRewrite() local
/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/mlir_hlo/stablehlo/dialect/
H A DStablehloOps.cpp1963 const int64_t featureGroupCount = op.feature_group_count(); in verifyConvolutionAttributes() local