Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/compiler/xla/service/gpu/
Dir_emitter_unnested.h134 using EmitElementFunction = std::function<void( variable
553 const IrEmitterUnnested::EmitElementFunction& emit_elem_function);
Dir_emitter_unnested.cc3974 const IrEmitterUnnested::EmitElementFunction* emit_elem_function) { in UnrollInnerTileLoop()
4005 const IrEmitterUnnested::EmitElementFunction& emit_elem_function) { in EmitTile()
4880 EmitElementFunction element_generator = in EmitHlo021Tile()
5436 EmitElementFunction emit_reduction_tile = in EmitIRForReduction()