Searched defs:reduction_axis (Results 1 – 7 of 7) sorted by relevance
| /external/eigen/unsupported/test/ |
| D | cxx11_tensor_reduction.cpp | 22 array<ptrdiff_t, 0> reduction_axis; in test_trivial_reductions() local
31 array<ptrdiff_t, 0> reduction_axis; in test_trivial_reductions() local
43 array<ptrdiff_t, 0> reduction_axis; in test_trivial_reductions() local
273 array<ptrdiff_t, 2> reduction_axis; in test_full_reductions() local
315 array<ptrdiff_t, 1> reduction_axis; in test_user_defined_reductions() local
341 array<ptrdiff_t, 2> reduction_axis; in test_tensor_maps() local
372 array<int, 2> reduction_axis; in test_static_dims() local
376 Eigen::IndexList<Eigen::type2index<1>, Eigen::type2index<3> > reduction_axis; in test_static_dims() local
402 array<int, 2> reduction_axis; in test_innermost_last_dims() local
407 Eigen::IndexList<Eigen::type2index<0>, Eigen::type2index<1> > reduction_axis; in test_innermost_last_dims() local
[all …]
|
| D | cxx11_tensor_index_list.cpp | 21 constexpr auto reduction_axis = make_index_list(0, 1, 2); in test_static_index_list() local
276 auto reduction_axis = make_index_list(dim1, dim2, dim3); in test_dynamic_index_list() local
307 auto reduction_axis = make_index_list(0, dim2, 2, dim4); in test_mixed_index_list() local
|
| D | cxx11_tensor_cuda.cu | 230 array<Eigen::DenseIndex, 2> reduction_axis; in test_cuda_reduction() local
|
| /external/tensorflow/tensorflow/core/kernels/image/ |
| D | adjust_contrast_op.h | 44 Eigen::array<int, 2> reduction_axis; in operator() local
59 reduction_axis; in operator() local
106 Eigen::array<int, 2> reduction_axis; in operator() local
126 reduction_axis; in operator() local
|
| /external/tensorflow/tensorflow/core/kernels/ |
| D | batch_norm_op.h | 94 Eigen::array<Index, 1> reduction_axis; in operator() local
101 Eigen::IndexList<Eigen::type2index<0> > reduction_axis; in operator() local
|
| D | fused_batch_norm_op.cu.cc | 70 Eigen::array<int, 1> reduction_axis{0}; in operator ()() local
75 Eigen::IndexList<Eigen::type2index<0> > reduction_axis; in operator ()() local
|
| D | redux_functor.h | 269 Eigen::array<Eigen::Index, 1> reduction_axis = {0}; in operator() local
|