Home
last modified time | relevance | path

Searched refs:populateUnfuseBatchNormInferencePattern (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.h105 void populateUnfuseBatchNormInferencePattern(MLIRContext *context,
117 populateUnfuseBatchNormInferencePattern(context, patterns); in populateUnfuseBatchNormPatterns()
/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/mlir_hlo/lib/Dialect/mhlo/transforms/
H A Dunfuse_batch_norm_pass.cc41 populateUnfuseBatchNormInferencePattern(&getContext(), &patterns); in runOnOperation()
H A Dunfuse_batch_norm.cc359 void populateUnfuseBatchNormInferencePattern(MLIRContext *context, in populateUnfuseBatchNormInferencePattern() function