| 187 | } |
| 188 | |
| 189 | Status Tensor::ReadData(CLCommandQueue* queue, TensorFloat32* dst) const { |
| 190 | RETURN_IF_ERROR(IsValid(dst->shape)); |
| 191 | return ReadDataBHWC(absl::MakeSpan(dst->data), queue); |
| 192 | } |
| 193 | |
| 194 | Status CreateTensor(const CLContext& context, const CLDevice& device, int width, |
| 195 | int height, int channels, DataType data_type, |