Searched refs:get_batch_norm_config (Results 1 – 1 of 1) sorted by relevance
1340 auto get_batch_norm_config = [](auto op, mlir::Value output) { in EmitBatchNormThunkFromMlir() local1381 /*config=*/get_batch_norm_config(bn_train, bn_train.output()), in EmitBatchNormThunkFromMlir()1431 /*config=*/get_batch_norm_config(bn_grad, bn_grad.grad_output()), in EmitBatchNormThunkFromMlir()1471 /*config=*/get_batch_norm_config(bn_inference, bn_inference.output()), in EmitBatchNormThunkFromMlir()