Searched refs:loop_body_emitter (Results 1 – 2 of 2) sorted by relevance
76 auto loop_body_emitter = [&](const IrArray::Index& update_index) -> Status { in EmitDynamicUpdateSliceInPlaceImpl() local99 return gpu::ParallelLoopEmitter(loop_body_emitter, update_shape, in EmitDynamicUpdateSliceInPlaceImpl()103 return LoopEmitter(loop_body_emitter, update_shape, b).EmitLoop(name); in EmitDynamicUpdateSliceInPlaceImpl()
834 auto loop_body_emitter = [=](const IrArray::Index& source_index) -> Status { in HandleSelectAndScatter() local972 return ParallelLoopEmitter(loop_body_emitter, source->shape(), in HandleSelectAndScatter()1085 auto loop_body_emitter = [&](const IrArray::Index& index) -> Status { in EmitScatter() local1198 return ParallelLoopEmitter(loop_body_emitter, updates->shape(), in EmitScatter()