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.cc177 std::unique_ptr<ForLoop> ForLoopNest::AddLoop(absl::string_view suffix, in AddLoop() function in xla::llvm_ir::ForLoopNest
182 return AddLoop(suffix, start_index, end_index, GetConstantWithIndexType(1), in AddLoop()
186 std::unique_ptr<ForLoop> ForLoopNest::AddLoop( in AddLoop() function in xla::llvm_ir::ForLoopNest
211 std::unique_ptr<ForLoop> ForLoopNest::AddLoop(int64 start_index, in AddLoop() function in xla::llvm_ir::ForLoopNest
217 return AddLoop(suffix, GetConstantWithIndexType(start_index), in AddLoop()
222 std::unique_ptr<ForLoop> ForLoopNest::AddLoop(int64 start_index, in AddLoop() function in xla::llvm_ir::ForLoopNest
228 return AddLoop(suffix, GetConstantWithIndexType(start_index), in AddLoop()
247 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 EmitStaticIndex()
123 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.cc590 std::unique_ptr<llvm_ir::ForLoop> reduction_loop = loop_nest.AddLoop( in EmitNaiveLlvmIrGemm()
Dir_emitter.cc1710 std::unique_ptr<llvm_ir::ForLoop> loop = loop_nest.AddLoop( in EmitVectorizedReduce()
1731 loop_nest.AddLoop(start_index, end_index, vectorization_factor, in EmitVectorizedReduce()
/external/deqp-deps/SPIRV-Tools/source/opt/
Dloop_descriptor.h504 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/swiftshader/third_party/SPIRV-Tools/source/opt/
Dloop_descriptor.h504 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/tensorflow/tensorflow/compiler/xla/service/
Delemental_ir_emitter.cc2856 .AddLoop( in EmitConvolution()
2863 .AddLoop(0, lhs->shape().dimensions(dnums.input_feature_dimension()), in EmitConvolution()