Lines Matching refs:add_tensor
163 pack.add_tensor(TensorType::ACL_SRC, src); in run()
164 pack.add_tensor(TensorType::ACL_DST, src_perm); in run()
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()
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()
205 pack.add_tensor(TensorType::ACL_SRC, dst_perm); in run()
206 pack.add_tensor(TensorType::ACL_DST, dst); in run()
214 pack.add_tensor(TensorType::ACL_SRC, dst); in run()
215 pack.add_tensor(TensorType::ACL_DST, dst); in run()
231 pack_opt.add_tensor(TensorType::ACL_SRC_1, weights); in prepare()
232 pack_opt.add_tensor(TensorType::ACL_SRC_2, bias); in prepare()
233 pack_opt.add_tensor(TensorType::ACL_INT_1, packed_weights); in prepare()
253 pack.add_tensor(TensorType::ACL_SRC, weights); in prepare()
254 pack.add_tensor(TensorType::ACL_DST, permuted_weights); in prepare()
261 pack_opt.add_tensor(TensorType::ACL_SRC_2, bias); in prepare()
262 pack_opt.add_tensor(TensorType::ACL_INT_1, packed_weights); in prepare()
270 pack_opt.add_tensor(TensorType::ACL_SRC_1, weights); in prepare()
271 pack_opt.add_tensor(TensorType::ACL_SRC_2, bias); in prepare()
272 pack_opt.add_tensor(TensorType::ACL_INT_1, packed_weights); in prepare()
386 pack.add_tensor(TensorType::ACL_SRC, src); in run()
387 pack.add_tensor(TensorType::ACL_DST, src_perm); in run()
393 pack_depth.add_tensor(TensorType::ACL_SRC_2, biases); in run()
394 pack_depth.add_tensor(TensorType::ACL_DST, dst_perm); in run()
400 pack_depth.add_tensor(TensorType::ACL_SRC_0, src); in run()
401 pack_depth.add_tensor(TensorType::ACL_SRC_1, weights); in run()
402 pack_depth.add_tensor(TensorType::ACL_SRC_2, biases); in run()
403 pack_depth.add_tensor(TensorType::ACL_DST, dst); in run()
411 pack.add_tensor(TensorType::ACL_SRC, dst_perm); in run()
412 pack.add_tensor(TensorType::ACL_DST, dst); in run()
419 pack.add_tensor(TensorType::ACL_SRC, dst); in run()
420 pack.add_tensor(TensorType::ACL_DST, dst); in run()
435 pack.add_tensor(TensorType::ACL_SRC, weights); in prepare()
436 pack.add_tensor(TensorType::ACL_DST, weights_perm); in prepare()