Searched refs:loop_body_emitter (Results 1 – 3 of 3) sorted by relevance
| /external/tensorflow/tensorflow/compiler/xla/service/llvm_ir/ |
| D | dynamic_update_slice_util.cc | 130 auto loop_body_emitter = [&](const IrArray::Index& update_index) -> Status { in EmitDynamicUpdateSliceInPlaceImpl() local
153 return gpu::ParallelLoopEmitter(loop_body_emitter, update_shape, in EmitDynamicUpdateSliceInPlaceImpl()
157 return LoopEmitter(loop_body_emitter, update_shape, b).EmitLoop(name); in EmitDynamicUpdateSliceInPlaceImpl()
|
| /external/tensorflow/tensorflow/compiler/xla/service/cpu/ |
| D | ir_emitter.cc | 2172 auto loop_body_emitter = in HandleSliceToDynamic() local
2183 return llvm_ir::LoopEmitter(loop_body_emitter, data_array.GetShape(), in HandleSliceToDynamic()
2238 auto loop_body_emitter = in HandlePadToStatic() local
2248 llvm_ir::LoopEmitter(loop_body_emitter, input_shape, dynamic_dims, &b_) in HandlePadToStatic()
|
| /external/tensorflow/tensorflow/compiler/xla/service/gpu/ |
| D | ir_emitter_unnested.cc | 2265 auto loop_body_emitter = [&](const IrArray::Index& source_index) -> Status { in EmitSelectAndScatterFromMlir() local
2426 return ParallelLoopEmitter(loop_body_emitter, source_shape, launch_dimensions, in EmitSelectAndScatterFromMlir()
2599 auto loop_body_emitter = [&](const IrArray::Index& index) -> Status { in EmitScatter() local
2727 return ParallelLoopEmitter(loop_body_emitter, desc.updates_shape, in EmitScatter()
|