Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/compiler/xla/service/gpu/
Dir_emitter_unnested.h124 BlockPrologueGenerator block_prologue_generator = {},
127 block_prologue_generator_(std::move(block_prologue_generator)), in tile_element_generator_()
Dir_emitter_unnested.cc3195 const BlockPrologueGenerator& block_prologue_generator = in EmitKernel() local
3197 if (block_prologue_generator) { in EmitKernel()
3198 block_prologue_generator(unnested_hlo, kernel_info); in EmitKernel()