Home
last modified time | relevance | path

Searched refs:loop_nest (Results 1 – 7 of 7) sorted by relevance

/external/tensorflow/tensorflow/compiler/xla/service/cpu/
Dparallel_loop_emitter.cc37 llvm_ir::ForLoopNest loop_nest(loop_name, ir_builder_); in EmitIndexAndSetExitBasicBlock() local
51 std::unique_ptr<llvm_ir::ForLoop> loop = loop_nest.AddLoop( in EmitIndexAndSetExitBasicBlock()
57 std::unique_ptr<llvm_ir::ForLoop> loop = loop_nest.AddLoop( in EmitIndexAndSetExitBasicBlock()
65 llvm_ir::SetToFirstInsertPoint(loop_nest.GetInnerLoopBodyBasicBlock(), in EmitIndexAndSetExitBasicBlock()
69 exit_bb_ = loop_nest.GetOuterLoopExitBasicBlock(); in EmitIndexAndSetExitBasicBlock()
Ddot_op_emitter.cc764 llvm_ir::ForLoopNest loop_nest(llvm_ir::IrName(&dot_), ir_builder_); in Emit() local
766 &loop_nest, lhs_array_, lhs_reduction_dimension, "lhs"); in Emit()
768 &loop_nest, rhs_array_, rhs_reduction_dimension, "rhs"); in Emit()
778 std::unique_ptr<llvm_ir::ForLoop> reduction_loop = loop_nest.AddLoop( in Emit()
872 ir_builder_->SetInsertPoint(loop_nest.GetOuterLoopExitBasicBlock()); in Emit()
1013 llvm_ir::ForLoopNest* loop_nest, const llvm_ir::IrArray& operand_array, in EmitOperandArrayLoopNest() argument
1030 loop_nest->AddLoopsForShapeOnDimensions(shape, dimensions, name_suffix); in EmitOperandArrayLoopNest()
Ddot_op_emitter.h99 llvm_ir::ForLoopNest* loop_nest, const llvm_ir::IrArray& operand_array,
Dir_emitter.cc1649 llvm_ir::ForLoopNest loop_nest(IrName(reduce), &ir_builder_); in EmitVectorizedReduce() local
1657 loop_nest.AddLoop(start_index, end_index, in EmitVectorizedReduce()
1667 loop_nest.GetInnerLoopBodyBasicBlock()) { in EmitVectorizedReduce()
1671 auto outermost_loop_exit_block = loop_nest.GetOuterLoopExitBasicBlock(); in EmitVectorizedReduce()
1677 std::unique_ptr<llvm_ir::ForLoop> loop = loop_nest.AddLoop( in EmitVectorizedReduce()
/external/tensorflow/tensorflow/compiler/xla/service/llvm_ir/
Dloop_emitter.cc103 ForLoopNest loop_nest(loop_name, ir_builder_); in EmitIndexAndSetExitBasicBlock() local
107 std::unique_ptr<ForLoop> loop = loop_nest.AddLoop( in EmitIndexAndSetExitBasicBlock()
116 llvm::BasicBlock* innermost_body_bb = loop_nest.GetInnerLoopBodyBasicBlock(); in EmitIndexAndSetExitBasicBlock()
121 exit_bb_ = loop_nest.GetOuterLoopExitBasicBlock(); in EmitIndexAndSetExitBasicBlock()
/external/tensorflow/tensorflow/compiler/xla/service/gpu/
Dir_emitter.cc504 llvm_ir::ForLoopNest loop_nest(IrName(dot), &ir_builder_); in HandleDot() local
506 lhs_array, lhs_reduction_dimension, "lhs", &loop_nest); in HandleDot()
508 rhs_array, rhs_reduction_dimension, "rhs", &loop_nest); in HandleDot()
511 std::unique_ptr<llvm_ir::ForLoop> reduction_loop = loop_nest.AddLoop( in HandleDot()
595 ir_builder_.SetInsertPoint(loop_nest.GetOuterLoopExitBasicBlock()); in HandleDot()
766 tensorflow::StringPiece name_suffix, llvm_ir::ForLoopNest* loop_nest) { in EmitOperandArrayLoopNest() argument
782 loop_nest->AddLoopsForShapeOnDimensions(shape, dimensions, name_suffix); in EmitOperandArrayLoopNest()
Dir_emitter.h181 tensorflow::StringPiece name_suffix, llvm_ir::ForLoopNest* loop_nest);