Home
last modified time | relevance | path

Searched refs:populateUnfuseBatchNormTrainingPattern (Results 1 – 3 of 3) sorted by relevance

/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/mlir_hlo/include/mlir-hlo/Dialect/mhlo/transforms/
H A Drewriters.h110 void populateUnfuseBatchNormTrainingPattern(MLIRContext *context,
118 populateUnfuseBatchNormTrainingPattern(context, patterns); in populateUnfuseBatchNormPatterns()
/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/mlir_hlo/lib/Dialect/mhlo/transforms/
H A Dunfuse_batch_norm_pass.cc42 populateUnfuseBatchNormTrainingPattern(&getContext(), &patterns); in runOnOperation()
H A Dunfuse_batch_norm.cc364 void populateUnfuseBatchNormTrainingPattern(MLIRContext *context, in populateUnfuseBatchNormTrainingPattern() function