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 | 38 LoopEmitter::LoopEmitter(const ElementGenerator& target_element_generator, in LoopEmitter() argument
43 target_element_generator(array_index)); in LoopEmitter()
51 const ElementGenerator& target_element_generator, in MakeBodyEmitterForMultiOutputFusion() argument
55 target_element_generator(array_index)); in MakeBodyEmitterForMultiOutputFusion()
70 LoopEmitter::LoopEmitter(const ElementGenerator& target_element_generator, in LoopEmitter() argument
74 target_element_generator, in LoopEmitter()
|
| D | loop_emitter.h | 47 LoopEmitter(const ElementGenerator& target_element_generator,
55 LoopEmitter(const ElementGenerator& target_element_generator,
|
| /external/tensorflow/tensorflow/compiler/xla/service/gpu/ |
| D | parallel_loop_emitter.cc | 41 const llvm_ir::ElementGenerator& target_element_generator, in ParallelLoopEmitter() argument
45 : LoopEmitter(target_element_generator, target_arrays, b), in ParallelLoopEmitter()
50 const llvm_ir::ElementGenerator& target_element_generator, in ParallelLoopEmitter() argument
54 : 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,
|