Method VirtualKernelInit

oneflow/core/kernel/slice_boxing_kernel.cpp:64–76

};

void SliceBoxingKernel::VirtualKernelInit(KernelContext* ctx) {
  const SliceBoxingConf& conf = GetCustomizedBoxingConf();
  if (/*is_0size_tensor=*/std::any_of(conf.out_shape().dim().begin(), conf.out_shape().dim().end(),
                                      [](int64_t dim) { return dim == 0; })) {
    return;
  }
  const TensorSliceView out_slice(conf.out_slice());
  for (const TensorSliceViewProto& in_slice_proto : conf.in_slice()) {
    const TensorSliceView in_slice(in_slice_proto);
    tensor_slice_copier_vec_.emplace_back(new TensorSliceCopier(
        out_slice, in_slice, this->kernel_conf().data_type(), ctx->stream()->device_type()));
  }
}

const std::vector<std::shared_ptr<TensorSliceCopier>>& SliceBoxingKernel::tensor_slice_copier_vec()
    const {