Home
last modified time | relevance | path

Searched refs:AddLoop (Results 1 – 13 of 13) sorted by relevance

/external/tensorflow/tensorflow/compiler/xla/service/llvm_ir/
Dllvm_loop.cc178 std::unique_ptr<ForLoop> ForLoopNest::AddLoop(absl::string_view suffix, in AddLoop() function in xla::llvm_ir::ForLoopNest
183 return AddLoop(suffix, start_index, end_index, GetConstantWithIndexType(1), in AddLoop()
187 std::unique_ptr<ForLoop> ForLoopNest::AddLoop( in AddLoop() function in xla::llvm_ir::ForLoopNest
212 std::unique_ptr<ForLoop> ForLoopNest::AddLoop(int64_t start_index, in AddLoop() function in xla::llvm_ir::ForLoopNest
218 return AddLoop(suffix, GetConstantWithIndexType(start_index), in AddLoop()
223 std::unique_ptr<ForLoop> ForLoopNest::AddLoop(int64_t start_index, in AddLoop() function in xla::llvm_ir::ForLoopNest
229 return AddLoop(suffix, GetConstantWithIndexType(start_index), in AddLoop()
248 std::unique_ptr<llvm_ir::ForLoop> loop = AddLoop( in AddLoopsForShapeOnDimensions()
Dllvm_loop.h194 std::unique_ptr<ForLoop> AddLoop(
201 std::unique_ptr<ForLoop> AddLoop(
209 std::unique_ptr<ForLoop> AddLoop(
216 std::unique_ptr<ForLoop> AddLoop(
Dloop_emitter.cc111 std::unique_ptr<ForLoop> loop = loop_nest->AddLoop( in EmitStaticIndex()
130 std::unique_ptr<ForLoop> loop = loop_nest->AddLoop( in EmitDynamicIndex()
/external/tensorflow/tensorflow/compiler/xla/service/cpu/
Dparallel_loop_emitter.cc60 std::unique_ptr<llvm_ir::ForLoop> loop = loop_nest.AddLoop( in EmitIndexAndSetExitBasicBlock()
66 std::unique_ptr<llvm_ir::ForLoop> loop = loop_nest.AddLoop( in EmitIndexAndSetExitBasicBlock()
Ddot_op_emitter.cc654 std::unique_ptr<llvm_ir::ForLoop> reduction_loop = loop_nest.AddLoop( in EmitNaiveLlvmIrGemm()
Dir_emitter.cc1812 std::unique_ptr<llvm_ir::ForLoop> loop = loop_nest.AddLoop( in EmitVectorizedReduce()
1833 loop_nest.AddLoop(start_index, end_index, vectorization_factor, in EmitVectorizedReduce()
/external/angle/third_party/vulkan-deps/spirv-tools/src/source/opt/
Dloop_descriptor.h505 inline void AddLoop(std::unique_ptr<Loop>&& loop_to_add, Loop* parent) { in AddLoop() function
Dloop_unroller.cpp512 loop_descriptor.AddLoop(std::move(new_loop), loop->GetParent()); in PartiallyUnrollResidualFactor()
/external/swiftshader/third_party/SPIRV-Tools/source/opt/
Dloop_descriptor.h505 inline void AddLoop(std::unique_ptr<Loop>&& loop_to_add, Loop* parent) { in AddLoop() function
Dloop_unroller.cpp513 loop_descriptor.AddLoop(std::move(new_loop), loop->GetParent()); in PartiallyUnrollResidualFactor()
/external/deqp-deps/SPIRV-Tools/source/opt/
Dloop_descriptor.h505 inline void AddLoop(std::unique_ptr<Loop>&& loop_to_add, Loop* parent) { in AddLoop() function
Dloop_unroller.cpp514 loop_descriptor.AddLoop(std::move(new_loop), loop->GetParent()); in PartiallyUnrollResidualFactor()
/external/tensorflow/tensorflow/compiler/xla/service/
Delemental_ir_emitter.cc3060 .AddLoop( in EmitConvolution()
3072 loops.AddLoop(0, input_group_size, "iz")->GetIndVarValue(); in EmitConvolution()