Searched refs:EmitLoop (Results 1 – 11 of 11) sorted by relevance
| /external/tensorflow/tensorflow/compiler/xla/service/gpu/ |
| D | parallel_loop_emitter.cc | 158 Status ParallelLoopEmitter::EmitLoop(absl::string_view loop_name, in EmitLoop() function in xla::gpu::ParallelLoopEmitter
169 return LoopEmitter::EmitLoop(loop_name, index_type); in EmitLoop()
|
| D | ir_emitter_nested.cc | 175 llvm_ir::LoopEmitter(element_generator, target_arrays, &b_).EmitLoop()); in EmitTargetElementLoop()
180 .EmitLoop(); in EmitTargetElementLoop()
|
| D | parallel_loop_emitter.h | 63 Status EmitLoop(absl::string_view loop_name = "",
|
| D | ir_emitter_unnested.cc | 931 .EmitLoop(ir_name, in EmitPadToStaticFromMlir()
1055 .EmitLoop(ir_name, GetIndexTypeForKernelFromMlir( in EmitSliceToDynamicFromMlir()
1861 .EmitLoop(context.name, index_type)); in EmitLoopFusionFromMlir()
1866 .EmitLoop(context.name, index_type)); in EmitLoopFusionFromMlir()
1926 .EmitLoop( in HandleFusion()
2428 .EmitLoop(name, index_type); in EmitSelectAndScatterFromMlir()
2729 .EmitLoop(desc.name, in EmitScatter()
3684 .EmitLoop(mlir::GetNameFromLoc(op->getLoc()))); in BuildInitializerThunkForMlir()
3750 .EmitLoop(mlir::GetNameFromLoc(fusion.getLoc()))); in BuildFusedInitializerThunkForMlir()
5820 .EmitLoop(IrName(mlir::GetNameFromLoc(fusion.getLoc())), in EmitInputFusibleNonStridedSlices()
|
| /external/tensorflow/tensorflow/compiler/xla/service/llvm_ir/ |
| D | dynamic_update_slice_util.cc | 155 .EmitLoop(name); in EmitDynamicUpdateSliceInPlaceImpl()
157 return LoopEmitter(loop_body_emitter, update_shape, b).EmitLoop(name); in EmitDynamicUpdateSliceInPlaceImpl()
|
| D | loop_emitter.h | 83 Status EmitLoop(absl::string_view loop_name = "",
|
| D | loop_emitter.cc | 166 Status LoopEmitter::EmitLoop(absl::string_view loop_name, in EmitLoop() function in xla::llvm_ir::LoopEmitter
|
| D | sort_util.cc | 393 .EmitLoop(name); in EmitSortInPlace()
|
| /external/swiftshader/src/Pipeline/ |
| D | SpirvShaderControlFlow.cpp | 276 EmitLoop(state); in EmitBlocks()
325 void SpirvShader::EmitLoop(EmitState *state) const in EmitLoop() function in sw::SpirvShader
|
| D | SpirvShader.hpp | 1165 void EmitLoop(EmitState *state) const;
|
| /external/tensorflow/tensorflow/compiler/xla/service/cpu/ |
| D | ir_emitter.cc | 2185 .EmitLoop(IrName(hlo)); in HandleSliceToDynamic()
2249 .EmitLoop(IrName(hlo))); in HandlePadToStatic()
3155 .EmitLoop(IrName(target_op))); in EmitTargetElementLoop()
3171 .EmitLoop(IrName(target_op))); in EmitTargetElementLoop()
3175 .EmitLoop(IrName(target_op))); in EmitTargetElementLoop()
|