Searched refs:reduction_axis2 (Results 1 – 2 of 2) sorted by relevance
60 array<ptrdiff_t, 2> reduction_axis2; in test_simple_reductions() local61 reduction_axis2[0] = 1; in test_simple_reductions()62 reduction_axis2[1] = 3; in test_simple_reductions()64 Tensor<float, 2, DataLayout> result = tensor.sum(reduction_axis2); in test_simple_reductions()94 reduction_axis2[0] = 0; in test_simple_reductions()95 reduction_axis2[1] = 2; in test_simple_reductions()96 result = tensor.prod(reduction_axis2); in test_simple_reductions()126 reduction_axis2[0] = 0; in test_simple_reductions()127 reduction_axis2[1] = 2; in test_simple_reductions()128 result = tensor.maximum(reduction_axis2); in test_simple_reductions()[all …]
112 const Dims2 reduction_axis2; in test_type2index_list() local113 Tensor<float, 2> result2 = tensor.sum(reduction_axis2); in test_type2index_list()