Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/compiler/xla/service/llvm_ir/
Dllvm_loop.cc176 std::unique_ptr<ForLoop> ForLoopNest::AddLoop(absl::string_view suffix, in AddLoop() function in xla::llvm_ir::ForLoopNest
181 return AddLoop(suffix, start_index, end_index, GetConstantWithIndexType(1), in AddLoop()
185 std::unique_ptr<ForLoop> ForLoopNest::AddLoop( in AddLoop() function in xla::llvm_ir::ForLoopNest
210 std::unique_ptr<ForLoop> ForLoopNest::AddLoop(int64 start_index, in AddLoop() function in xla::llvm_ir::ForLoopNest
216 return AddLoop(suffix, GetConstantWithIndexType(start_index), in AddLoop()
221 std::unique_ptr<ForLoop> ForLoopNest::AddLoop(int64 start_index, in AddLoop() function in xla::llvm_ir::ForLoopNest
227 return AddLoop(suffix, GetConstantWithIndexType(start_index), in AddLoop()
246 std::unique_ptr<llvm_ir::ForLoop> loop = AddLoop( in AddLoopsForShapeOnDimensions()
Dllvm_loop.h196 std::unique_ptr<ForLoop> AddLoop(
203 std::unique_ptr<ForLoop> AddLoop(
211 std::unique_ptr<ForLoop> AddLoop(
218 std::unique_ptr<ForLoop> AddLoop(
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()
Dir_emitter.cc985 .AddLoop( in EmitElementalConvolution()
992 .AddLoop(0, lhs->shape().dimensions(dnums.input_feature_dimension()), in EmitElementalConvolution()
1730 std::unique_ptr<llvm_ir::ForLoop> loop = loop_nest.AddLoop( in EmitVectorizedReduce()
1751 loop_nest.AddLoop(start_index, end_index, vectorization_factor, in EmitVectorizedReduce()
Ddot_op_emitter.cc447 std::unique_ptr<llvm_ir::ForLoop> reduction_loop = loop_nest.AddLoop( in EmitNaiveLlvmIrGemm()
/external/deqp-deps/SPIRV-Tools/source/opt/
Dloop_descriptor.h502 inline void AddLoop(Loop* loop_to_add, Loop* parent) { in AddLoop() function
Dloop_unroller.cpp510 loop_descriptor.AddLoop(new_loop, loop->GetParent()); in PartiallyUnrollResidualFactor()
/external/swiftshader/third_party/SPIRV-Tools/source/opt/
Dloop_descriptor.h502 inline void AddLoop(std::unique_ptr<Loop>&& loop_to_add, Loop* parent) { in AddLoop() function
Dloop_unroller.cpp510 loop_descriptor.AddLoop(std::move(new_loop), loop->GetParent()); in PartiallyUnrollResidualFactor()
/external/tensorflow/tensorflow/compiler/xla/service/gpu/
Dir_emitter.cc546 std::unique_ptr<llvm_ir::ForLoop> reduction_loop = loop_nest.AddLoop( in HandleDot()