Home
last modified time | relevance | path

Searched refs:reduction_axis (Results 1 – 7 of 7) sorted by relevance

/external/eigen/unsupported/test/
Dcxx11_tensor_reduction.cpp22 array<ptrdiff_t, 0> reduction_axis; in test_trivial_reductions() local
24 Tensor<float, 0, DataLayout> result = tensor.sum(reduction_axis); in test_trivial_reductions()
31 array<ptrdiff_t, 0> reduction_axis; in test_trivial_reductions() local
33 Tensor<float, 1, DataLayout> result = tensor.sum(reduction_axis); in test_trivial_reductions()
43 array<ptrdiff_t, 0> reduction_axis; in test_trivial_reductions() local
45 Tensor<float, 2, DataLayout> result = tensor.sum(reduction_axis); in test_trivial_reductions()
273 array<ptrdiff_t, 2> reduction_axis; in test_full_reductions() local
274 reduction_axis[0] = 0; in test_full_reductions()
275 reduction_axis[1] = 1; in test_full_reductions()
277 Tensor<float, 0, DataLayout> result = tensor.sum(reduction_axis); in test_full_reductions()
[all …]
Dcxx11_tensor_index_list.cpp21 constexpr auto reduction_axis = make_index_list(0, 1, 2); in test_static_index_list() local
22 VERIFY_IS_EQUAL(internal::array_get<0>(reduction_axis), 0); in test_static_index_list()
23 VERIFY_IS_EQUAL(internal::array_get<1>(reduction_axis), 1); in test_static_index_list()
24 VERIFY_IS_EQUAL(internal::array_get<2>(reduction_axis), 2); in test_static_index_list()
25 VERIFY_IS_EQUAL(static_cast<DenseIndex>(reduction_axis[0]), 0); in test_static_index_list()
26 VERIFY_IS_EQUAL(static_cast<DenseIndex>(reduction_axis[1]), 1); in test_static_index_list()
27 VERIFY_IS_EQUAL(static_cast<DenseIndex>(reduction_axis[2]), 2); in test_static_index_list()
29 …EIGEN_STATIC_ASSERT((internal::array_get<0>(reduction_axis) == 0), YOU_MADE_A_PROGRAMMING_MISTAKE); in test_static_index_list()
30 …EIGEN_STATIC_ASSERT((internal::array_get<1>(reduction_axis) == 1), YOU_MADE_A_PROGRAMMING_MISTAKE); in test_static_index_list()
31 …EIGEN_STATIC_ASSERT((internal::array_get<2>(reduction_axis) == 2), YOU_MADE_A_PROGRAMMING_MISTAKE); in test_static_index_list()
[all …]
Dcxx11_tensor_cuda.cu230 array<Eigen::DenseIndex, 2> reduction_axis; in test_cuda_reduction() local
231 reduction_axis[0] = 1; in test_cuda_reduction()
232 reduction_axis[1] = 3; in test_cuda_reduction()
234 gpu_out.device(gpu_device) = gpu_in1.maximum(reduction_axis); in test_cuda_reduction()
/external/tensorflow/tensorflow/core/kernels/image/
Dadjust_contrast_op.h44 Eigen::array<int, 2> reduction_axis; in operator() local
45 reduction_axis[0] = 1; in operator()
46 reduction_axis[1] = 2; in operator()
59 reduction_axis; in operator() local
72 (input.template cast<float>().sum(reduction_axis).eval() / in operator()
106 Eigen::array<int, 2> reduction_axis; in operator() local
107 reduction_axis[0] = 0; in operator()
108 reduction_axis[1] = 1; in operator()
126 reduction_axis; in operator() local
143 .sum(reduction_axis) in operator()
/external/tensorflow/tensorflow/core/kernels/
Dbatch_norm_op.h94 Eigen::array<Index, 1> reduction_axis; in operator() local
95 reduction_axis[0] = 0; // Reduces on first dimension. in operator()
101 Eigen::IndexList<Eigen::type2index<0> > reduction_axis; in operator() local
114 db.device(d) = out_backprop.reshape(rest_by_depth).sum(reduction_axis); in operator()
123 .sum(reduction_axis); in operator()
Dredux_functor.h269 Eigen::array<Eigen::Index, 1> reduction_axis = {0}; in operator() local
275 reduction_axis, reducer, binary_op]( in operator()
290 .reduce(reduction_axis, reducer) in operator()
298 .reduce(reduction_axis, reducer) in operator()
306 .reduce(reduction_axis, reducer) in operator()
Dfused_batch_norm_op.cu.cc70 Eigen::array<int, 1> reduction_axis{0}; in operator ()() local
75 Eigen::IndexList<Eigen::type2index<0> > reduction_axis; in operator ()() local
88 offset_backprop.device(d) = y_backprop_rest_by_depth.sum(reduction_axis); in operator ()()
98 .sum(reduction_axis); in operator ()()