Method DataFromBHWC

tensorflow/lite/delegates/gpu/cl/tensor.cc:332–358

template <typename T>
void Tensor::DataFromBHWC(absl::Span<const float> src,
                          absl::Span<T> dst) const {
  int channels =
      storage_type_ == TensorStorageType::SINGLE_TEXTURE_2D ? channels_ : 4;
  BHWC src_shape;
  src_shape.b = 1;
  src_shape.h = height_;
  src_shape.w = width_;
  src_shape.c = channels_;
  for (int d = 0; d < Depth(); ++d) {
    for (int y = 0; y < height_; ++y) {
      for (int x = 0; x < width_; ++x) {
        for (int c = 0; c < channels; ++c) {
          float value;
          if (d * 4 + c < channels_) {
            const int cpu_index = src_shape.LinearIndex({0, y, x, d * 4 + c});
            value = src[cpu_index];
          } else {
            value = 0.0f;
          }
          const int gpu_index = GetLinearIndex(x, y, d, c);
          dst[gpu_index] = value;
        }
      }
    }
  }
}

template void Tensor::DataFromBHWC<float>(absl::Span<const float> src,
                                          absl::Span<float> dst) const;