Home
last modified time | relevance | path

Searched refs:reduction_axis (Results 1 – 8 of 8) 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/
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()
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()
Dfused_batch_norm_op.h88 Eigen::array<int, 1> reduction_axis{0}; in operator()
93 Eigen::IndexList<Eigen::type2index<0> > reduction_axis; in operator() local
106 offset_backprop.device(d) = y_backprop_rest_by_depth.sum(reduction_axis); in operator()
116 .sum(reduction_axis); in operator()
/external/tensorflow/tensorflow/contrib/layers/python/layers/
Dnormalization.py105 reduction_axis = 1
112 reduction_axis = inputs_rank - 1
115 del moments_axes[reduction_axis]
117 params_shape = inputs_shape[reduction_axis:reduction_axis + 1]
/external/tensorflow/tensorflow/contrib/gan/python/features/python/
Dvirtual_batchnorm_test.py78 for reduction_axis in range(1, 4):
81 del reduction_axes[reduction_axis]
86 del vb_reduction_axes[reduction_axis]
88 vbn = virtual_batchnorm.VBN(partial_batch, reduction_axis)