Searched refs:EmitBatchNormThunkFromMlir (Results 1 – 2 of 2) sorted by relevance
178 Status EmitBatchNormThunkFromMlir(MlirEmitterInput input);
1096 return EmitBatchNormThunkFromMlir(input); in EmitCustomCallFromMlir()1339 Status IrEmitterUnnested::EmitBatchNormThunkFromMlir(MlirEmitterInput input) { in EmitBatchNormThunkFromMlir() function in xla::gpu::IrEmitterUnnested