Searched refs:AddLoopsForShapeOnDimensions (Results 1 – 5 of 5) sorted by relevance
| /external/tensorflow/tensorflow/compiler/xla/service/llvm_ir/ |
| D | llvm_loop.cc | 237 return IrArray::Index(AddLoopsForShapeOnDimensions(shape, dimensions, suffix), in AddLoopsForShape()
241 std::vector<llvm::Value*> ForLoopNest::AddLoopsForShapeOnDimensions( in AddLoopsForShapeOnDimensions() function in xla::llvm_ir::ForLoopNest
273 AddLoopsForShapeOnDimensions(shape, dimensions, name_suffix); in EmitOperandArrayLoopNest()
|
| D | llvm_loop.h | 242 std::vector<llvm::Value*> AddLoopsForShapeOnDimensions(
|
| /external/tensorflow/tensorflow/compiler/xla/service/gpu/ |
| D | elemental_ir_emitter.cc | 457 loops.AddLoopsForShapeOnDimensions( in MakeElementGenerator()
|
| D | ir_emitter.cc | 690 loops.AddLoopsForShapeOnDimensions(arg->shape(), dimensions, in HandleReduce()
|
| /external/tensorflow/tensorflow/compiler/xla/service/cpu/ |
| D | ir_emitter.cc | 1596 reduction_loop_nest.AddLoopsForShapeOnDimensions(arg->shape(), dimensions, in EmitInnerLoopForVectorizedReduction()
1839 loops.AddLoopsForShapeOnDimensions(arg->shape(), dimensions, in EmitElementalReduce()
2009 loops.AddLoopsForShapeOnDimensions(slice->shape(), outer_dims, "slice"); in HandleSlice()
2421 loops.AddLoopsForShapeOnDimensions(output_shape, outer_dims, "concat"); in EmitFastConcatenate()
|