Searched refs:kernel_generator (Results 1 – 4 of 4) sorted by relevance
| /external/tensorflow/tensorflow/compiler/xla/service/gpu/ |
| D | ir_emitter_unnested.h | 249 const KernelCodeGenerator& kernel_generator,
|
| D | ir_emitter_unnested.cc | 3052 const KernelCodeGenerator& kernel_generator, in EmitKernel() argument
3182 kernel_generator.GetTileElementGenerator()( in EmitKernel()
3196 kernel_generator.GetBlockPrologueGenerator(); in EmitKernel()
3204 kernel_generator.GetBlockEpilogueGenerator(); in EmitKernel()
3265 KernelCodeGenerator kernel_generator(std::move(element_generator)); in EmitHlo021Tile() local
3266 return EmitKernel(hlo, tiled_param_ids, kernel_generator, &kernel_info); in EmitHlo021Tile()
3776 KernelCodeGenerator kernel_generator( in EmitReductionToVector() local
3794 EmitKernel(unnested_hlo, {}, kernel_generator, &reduction_info); in EmitReductionToVector()
|
| /external/tensorflow/tensorflow/compiler/xla/service/cpu/ |
| D | ir_emitter.h | 130 const llvm_ir::ElementGenerator& kernel_generator,
|
| D | ir_emitter.cc | 953 const llvm_ir::ElementGenerator& kernel_generator, in EmitElementalConvolution() argument
1084 kernel_generator(kernel_index)); in EmitElementalConvolution()
|