Searched refs:ElementGenerator (Results 1 – 17 of 17) sorted by relevance
/external/tensorflow/tensorflow/compiler/xla/service/llvm_ir/ |
D | loop_emitter.h | 36 using ElementGenerator = variable 54 LoopEmitter(const ElementGenerator& target_element_generator, 62 LoopEmitter(const ElementGenerator& target_element_generator,
|
D | fused_ir_emitter.h | 55 using IndexedGenerator = llvm_ir::ElementGenerator; 63 llvm_ir::ElementGenerator generator) { in BindGenerator()
|
D | dynamic_update_slice_util.cc | 97 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()
|
D | loop_emitter.cc | 46 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/ |
D | parallel_loop_emitter.h | 40 ParallelLoopEmitter(const llvm_ir::ElementGenerator& target_element_generator, 50 ParallelLoopEmitter(const llvm_ir::ElementGenerator& target_element_generator,
|
D | ir_emitter_unnested.h | 228 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;
|
D | parallel_loop_emitter.cc | 43 const llvm_ir::ElementGenerator& target_element_generator, in ParallelLoopEmitter() 52 const llvm_ir::ElementGenerator& target_element_generator, in ParallelLoopEmitter()
|
D | ir_emitter_nested.h | 58 const llvm_ir::ElementGenerator& body_emitter) override;
|
D | ir_emitter.h | 147 const llvm_ir::ElementGenerator& body_emitter) = 0;
|
D | ir_emitter_nested.cc | 168 const llvm_ir::ElementGenerator& element_generator) { in EmitTargetElementLoop()
|
D | ir_emitter_unnested.cc | 2120 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/ |
D | elemental_ir_emitter.h | 39 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,
|
D | elemental_ir_emitter.cc | 1916 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/ |
D | parallel_loop_emitter.h | 54 ParallelLoopEmitter(const llvm_ir::ElementGenerator& target_element_generator,
|
D | parallel_loop_emitter.cc | 26 const llvm_ir::ElementGenerator& target_element_generator, in ParallelLoopEmitter()
|
D | ir_emitter.h | 313 const llvm_ir::ElementGenerator& element_generator); 316 const llvm_ir::ElementGenerator& element_generator);
|
D | ir_emitter.cc | 3125 const llvm_ir::ElementGenerator& element_generator) { in EmitTargetElementLoop() 3131 const llvm_ir::ElementGenerator& element_generator) { in EmitTargetElementLoop()
|