Home
last modified time | relevance | path

Searched refs:ElementGenerator (Results 1 – 17 of 17) sorted by relevance

/external/tensorflow/tensorflow/compiler/xla/service/llvm_ir/
Dloop_emitter.h36 using ElementGenerator = variable
54 LoopEmitter(const ElementGenerator& target_element_generator,
62 LoopEmitter(const ElementGenerator& target_element_generator,
Dfused_ir_emitter.h55 using IndexedGenerator = llvm_ir::ElementGenerator;
63 llvm_ir::ElementGenerator generator) { in BindGenerator()
Ddynamic_update_slice_util.cc97 bool is_signed, ElementGenerator update_array_generator, in EmitDynamicUpdateSliceInPlaceImpl()
177 ElementGenerator update_array_generator = [&](const IrArray::Index& index) { in EmitDynamicUpdateSliceInPlace()
220 TF_ASSIGN_OR_RETURN(ElementGenerator update_array_generator, in EmitFusedDynamicUpdateSliceInPlaceImpl()
226 ElementGenerator element_generator, in EmitFusedDynamicUpdateSliceInPlaceImpl()
Dloop_emitter.cc46 LoopEmitter::LoopEmitter(const ElementGenerator& target_element_generator, in LoopEmitter()
59 const ElementGenerator& target_element_generator, in MakeBodyEmitterForMultiOutput()
78 LoopEmitter::LoopEmitter(const ElementGenerator& target_element_generator, in LoopEmitter()
/external/tensorflow/tensorflow/compiler/xla/service/gpu/
Dparallel_loop_emitter.h40 ParallelLoopEmitter(const llvm_ir::ElementGenerator& target_element_generator,
50 ParallelLoopEmitter(const llvm_ir::ElementGenerator& target_element_generator,
Dir_emitter_unnested.h228 const llvm_ir::ElementGenerator& body_emitter) override;
234 const HloInstruction& hlo, const llvm_ir::ElementGenerator& body_emitter,
368 absl::Span<const std::pair<llvm_ir::ElementGenerator, int>>
470 const llvm_ir::ElementGenerator& scatter_indices_gen,
471 const llvm_ir::ElementGenerator& updates_gen,
486 llvm_ir::ElementGenerator scatter_indices_gen;
487 llvm_ir::ElementGenerator updates_gen;
Dparallel_loop_emitter.cc43 const llvm_ir::ElementGenerator& target_element_generator, in ParallelLoopEmitter()
52 const llvm_ir::ElementGenerator& target_element_generator, in ParallelLoopEmitter()
Dir_emitter_nested.h58 const llvm_ir::ElementGenerator& body_emitter) override;
Dir_emitter.h147 const llvm_ir::ElementGenerator& body_emitter) = 0;
Dir_emitter_nested.cc168 const llvm_ir::ElementGenerator& element_generator) { in EmitTargetElementLoop()
Dir_emitter_unnested.cc2120 absl::Span<const std::pair<llvm_ir::ElementGenerator, int>> in EmitExtraOutputsForReduce() argument
2567 const llvm_ir::ElementGenerator& scatter_indices_gen, in EmitScatter()
2568 const llvm_ir::ElementGenerator& updates_gen, in EmitScatter()
3935 const HloInstruction& hlo, const llvm_ir::ElementGenerator& body_emitter) { in EmitTargetElementLoop()
4157 llvm_ir::ElementGenerator gen; in EmitTileElementForFusion()
4184 llvm_ir::ElementGenerator output_generator = in EmitTileElementForFusion()
4606 InlinedVector<llvm_ir::ElementGenerator, 1> input_gens; in EmitTileElementForReduction()
4607 std::vector<std::pair<llvm_ir::ElementGenerator, int>> extra_output_gens; in EmitTileElementForReduction()
/external/tensorflow/tensorflow/compiler/xla/service/
Delemental_ir_emitter.h39 std::unordered_map<const HloInstruction*, llvm_ir::ElementGenerator>;
48 llvm_ir::ElementGenerator MakeElementGenerator(
253 std::vector<llvm_ir::ElementGenerator> input_generators,
254 std::vector<llvm_ir::ElementGenerator> initial_value_generators,
259 std::vector<llvm_ir::ElementGenerator> input_generators,
260 std::vector<llvm_ir::ElementGenerator> initial_value_generators,
Delemental_ir_emitter.cc1916 const llvm_ir::ElementGenerator& operand_generator = in EmitElementalGather()
1918 const llvm_ir::ElementGenerator& indices_generator = in EmitElementalGather()
2244 llvm_ir::ElementGenerator ElementalIrEmitter::MakeElementGenerator( in MakeElementGenerator()
2494 std::vector<llvm_ir::ElementGenerator> input_generators; in MakeElementGenerator()
2500 std::vector<llvm_ir::ElementGenerator> initial_value_generators; in MakeElementGenerator()
2511 std::vector<llvm_ir::ElementGenerator> input_generators; in MakeElementGenerator()
2516 std::vector<llvm_ir::ElementGenerator> initial_value_generators; in MakeElementGenerator()
2591 std::vector<llvm_ir::ElementGenerator> input_generators, in EmitElementalReduceWindow()
2592 std::vector<llvm_ir::ElementGenerator> initial_value_generators, in EmitElementalReduceWindow()
2710 std::vector<llvm_ir::ElementGenerator> input_generators, in EmitElementalReduce()
[all …]
/external/tensorflow/tensorflow/compiler/xla/service/cpu/
Dparallel_loop_emitter.h54 ParallelLoopEmitter(const llvm_ir::ElementGenerator& target_element_generator,
Dparallel_loop_emitter.cc26 const llvm_ir::ElementGenerator& target_element_generator, in ParallelLoopEmitter()
Dir_emitter.h313 const llvm_ir::ElementGenerator& element_generator);
316 const llvm_ir::ElementGenerator& element_generator);
Dir_emitter.cc3125 const llvm_ir::ElementGenerator& element_generator) { in EmitTargetElementLoop()
3131 const llvm_ir::ElementGenerator& element_generator) { in EmitTargetElementLoop()