Searched refs:red_axis (Results 1 – 2 of 2) sorted by relevance
68 Eigen::array<int, 1> red_axis; in test_first_dim_reductions() local69 red_axis[0] = 0; in test_first_dim_reductions()70 Tensor<Type, 2, DataLayout> redux = in.sum(red_axis); in test_first_dim_reductions()84 gpu_out.device(dev) = gpu_in.sum(red_axis); in test_first_dim_reductions()85 gpu_out.device(dev) += gpu_in.sum(red_axis); in test_first_dim_reductions()108 Eigen::array<int, 1> red_axis; in test_last_dim_reductions() local109 red_axis[0] = 2; in test_last_dim_reductions()110 Tensor<Type, 2, DataLayout> redux = in.sum(red_axis); in test_last_dim_reductions()124 gpu_out.device(dev) = gpu_in.sum(red_axis); in test_last_dim_reductions()125 gpu_out.device(dev) += gpu_in.sum(red_axis); in test_last_dim_reductions()
62 Eigen::array<int, 1> red_axis; in test_first_dim_reductions_sycl() local63 red_axis[0] = 0; in test_first_dim_reductions_sycl()72 redux= in.sum(red_axis); in test_first_dim_reductions_sycl()81 out_gpu.device(sycl_device) = in_gpu.sum(red_axis); in test_first_dim_reductions_sycl()100 Eigen::array<int, 1> red_axis; in test_last_dim_reductions_sycl() local101 red_axis[0] = 2; in test_last_dim_reductions_sycl()110 redux= in.sum(red_axis); in test_last_dim_reductions_sycl()119 out_gpu.device(sycl_device) = in_gpu.sum(red_axis); in test_last_dim_reductions_sycl()