Home
last modified time | relevance | path

Searched refs:ForLoop (Results 1 – 12 of 12) sorted by relevance

/external/tensorflow/tensorflow/compiler/xla/service/llvm_ir/
Dllvm_loop.cc33 ForLoop::ForLoop(absl::string_view prefix, absl::string_view suffix, in ForLoop() function in xla::llvm_ir::ForLoop
46 /* static */ std::unique_ptr<ForLoop> ForLoop::EmitForLoop( in EmitForLoop()
50 std::unique_ptr<ForLoop> loop(new ForLoop(prefix, /*suffix=*/"", start_index, in EmitForLoop()
57 void ForLoop::Emit(llvm::IRBuilder<>* b) { in Emit()
142 std::vector<llvm::Metadata*> ForLoop::GetLoopMetadata(llvm::IRBuilder<>* b) { in GetLoopMetadata()
167 string ForLoop::GetQualifiedName(absl::string_view name) { in GetQualifiedName()
171 llvm::BasicBlock* ForLoop::CreateLoopBB(absl::string_view name, in CreateLoopBB()
176 std::unique_ptr<ForLoop> ForLoopNest::AddLoop(absl::string_view suffix, in AddLoop()
185 std::unique_ptr<ForLoop> ForLoopNest::AddLoop( in AddLoop()
192 std::unique_ptr<ForLoop> loop(new ForLoop( in AddLoop()
[all …]
Dllvm_loop.h44 class ForLoop {
46 ForLoop(const ForLoop&) = delete;
47 ForLoop& operator=(const ForLoop&) = delete;
83 static std::unique_ptr<ForLoop> EmitForLoop(
139 ForLoop(absl::string_view prefix, absl::string_view suffix,
196 std::unique_ptr<ForLoop> AddLoop(
203 std::unique_ptr<ForLoop> AddLoop(
211 std::unique_ptr<ForLoop> AddLoop(
218 std::unique_ptr<ForLoop> AddLoop(
Dkernel_support_library.cc45 std::unique_ptr<llvm_ir::ForLoop> loop = llvm_ir::ForLoop::EmitForLoop( in ForWithStatus()
Dloop_emitter.cc104 std::unique_ptr<ForLoop> loop = loop_nest.AddLoop( in EmitIndexAndSetExitBasicBlock()
/external/tensorflow/tensorflow/compiler/xla/service/cpu/
Dparallel_loop_emitter.cc54 std::unique_ptr<llvm_ir::ForLoop> loop = loop_nest.AddLoop( in EmitIndexAndSetExitBasicBlock()
60 std::unique_ptr<llvm_ir::ForLoop> loop = loop_nest.AddLoop( in EmitIndexAndSetExitBasicBlock()
Ddot_op_emitter.cc447 std::unique_ptr<llvm_ir::ForLoop> reduction_loop = loop_nest.AddLoop( in EmitNaiveLlvmIrGemm()
Dir_emitter.cc1730 std::unique_ptr<llvm_ir::ForLoop> loop = loop_nest.AddLoop( in EmitVectorizedReduce()
1750 std::unique_ptr<llvm_ir::ForLoop> loop = in EmitVectorizedReduce()
/external/adhd/cras/src/tests/
Darray_unittest.cc86 TEST(ArrayTest, ForLoop) { in TEST() argument
/external/swiftshader/third_party/SPIRV-Tools/test/opt/
Dlocal_ssa_elim_test.cpp28 TEST_F(LocalSSAElimTest, ForLoop) { in TEST_F() argument
/external/deqp-deps/SPIRV-Tools/test/opt/
Dlocal_ssa_elim_test.cpp28 TEST_F(LocalSSAElimTest, ForLoop) { in TEST_F() argument
/external/tensorflow/tensorflow/compiler/xla/service/gpu/
Dir_emitter.cc546 std::unique_ptr<llvm_ir::ForLoop> reduction_loop = loop_nest.AddLoop( in HandleDot()
/external/tensorflow/tensorflow/compiler/xla/service/
Delemental_ir_emitter.cc2132 std::unique_ptr<llvm_ir::ForLoop> inner_loop = llvm_ir::ForLoop::EmitForLoop( in EmitElementalDot()