MCPcopy Create free account
hub / github.com/DeepRec-AI/DeepRec / EmitTuple

Function EmitTuple

tensorflow/compiler/xla/service/llvm_ir/tuple_ops.cc:61–71  ·  view source on GitHub ↗

Source from the content-addressed store, hash-verified

59}
60
61void EmitTuple(const IrArray& tuple, absl::Span<llvm::Value* const> operands,
62 llvm::IRBuilder<>* b) {
63 llvm::Module* module = getModuleFromBuilder(b);
64 for (size_t i = 0; i < operands.size(); ++i) {
65 auto* store = b->CreateStore(
66 b->CreatePointerCast(operands[i], PrimitiveTypeToIrType(TUPLE, module)),
67 b->CreateInBoundsGEP(tuple.GetBasePointer(),
68 {b->getInt64(0), b->getInt64(i)}));
69 tuple.AnnotateLoadStoreInstructionWithMetadata(store);
70 }
71}
72
73void EmitTuple(const IrArray& tuple, absl::Span<const IrArray> buffers,
74 llvm::IRBuilder<>* b) {

Callers 12

EmitTargetElementLoopMethod · 0.85
HandleTupleMethod · 0.85
HandleReduceMethod · 0.85
EmitHlo021TileMethod · 0.85
HandleInfeedMethod · 0.85
HandleSortMethod · 0.85
HandleTupleMethod · 0.85
HandleCustomCallMethod · 0.85
EmitTargetElementLoopMethod · 0.85
EmitThreadLocalCallMethod · 0.85

Calls 6

getModuleFromBuilderFunction · 0.85
PrimitiveTypeToIrTypeFunction · 0.85
sizeMethod · 0.45
GetBasePointerMethod · 0.45
reserveMethod · 0.45

Tested by

no test coverage detected