Searched refs:target_element_generator (Results 1 – 6 of 6) sorted by relevance
| /external/tensorflow/tensorflow/compiler/xla/service/llvm_ir/ |
| D | loop_emitter.cc | 46 LoopEmitter::LoopEmitter(const ElementGenerator& target_element_generator, in LoopEmitter() argument
51 target_element_generator(array_index)); in LoopEmitter()
59 const ElementGenerator& target_element_generator, in MakeBodyEmitterForMultiOutput() argument
63 target_element_generator(array_index)); in MakeBodyEmitterForMultiOutput()
78 LoopEmitter::LoopEmitter(const ElementGenerator& target_element_generator, in LoopEmitter() argument
82 target_element_generator, in LoopEmitter()
|
| D | loop_emitter.h | 54 LoopEmitter(const ElementGenerator& target_element_generator,
62 LoopEmitter(const ElementGenerator& target_element_generator,
|
| /external/tensorflow/tensorflow/compiler/xla/service/gpu/ |
| D | parallel_loop_emitter.cc | 43 const llvm_ir::ElementGenerator& target_element_generator, in ParallelLoopEmitter() argument
47 : LoopEmitter(target_element_generator, target_arrays, b), in ParallelLoopEmitter()
52 const llvm_ir::ElementGenerator& target_element_generator, in ParallelLoopEmitter() argument
56 : LoopEmitter(target_element_generator, target_array, b), in ParallelLoopEmitter()
|
| D | parallel_loop_emitter.h | 40 ParallelLoopEmitter(const llvm_ir::ElementGenerator& target_element_generator,
50 ParallelLoopEmitter(const llvm_ir::ElementGenerator& target_element_generator,
|
| /external/tensorflow/tensorflow/compiler/xla/service/cpu/ |
| D | parallel_loop_emitter.cc | 26 const llvm_ir::ElementGenerator& target_element_generator, in ParallelLoopEmitter() argument
29 : LoopEmitter(target_element_generator, target_array, b), in ParallelLoopEmitter()
|
| D | parallel_loop_emitter.h | 54 ParallelLoopEmitter(const llvm_ir::ElementGenerator& target_element_generator,
|