Home
last modified time | relevance | path

Searched refs:type2index (Results 1 – 25 of 29) sorted by relevance

12

/external/tensorflow/tensorflow/core/kernels/
Dadjust_contrast_op.h58 Eigen::IndexList<Eigen::type2index<1>, Eigen::type2index<2> > in operator()
60 Eigen::IndexList<Eigen::type2index<1>, int, int, Eigen::type2index<1> > in operator()
64 Eigen::IndexList<int, Eigen::type2index<1>, Eigen::type2index<1>, int> in operator()
125 Eigen::IndexList<Eigen::type2index<0>, Eigen::type2index<1> > in operator()
127 Eigen::IndexList<Eigen::type2index<1>, int, int, Eigen::type2index<1> > in operator()
131 Eigen::IndexList<int, Eigen::type2index<1>, Eigen::type2index<1>, int> in operator()
135 Eigen::IndexList<Eigen::type2index<1>, Eigen::type2index<2>, in operator()
136 Eigen::type2index<0>, Eigen::type2index<3> > in operator()
Deigen_pooling.h59 const Eigen::IndexList<Eigen::type2index<1>, Eigen::type2index<2> >,
60 const Eigen::IndexList<Eigen::type2index<2>,
61 Eigen::type2index<3> > >::type,
125 const Eigen::IndexList<Eigen::type2index<1>, Eigen::type2index<2> >,
126 const Eigen::IndexList<Eigen::type2index<2>,
127 Eigen::type2index<3> > >::type reduction_dims;
174 const Eigen::IndexList<Eigen::type2index<1> >,
247 Eigen::IndexList<Eigen::type2index<1> > reduction_dims; in CuboidMaxPooling()
416 const Eigen::IndexList<Eigen::type2index<1>, Eigen::type2index<2> >,
417 const Eigen::IndexList<Eigen::type2index<2>,
[all …]
Dbatch_norm_op.h44 Eigen::IndexList<int, Eigen::type2index<1> > rest_by_one; in operator()
46 Eigen::IndexList<Eigen::type2index<1>, int> one_by_depth; in operator()
48 Eigen::IndexList<int, Eigen::type2index<1> > depth_by_one; in operator()
97 Eigen::IndexList<Index, Eigen::type2index<1> > rest_by_one; in operator()
99 Eigen::IndexList<Eigen::type2index<1>, Index> one_by_depth; in operator()
101 Eigen::IndexList<Eigen::type2index<0> > reduction_axis; in operator()
Dsoftmax_op_functor.h59 Eigen::IndexList<Eigen::type2index<kClassDim> > along_class; in Compute()
60 Eigen::IndexList<int, Eigen::type2index<1> > batch_by_one; in Compute()
62 Eigen::IndexList<Eigen::type2index<1>, int> one_by_class; in Compute()
Deigen_softmax.h60 Eigen::IndexList<Eigen::type2index<0> > depth_dim; in eval()
61 Eigen::IndexList<int, Eigen::type2index<1> > bcast; in eval()
63 Eigen::IndexList<Eigen::type2index<1>, in eval()
Deigen_backward_spatial_convolutions.h48 typedef IndexList<type2index<0>, type2index<0>, type2index<1>, type2index<1> >
50 typedef IndexList<type2index<1>, type2index<1>, type2index<0>, type2index<0> >
Dcwise_op_gpu_select.cu.cc46 Eigen::IndexList<Eigen::type2index<1> > rank1; in operator ()()
71 Eigen::IndexList<Eigen::type2index<1>, int> broadcast_dims; in operator ()()
73 Eigen::IndexList<int, Eigen::type2index<1> > reshape_dims; in operator ()()
Dxent_op.h86 Eigen::IndexList<Eigen::type2index<kClassDim> > along_class; in Compute()
87 Eigen::IndexList<int, Eigen::type2index<1> > batch_by_one; in Compute()
91 Eigen::IndexList<Eigen::type2index<1>, int> one_by_class; in Compute()
Dmultinomial_op_gpu.cu.cc76 Eigen::IndexList<int, Eigen::type2index<1>, int> boc; in operator ()()
80 Eigen::IndexList<Eigen::type2index<1>, int, Eigen::type2index<1>> oso; in operator ()()
Dfused_batch_norm_op.h91 Eigen::IndexList<Eigen::type2index<1>, Eigen::Index> one_by_depth; in operator()
93 Eigen::IndexList<Eigen::type2index<0> > reduction_axis; in operator()
94 Eigen::IndexList<Eigen::Index, Eigen::type2index<1> > rest_by_one; in operator()
Dsparse_xent_op.h145 Eigen::IndexList<Eigen::type2index<1> > along_row; in Compute()
203 Eigen::IndexList<Eigen::type2index<kClassDim> > along_class; in Compute()
204 Eigen::IndexList<int, Eigen::type2index<1> > batch_by_one; in Compute()
208 Eigen::IndexList<Eigen::type2index<1>, int> one_by_class; in Compute()
Dreduction_ops_common.h68 const Eigen::IndexList<Eigen::type2index<0>> kZero;
69 const Eigen::IndexList<Eigen::type2index<1>> kOne;
70 const Eigen::IndexList<Eigen::type2index<0>, Eigen::type2index<2>> kZeroTwo;
Dquantized_instance_norm.cc304 const Eigen::IndexList<Eigen::type2index<1>, Eigen::type2index<2>> in Compute()
306 Eigen::IndexList<Eigen::type2index<1>, Index, Index, Eigen::type2index<1>> in Compute()
310 Eigen::IndexList<Index, Eigen::type2index<1>, Eigen::type2index<1>, Index> in Compute()
Dfused_batch_norm_op.cc102 Eigen::IndexList<Eigen::type2index<1>, Eigen::Index> one_by_depth; in operator ()()
104 Eigen::IndexList<Eigen::type2index<0> > reduce_dims; in operator ()()
105 Eigen::IndexList<Eigen::Index, Eigen::type2index<1> > bcast_spec; in operator ()()
194 Eigen::IndexList<Eigen::type2index<1>, Eigen::Index> one_by_depth; in operator ()()
196 Eigen::IndexList<Eigen::type2index<0> > reduce_dims; in operator ()()
197 Eigen::IndexList<Eigen::Index, Eigen::type2index<1> > bcast_spec; in operator ()()
Dtopk_op.cc114 typename Eigen::IndexList<Eigen::type2index<1>> reduce_on_cols; in Compute()
115 typename Eigen::IndexList<int, Eigen::type2index<1>> rows_by_one; in Compute()
Dcwise_op_select.cc272 Eigen::IndexList<Eigen::type2index<1>, Eigen::DenseIndex> broadcast_dims; in operator ()()
274 Eigen::IndexList<Eigen::DenseIndex, Eigen::type2index<1> > reshape_dims; in operator ()()
Dcolorspace_op.h44 Eigen::IndexList<Eigen::type2index<1> > channel_axis; in operator()
Dsparse_matmul_op.cc71 inline Eigen::IndexList<Eigen::type2index<0>, Eigen::type2index<0>>
73 return Eigen::IndexList<Eigen::type2index<0>, Eigen::type2index<0>>(); in dsizes_00()
75 inline Eigen::IndexList<Eigen::type2index<1>, Eigen::type2index<0>>
77 return Eigen::IndexList<Eigen::type2index<1>, Eigen::type2index<0>>(); in dsizes_10()
Dcwise_ops_common.h375 inline Eigen::IndexList<int, Eigen::type2index<1>> NByOne(int n) {
376 Eigen::IndexList<int, Eigen::type2index<1>> ret;
380 inline Eigen::IndexList<Eigen::type2index<1>, int> OneByM(int m) {
381 Eigen::IndexList<Eigen::type2index<1>, int> ret;
Dfill_functor.cc150 Eigen::IndexList<Eigen::type2index<1> > rank1; in operator ()()
/external/eigen/unsupported/test/
Dcxx11_tensor_index_list.cpp54 typedef Eigen::IndexList<Eigen::type2index<0>> Dims0; in test_type2index_list()
55 typedef Eigen::IndexList<Eigen::type2index<0>, Eigen::type2index<1>> Dims1; in test_type2index_list()
56 typedef Eigen::IndexList<Eigen::type2index<0>, Eigen::type2index<1>, Eigen::type2index<2>> Dims2; in test_type2index_list()
57 …typedef Eigen::IndexList<Eigen::type2index<0>, Eigen::type2index<1>, Eigen::type2index<2>, Eigen::… in test_type2index_list()
58 …ef Eigen::IndexList<Eigen::type2index<0>, Eigen::type2index<1>, Eigen::type2index<2>, Eigen::type2… in test_type2index_list()
318 typedef IndexList<type2index<0>, int, type2index<2>, int> ReductionIndices; in test_mixed_index_list()
333 typedef IndexList<type2index<0>, type2index<1>, type2index<2>, type2index<3>> ReductionList; in test_mixed_index_list()
366 Eigen::IndexList<Eigen::type2index<1>, int> dim1; in test_dim_check()
368 Eigen::IndexList<Eigen::type2index<1>, int> dim2; in test_dim_check()
Dcxx11_tensor_reduction.cpp376 Eigen::IndexList<Eigen::type2index<1>, Eigen::type2index<3> > reduction_axis; in test_static_dims()
407 Eigen::IndexList<Eigen::type2index<0>, Eigen::type2index<1> > reduction_axis; in test_innermost_last_dims()
438 Eigen::IndexList<Eigen::type2index<2>, Eigen::type2index<3>> reduction_axis; in test_innermost_first_dims()
469 Eigen::IndexList<Eigen::type2index<1>, Eigen::type2index<2>> reduction_axis; in test_reduce_middle_dims()
Dcxx11_tensor_broadcasting.cpp119 Eigen::IndexList<Eigen::type2index<2>, Eigen::type2index<3>, Eigen::type2index<4>> broadcasts; in test_static_broadcasting()
/external/eigen/unsupported/Eigen/CXX11/src/Tensor/
DTensorIndexList.h41 struct type2index { struct
67 template<DenseIndex n> struct NumTraits<type2index<n> > argument
90 EIGEN_DEVICE_FUNC void update_value(type2index<n>& val, DenseIndex new_val) {
110 struct is_compile_time_constant<type2index<idx> > {
114 struct is_compile_time_constant<const type2index<idx> > {
118 struct is_compile_time_constant<type2index<idx>& > {
122 struct is_compile_time_constant<const type2index<idx>& > {
/external/eigen/bench/tensors/
Dtensor_benchmarks.h216 Eigen::IndexList<Eigen::type2index<1>, Eigen::type2index<2> > strides; in striding()
244 Eigen::IndexList<Eigen::type2index<1>, int> broadcast; in broadcasting()
326 Eigen::IndexList<Eigen::type2index<0>> sum_along_dim; in rowReduction()
356 Eigen::IndexList<Eigen::type2index<1>> sum_along_dim; in colReduction()

12