Class TensorBHWC

tensorflow/lite/delegates/gpu/cl/tensor.h:106–148

};

class TensorBHWC : public Tensor {
 public:
  TensorBHWC() = default;
  TensorBHWC(cl_mem memory, int width, int height, int channels,
             enum DataType data_type, TensorStorageType storage_type)
      : Tensor(memory, width, height, channels, data_type, storage_type) {}

  // Move only
  TensorBHWC(TensorBHWC&& tensor);
  TensorBHWC& operator=(TensorBHWC&& tensor);
  TensorBHWC(const TensorBHWC&) = delete;
  TensorBHWC& operator=(const TensorBHWC&) = delete;

  Status WriteData(CLCommandQueue* queue, void* data_ptr) const {
    const size_t data_size =
        Width() * Height() * Channels() * SizeOf(DataType());
    RETURN_IF_ERROR(
        queue->EnqueueWriteBuffer(GetMemoryPtr(), data_size, data_ptr));
    return OkStatus();
  }

  Status ReadData(CLCommandQueue* queue, void* data_ptr) const {
    const size_t data_size =
        Width() * Height() * Channels() * SizeOf(DataType());
    RETURN_IF_ERROR(
        queue->EnqueueReadBuffer(GetMemoryPtr(), data_size, data_ptr));
    return OkStatus();
  }

  ~TensorBHWC() override { ReleaseBHWC(); }

 private:
  friend Status CreateTensorBHWCFromOpenGlObject(const CLContext& context,
                                                 cl_int ssbo_id,
                                                 const HWC& shape,
                                                 bool is_readonly,
                                                 TensorBHWC* tensor);

  void ReleaseBHWC();

  // When object created from GL object it isn't owner
  bool owner_ = true;
};

using TensorPtr = std::shared_ptr<Tensor>;