Lines Matching refs:tensor2
261 inline void CPU_tensor_apply2(Tensor tensor1, Tensor tensor2, const Op op) { in CPU_tensor_apply2() argument
262 if (!_apply_preamble({tensor1, tensor2})) in CPU_tensor_apply2()
264 if (_max_dim_tensors({tensor1, tensor2}) <= 8) { in CPU_tensor_apply2()
270 strided_tensor_iter_fixed<scalar2, 8>(tensor2)); in CPU_tensor_apply2()
277 strided_tensor_iter<scalar2>(tensor2)); in CPU_tensor_apply2()
284 Tensor tensor2, in CPU_tensor_apply3() argument
287 if (!_apply_preamble({tensor1, tensor2, tensor3})) in CPU_tensor_apply3()
289 if (_max_dim_tensors({tensor1, tensor2, tensor3}) <= 8) { in CPU_tensor_apply3()
295 strided_tensor_iter_fixed<scalar2, 8>(tensor2), in CPU_tensor_apply3()
303 strided_tensor_iter<scalar2>(tensor2), in CPU_tensor_apply3()
316 Tensor tensor2, in CPU_tensor_apply4() argument
320 if (!_apply_preamble({tensor1, tensor2, tensor3, tensor4})) in CPU_tensor_apply4()
322 if (_max_dim_tensors({tensor1, tensor2, tensor3, tensor4}) <= 8) { in CPU_tensor_apply4()
328 strided_tensor_iter_fixed<scalar2, 8>(tensor2), in CPU_tensor_apply4()
337 strided_tensor_iter<scalar2>(tensor2), in CPU_tensor_apply4()