Searched refs:AddLoopsForShapeOnDimensions (Results 1 – 4 of 4) sorted by relevance
| /external/tensorflow/tensorflow/compiler/xla/service/llvm_ir/ |
| D | llvm_loop.cc | 238 return IrArray::Index(AddLoopsForShapeOnDimensions(shape, dimensions, suffix), in AddLoopsForShape()
242 std::vector<llvm::Value*> ForLoopNest::AddLoopsForShapeOnDimensions( in AddLoopsForShapeOnDimensions() function in xla::llvm_ir::ForLoopNest
276 AddLoopsForShapeOnDimensions(shape, dimensions, name_suffix); in EmitOperandArrayLoopNest()
|
| D | llvm_loop.h | 242 std::vector<llvm::Value*> AddLoopsForShapeOnDimensions(
|
| /external/tensorflow/tensorflow/compiler/xla/service/cpu/ |
| D | ir_emitter.cc | 1561 reduction_loop_nest.AddLoopsForShapeOnDimensions(arg->shape(), dimensions, in EmitInnerLoopForVectorizedReduction()
1923 loops.AddLoopsForShapeOnDimensions(slice->shape(), outer_dims, "slice"); in HandleSlice()
2482 loops.AddLoopsForShapeOnDimensions(output_shape, outer_dims, "concat"); in EmitFastConcatenate()
|
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | elemental_ir_emitter.cc | 2752 loops.AddLoopsForShapeOnDimensions(arg->shape(), reduced_dimensions, in EmitElementalReduce()
|