Searched refs:target_arrays (Results 1 – 5 of 5) sorted by relevance
| /external/tensorflow/tensorflow/compiler/xla/service/llvm_ir/ |
| D | loop_emitter.cc | 52 const std::vector<IrArray>& target_arrays, llvm::IRBuilder<>* b) { in MakeBodyEmitterForMultiOutputFusion() argument
60 target_arrays.size()); in MakeBodyEmitterForMultiOutputFusion()
62 for (int64 i = 0; i < target_arrays.size(); ++i) { in MakeBodyEmitterForMultiOutputFusion()
63 target_arrays[i].EmitWriteArrayElement( in MakeBodyEmitterForMultiOutputFusion()
71 absl::Span<const IrArray> target_arrays, in LoopEmitter() argument
75 std::vector<IrArray>(target_arrays.begin(), target_arrays.end()), b)), in LoopEmitter()
76 shape_(target_arrays[0].GetShape()), in LoopEmitter()
80 for (const IrArray& array : target_arrays) { in LoopEmitter()
|
| D | loop_emitter.h | 56 absl::Span<const IrArray> target_arrays, llvm::IRBuilder<>* b);
|
| /external/tensorflow/tensorflow/compiler/xla/service/gpu/ |
| D | ir_emitter_nested.cc | 122 std::vector<llvm_ir::IrArray> target_arrays = in EmitTargetElementLoop() local
125 llvm_ir::LoopEmitter(element_generator, target_arrays, &b_).EmitLoop()); in EmitTargetElementLoop()
126 llvm_ir::EmitTuple(GetIrArray(hlo, hlo), target_arrays, &b_); in EmitTargetElementLoop()
|
| D | parallel_loop_emitter.cc | 42 absl::Span<const llvm_ir::IrArray> target_arrays, in ParallelLoopEmitter() argument
45 : LoopEmitter(target_element_generator, target_arrays, b), in ParallelLoopEmitter()
|
| D | parallel_loop_emitter.h | 51 absl::Span<const llvm_ir::IrArray> target_arrays,
|