Searched refs:MlirEmitterInput (Results 1 – 3 of 3) sorted by relevance
60 struct MlirEmitterInput { struct161 Status EmitUsingElementalIrEmitter(MlirEmitterInput input);167 Status EmitConstant(MlirEmitterInput mlir_input);170 Status EmitCopyForMlir(MlirEmitterInput input);175 Status EmitCustomCallFromMlir(MlirEmitterInput input);176 Status EmitConvolutionThunkFromMlir(MlirEmitterInput input);177 Status EmitGemmThunkFromMlir(MlirEmitterInput input);178 Status EmitBatchNormThunkFromMlir(MlirEmitterInput input);180 Status EmitCholeskyThunkFromMlir(MlirEmitterInput input);182 Status EmitCustomCallThunkFromMlir(MlirEmitterInput input);[all …]
634 Status IrEmitterUnnested::EmitUsingElementalIrEmitter(MlirEmitterInput input) { in EmitUsingElementalIrEmitter()741 Status IrEmitterUnnested::EmitConstant(MlirEmitterInput mlir_input) { in EmitConstant()814 Status IrEmitterUnnested::EmitPadToStaticFromMlir(MlirEmitterInput mlir_input) { in EmitPadToStaticFromMlir()941 MlirEmitterInput mlir_input) { in EmitSliceToDynamicFromMlir()1067 Status IrEmitterUnnested::EmitCustomCallFromMlir(MlirEmitterInput input) { in EmitCustomCallFromMlir()1109 Status IrEmitterUnnested::EmitConvolutionThunkFromMlir(MlirEmitterInput input) { in EmitConvolutionThunkFromMlir()1223 Status IrEmitterUnnested::EmitGemmThunkFromMlir(MlirEmitterInput input) { in EmitGemmThunkFromMlir()1339 Status IrEmitterUnnested::EmitBatchNormThunkFromMlir(MlirEmitterInput input) { in EmitBatchNormThunkFromMlir()1485 Status IrEmitterUnnested::EmitCholeskyThunkFromMlir(MlirEmitterInput input) { in EmitCholeskyThunkFromMlir()1534 Status IrEmitterUnnested::EmitCustomCallThunkFromMlir(MlirEmitterInput input) { in EmitCustomCallThunkFromMlir()[all …]
1124 MlirEmitterInput input; in CompileLmhloToExecutable()