Searched refs:target_arrays (Results 1 – 5 of 5) sorted by relevance
| /external/tensorflow/tensorflow/compiler/xla/service/llvm_ir/ |
| D | loop_emitter.cc | 60 const std::vector<IrArray>& target_arrays, llvm::IRBuilder<>* b) { in MakeBodyEmitterForMultiOutput() argument
68 target_arrays.size()); in MakeBodyEmitterForMultiOutput()
70 for (int64 i = 0; i < target_arrays.size(); ++i) { in MakeBodyEmitterForMultiOutput()
71 target_arrays[i].EmitWriteArrayElement( in MakeBodyEmitterForMultiOutput()
79 absl::Span<const IrArray> target_arrays, in LoopEmitter() argument
83 std::vector<IrArray>(target_arrays.begin(), target_arrays.end()), b)), in LoopEmitter()
84 shape_(target_arrays[0].GetShape()), in LoopEmitter()
88 for (const IrArray& array : target_arrays) { in LoopEmitter()
|
| D | loop_emitter.h | 63 absl::Span<const IrArray> target_arrays, llvm::IRBuilder<>* b);
|
| /external/tensorflow/tensorflow/compiler/xla/service/gpu/ |
| D | ir_emitter_nested.cc | 172 std::vector<llvm_ir::IrArray> target_arrays = in EmitTargetElementLoop() local
175 llvm_ir::LoopEmitter(element_generator, target_arrays, &b_).EmitLoop()); in EmitTargetElementLoop()
176 llvm_ir::EmitTuple(GetIrArray(hlo, hlo), target_arrays, &b_); in EmitTargetElementLoop()
|
| D | parallel_loop_emitter.cc | 44 absl::Span<const llvm_ir::IrArray> target_arrays, in ParallelLoopEmitter() argument
47 : LoopEmitter(target_element_generator, target_arrays, b), in ParallelLoopEmitter()
|
| D | parallel_loop_emitter.h | 51 absl::Span<const llvm_ir::IrArray> target_arrays,
|