• Home
  • Raw
  • Download

Lines Matching full:pack

160         ITensorPack pack;  in run()  local
163 pack.add_tensor(TensorType::ACL_SRC, src); in run()
164 pack.add_tensor(TensorType::ACL_DST, src_perm); in run()
165 _permute_input->run(pack); in run()
175 ITensorPack pack; in run() local
176 pack.add_tensor(TensorType::ACL_SRC_0, src_perm); in run()
177 pack.add_tensor(TensorType::ACL_SRC_1, weights_perm); in run()
178 pack.add_tensor(TensorType::ACL_SRC_2, bias); in run()
179 pack.add_tensor(TensorType::ACL_INT_0, workspace); in run()
180 pack.add_tensor(TensorType::ACL_INT_1, packed_weights); in run()
181 pack.add_tensor(TensorType::ACL_DST, dst_perm); in run()
182 _dwc_optimized_func->run(pack); in run()
190 ITensorPack pack; in run() local
191 pack.add_tensor(TensorType::ACL_SRC_0, src); in run()
192 pack.add_tensor(TensorType::ACL_SRC_1, weights); in run()
193 pack.add_tensor(TensorType::ACL_SRC_2, bias); in run()
194 pack.add_tensor(TensorType::ACL_INT_0, workspace); in run()
195 pack.add_tensor(TensorType::ACL_INT_1, packed_weights); in run()
196 pack.add_tensor(TensorType::ACL_DST, dst); in run()
197 _dwc_optimized_func->run(pack); in run()
203 ITensorPack pack; in run() local
205 pack.add_tensor(TensorType::ACL_SRC, dst_perm); in run()
206 pack.add_tensor(TensorType::ACL_DST, dst); in run()
207 _permute_output->run(pack); in run()
213 ITensorPack pack; in run() local
214 pack.add_tensor(TensorType::ACL_SRC, dst); in run()
215 pack.add_tensor(TensorType::ACL_DST, dst); in run()
216 _activationlayer_function->run(pack); in run()
252 ITensorPack pack; in prepare() local
253 pack.add_tensor(TensorType::ACL_SRC, weights); in prepare()
254 pack.add_tensor(TensorType::ACL_DST, permuted_weights); in prepare()
255 _permute_weights->run(pack); in prepare()
385 ITensorPack pack; in run() local
386 pack.add_tensor(TensorType::ACL_SRC, src); in run()
387 pack.add_tensor(TensorType::ACL_DST, src_perm); in run()
388 _permute_input->run(pack); in run()
409 ITensorPack pack; in run() local
411 pack.add_tensor(TensorType::ACL_SRC, dst_perm); in run()
412 pack.add_tensor(TensorType::ACL_DST, dst); in run()
413 _permute_output->run(pack); in run()
418 ITensorPack pack; in run() local
419 pack.add_tensor(TensorType::ACL_SRC, dst); in run()
420 pack.add_tensor(TensorType::ACL_DST, dst); in run()
421 _activationlayer_function->run(pack); in run()
434 ITensorPack pack; in prepare() local
435 pack.add_tensor(TensorType::ACL_SRC, weights); in prepare()
436 pack.add_tensor(TensorType::ACL_DST, weights_perm); in prepare()
438 _permute_weights->run(pack); in prepare()