Searched refs:GetIrArray (Results 1 – 6 of 6) sorted by relevance
| /external/tensorflow/tensorflow/compiler/xla/service/gpu/ |
| D | ir_emitter.cc | 72 return GetIrArray(*operand, *hlo).EmitReadArrayElement(index, &b_); in DefaultAction()
147 llvm_ir::EmitTuple(GetIrArray(*tuple, *tuple), base_ptrs, &b_); in HandleTuple()
434 llvm_ir::EmitTupleSelect(GetIrArray(*tuple_select, *tuple_select), in HandleTupleSelect()
435 GetIrArray(*pred, *tuple_select), in HandleTupleSelect()
470 const llvm_ir::IrArray& target_array = GetIrArray(*dot, *dot); in HandleDot()
471 const llvm_ir::IrArray& lhs_array = GetIrArray(*lhs_instruction, *dot); in HandleDot()
472 const llvm_ir::IrArray& rhs_array = GetIrArray(*rhs_instruction, *dot); in HandleDot()
713 GetIrArray(*arg, *reduce).EmitArrayElementAddress(input_index, &b_); in HandleReduce()
806 output_arrays.push_back(GetIrArray(hlo, hlo, {i})); in ConstructIrArrayForOutputs()
809 output_arrays.push_back(GetIrArray(hlo, hlo)); in ConstructIrArrayForOutputs()
|
| D | ir_emitter.h | 122 llvm_ir::IrArray GetIrArray(const HloInstruction& inst,
125 return bindings_.GetIrArray(inst, consumer, shape_index);
194 return GetIrArray(*operand, *fusion); in GetGeneratorForOperandIrArrays()
|
| D | ir_emitter_nested.cc | 126 llvm_ir::EmitTuple(GetIrArray(hlo, hlo), target_arrays, &b_); in EmitTargetElementLoop()
129 return llvm_ir::LoopEmitter(element_generator, GetIrArray(hlo, hlo), &b_) in EmitTargetElementLoop()
|
| D | ir_emitter_unnested.cc | 662 IrArray output_array = GetIrArray(*fusion, *fusion); in HandleFusion()
712 GetIrArray(*unnested_hlo, *unnested_hlo, extra_output_gens[i].second) in EmitExtraOutputsForReduce()
899 IrArray operand_array = GetIrArray(*operand, *select_and_scatter); in HandleSelectAndScatter()
950 GetIrArray(*source, *select_and_scatter) in HandleSelectAndScatter()
956 GetIrArray(*select_and_scatter, *select_and_scatter) in HandleSelectAndScatter()
1003 return GetIrArray(*operand, *rng).EmitReadArrayElement(index, &b_); in HandleRng()
1054 return GetIrArray(*scatter_indices, *scatter) in HandleScatter()
1059 return GetIrArray(*updates, *scatter) in HandleScatter()
1180 GetIrArray(*output_hlo, *output_hlo) in EmitScatter()
1331 values_arrays.push_back(GetIrArray(*sort, *sort, shape_index)); in HandleSort()
[all …]
|
| D | hlo_to_ir_bindings.h | 87 llvm_ir::IrArray GetIrArray(const HloInstruction& hlo,
|
| D | hlo_to_ir_bindings.cc | 213 llvm_ir::IrArray HloToIrBindings::GetIrArray(const HloInstruction& hlo, in GetIrArray() function in xla::gpu::HloToIrBindings
|