Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/compiler/xla/service/gpu/
Dir_emitter_unnested.h191 Status EmitReduceFromMlir(MlirEmitterInput mlir_input);
Dir_emitter_unnested.cc2139 return EmitReduceFromMlir(mlir_input); in HandleReduce()
2142 Status IrEmitterUnnested::EmitReduceFromMlir(MlirEmitterInput mlir_input) { in EmitReduceFromMlir() function in xla::gpu::IrEmitterUnnested