Searched refs:AddLoopsForShapeOnDimensions (Results 1 – 4 of 4) sorted by relevance
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/llvm_ir/ |
| H A D | llvm_loop.cc | 239 return IrArray::Index(AddLoopsForShapeOnDimensions(shape, dimensions, suffix), in AddLoopsForShape()
243 std::vector<llvm::Value*> ForLoopNest::AddLoopsForShapeOnDimensions( in AddLoopsForShapeOnDimensions() function in xla::llvm_ir::ForLoopNest
277 AddLoopsForShapeOnDimensions(shape, dimensions, name_suffix); in EmitOperandArrayLoopNest()
|
| H A D | llvm_loop.h | 240 std::vector<llvm::Value*> AddLoopsForShapeOnDimensions(
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/cpu/ |
| H A D | ir_emitter.cc | 1659 reduction_loop_nest.AddLoopsForShapeOnDimensions(arg->shape(), dimensions, in EmitInnerLoopForVectorizedReduction()
2034 loops.AddLoopsForShapeOnDimensions(slice->shape(), outer_dims, "slice"); in HandleSlice()
2643 loops.AddLoopsForShapeOnDimensions(output_shape, outer_dims, "concat"); in EmitFastConcatenate()
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/ |
| H A D | elemental_ir_emitter.cc | 2953 loops.AddLoopsForShapeOnDimensions(arg->shape(), reduced_dimensions, in EmitElementalReduce()
|