Searched refs:GetIrArray (Results 1 – 5 of 5) sorted by relevance
| /external/tensorflow/tensorflow/compiler/xla/service/gpu/ |
| D | ir_emitter.h | 125 llvm_ir::IrArray GetIrArray(const HloInstruction& inst,
128 return bindings_.GetIrArray(inst, consumer, shape_index);
|
| D | ir_emitter.cc | 94 return GetIrArray(*operand, *hlo) in DefaultAction()
220 llvm_ir::EmitTuple(GetIrArray(*tuple, *tuple), base_ptrs, &b_); in HandleTuple()
677 output_arrays.push_back(GetIrArray(hlo, hlo, {i})); in ConstructIrArrayForOutputs()
680 output_arrays.push_back(GetIrArray(hlo, hlo)); in ConstructIrArrayForOutputs()
692 return GetIrArray(*operand, *fusion) in BindFusionArguments()
|
| D | ir_emitter_nested.cc | 176 llvm_ir::EmitTuple(GetIrArray(hlo, hlo), target_arrays, &b_); in EmitTargetElementLoop()
179 return llvm_ir::LoopEmitter(element_generator, GetIrArray(hlo, hlo), &b_) in EmitTargetElementLoop()
|
| D | hlo_to_ir_bindings.h | 85 llvm_ir::IrArray GetIrArray(const HloInstruction& hlo,
|
| D | hlo_to_ir_bindings.cc | 191 llvm_ir::IrArray HloToIrBindings::GetIrArray(const HloInstruction& hlo, in GetIrArray() function in xla::gpu::HloToIrBindings
|