Searched refs:reduction_axis4 (Results 1 – 2 of 2) sorted by relevance
83 array<ptrdiff_t, 4> reduction_axis4; in test_simple_reductions() local84 reduction_axis4[0] = 0; in test_simple_reductions()85 reduction_axis4[1] = 1; in test_simple_reductions()86 reduction_axis4[2] = 2; in test_simple_reductions()87 reduction_axis4[3] = 3; in test_simple_reductions()88 Tensor<float, 0, DataLayout> sum2 = tensor.sum(reduction_axis4); in test_simple_reductions()115 array<ptrdiff_t, 4> reduction_axis4; in test_simple_reductions() local116 reduction_axis4[0] = 0; in test_simple_reductions()117 reduction_axis4[1] = 1; in test_simple_reductions()118 reduction_axis4[2] = 2; in test_simple_reductions()[all …]
144 const Dims4 reduction_axis4; in test_type2index_list() local145 Tensor<float, 0> result4 = tensor.sum(reduction_axis4); in test_type2index_list()