MCPcopy Create free account
hub / github.com/Oneflow-Inc/oneflow / ForwardDataContent

Method ForwardDataContent

oneflow/core/kernel/slice_boxing_kernel.cpp:87–94  ·  view source on GitHub ↗

Source from the content-addressed store, hash-verified

85}
86
87void SliceBoxingCopyKernel::ForwardDataContent(KernelContext* ctx) const {
88 Blob* out = ctx->BnInOp2Blob("out");
89 if (out->shape_view().elem_cnt() == 0) { return; }
90 FOR_RANGE(int64_t, i, 0, this->op_attribute().input_bns().size()) {
91 const Blob* in_i = ctx->BnInOp2Blob(GenRepeatedBn("in", i));
92 this->tensor_slice_copier_vec().at(i)->Copy(ctx->stream(), out, in_i);
93 }
94}
95
96const SliceBoxingConf& SliceBoxingAddKernel::GetCustomizedBoxingConf() const {
97 return this->op_conf().slice_boxing_add_conf().slice_boxing_conf();

Callers

nothing calls this directly

Calls 6

BnInOp2BlobMethod · 0.45
elem_cntMethod · 0.45
shape_viewMethod · 0.45
device_typeMethod · 0.45
streamMethod · 0.45
data_typeMethod · 0.45

Tested by

no test coverage detected