Searched refs:ir_value (Results 1 – 2 of 2) sorted by relevance
| /external/tensorflow/tensorflow/compiler/xla/service/gpu/ |
| D | hlo_to_ir_bindings.cc | 122 llvm::Value* CastToTypedValue(const Shape& shape, llvm::Value* ir_value, in CastToTypedValue() argument
130 if (llvm::isa<llvm::GlobalVariable>(ir_value)) { in CastToTypedValue()
132 llvm::cast<llvm::GlobalVariable>(ir_value), dest_type); in CastToTypedValue()
135 ir_value, pointee_type->getPointerTo()); in CastToTypedValue()
142 llvm::Value* ir_value) { in GetTypedIrValue() argument
144 ShapeUtil::GetSubshape(hlo.shape(), shape_index), ir_value, b_); in GetTypedIrValue()
145 if (!HasMeaningfulName(ir_value)) { in GetTypedIrValue()
146 ir_value->setName(llvm_ir::IrName(&hlo, "raw")); in GetTypedIrValue()
155 llvm::Value* ir_value, in BindHloToIrValue() argument
160 llvm::Value* typed_ir_value = GetTypedIrValue(hlo, shape_index, ir_value); in BindHloToIrValue()
|
| D | hlo_to_ir_bindings.h | 51 void BindHloToIrValue(const HloInstruction& hlo, llvm::Value* ir_value,
99 llvm::Value* ir_value);
120 llvm::Value* CastToTypedValue(const Shape& shape, llvm::Value* ir_value,
|