| 21 | |
| 22 | namespace BlstmComputerTest { |
| 23 | shared_ptr<Tensor> createTensorFromData(const vector<int> &shape, float *data, Tensor::DimensionType dimType, |
| 24 | CPUBackend *bn) { |
| 25 | auto srcTensor = shared_ptr<Tensor>(Tensor::create(shape, halide_type_of<float>(), data, Tensor::CAFFE)); |
| 26 | auto tgtTensor = shared_ptr<Tensor>(Tensor::createDevice<float>(shape, dimType)); |
| 27 | bn->onAcquireBuffer(tgtTensor.get(), Backend::DYNAMIC); |
| 28 | bn->onCopyBuffer(srcTensor.get(), tgtTensor.get()); |
| 29 | return tgtTensor; |
| 30 | } |
| 31 | |
| 32 | shared_ptr<Tensor> createAndRun(int timesteps, int batch, int inDim, int stateDim, bool bidirectional, float *input, |
| 33 | float *weights, float *initH, float *initC, vector<int> lengths, |
no test coverage detected