Home
last modified time | relevance | path

Searched refs:EmitBatchNormThunkFromMlir (Results 1 – 2 of 2) sorted by relevance

/external/tensorflow/tensorflow/compiler/xla/service/gpu/
Dir_emitter_unnested.h178 Status EmitBatchNormThunkFromMlir(MlirEmitterInput input);
Dir_emitter_unnested.cc1096 return EmitBatchNormThunkFromMlir(input); in EmitCustomCallFromMlir()
1339 Status IrEmitterUnnested::EmitBatchNormThunkFromMlir(MlirEmitterInput input) { in EmitBatchNormThunkFromMlir() function in xla::gpu::IrEmitterUnnested