Searched refs:element_generator (Results 1 – 5 of 5) sorted by relevance
| /external/tensorflow/tensorflow/compiler/xla/service/gpu/ |
| D | ir_emitter_nested.cc | 168 const llvm_ir::ElementGenerator& element_generator) { in EmitTargetElementLoop() argument
175 llvm_ir::LoopEmitter(element_generator, target_arrays, &b_).EmitLoop()); in EmitTargetElementLoop()
179 return llvm_ir::LoopEmitter(element_generator, GetIrArray(hlo, hlo), &b_) in EmitTargetElementLoop()
|
| D | ir_emitter_unnested.cc | 1808 auto element_generator, in EmitLoopFusionFromMlir()
1859 ParallelLoopEmitter(element_generator, output_element_arrays, in EmitLoopFusionFromMlir()
1864 ParallelLoopEmitter(element_generator, output_element_arrays[0], in EmitLoopFusionFromMlir()
4880 EmitElementFunction element_generator = in EmitHlo021Tile() local
4941 tile_height, tile_width, element_generator); in EmitHlo021Tile()
|
| /external/tensorflow/tensorflow/compiler/xla/service/llvm_ir/ |
| D | dynamic_update_slice_util.cc | 226 ElementGenerator element_generator, in EmitFusedDynamicUpdateSliceInPlaceImpl()
228 return element_generator(IrArray::Index(b->getInt64Ty())); in EmitFusedDynamicUpdateSliceInPlaceImpl()
|
| /external/tensorflow/tensorflow/compiler/xla/service/cpu/ |
| D | ir_emitter.h | 313 const llvm_ir::ElementGenerator& element_generator);
316 const llvm_ir::ElementGenerator& element_generator);
|
| D | ir_emitter.cc | 3125 const llvm_ir::ElementGenerator& element_generator) { in EmitTargetElementLoop() argument
3126 return EmitTargetElementLoop(target_op, /*desc=*/"", element_generator); in EmitTargetElementLoop()
3131 const llvm_ir::ElementGenerator& element_generator) { in EmitTargetElementLoop() argument
3154 llvm_ir::LoopEmitter(element_generator, output_arrays, &b_) in EmitTargetElementLoop()
3169 TF_RETURN_IF_ERROR(ParallelLoopEmitter(element_generator, target_array, in EmitTargetElementLoop()
3174 llvm_ir::LoopEmitter(element_generator, target_array, &b_) in EmitTargetElementLoop()
|