Home
last modified time | relevance | path

Searched refs:type2index (Results 1 – 25 of 27) 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()
419 const Eigen::IndexList<Eigen::type2index<1>, Eigen::type2index<2> >,
420 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()
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()
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.cc62 Eigen::IndexList<Eigen::type2index<1> > rank1; in operator ()()
87 Eigen::IndexList<Eigen::type2index<1>, int> broadcast_dims; in operator ()()
89 Eigen::IndexList<int, Eigen::type2index<1> > reshape_dims; in operator ()()
Dmultinomial_op_gpu.cu.cc83 Eigen::IndexList<int, Eigen::type2index<1>, int> boc; in operator ()()
87 Eigen::IndexList<Eigen::type2index<1>, int, Eigen::type2index<1>> oso; 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()
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()
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;
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()
Dquantize_and_dequantize_op.h217 Eigen::IndexList<Eigen::type2index<0>, Eigen::type2index<2> > reduce_dims; in Compute()
Dfused_batch_norm_op.cu.cc69 Eigen::IndexList<Eigen::type2index<1>, Eigen::Index> one_by_depth; in operator ()()
71 Eigen::IndexList<Eigen::type2index<0> > reduction_axis; in operator ()()
72 Eigen::IndexList<Eigen::Index, Eigen::type2index<1> > rest_by_one; in operator ()()
Dcolorspace_op.h44 Eigen::IndexList<Eigen::type2index<1> > channel_axis; in operator()
Dfused_batch_norm_op.cc169 Eigen::IndexList<Eigen::type2index<1>, Eigen::Index> one_by_depth; in operator ()()
171 Eigen::IndexList<Eigen::type2index<0>> reduce_dims; in operator ()()
172 Eigen::IndexList<Eigen::Index, Eigen::type2index<1>> bcast_spec; in operator ()()
322 Eigen::IndexList<Eigen::type2index<1>, Eigen::Index> one_by_depth; in operator ()()
324 Eigen::IndexList<Eigen::Index, Eigen::type2index<1>> bcast_spec; in operator ()()
474 Eigen::IndexList<Eigen::type2index<1>, Eigen::Index> one_by_depth; in operator ()()
476 Eigen::IndexList<Eigen::Index, Eigen::type2index<1>> rest_by_one; in operator ()()
Dsparse_matmul_op.cc70 inline Eigen::IndexList<Eigen::type2index<0>, Eigen::type2index<0>>
72 return Eigen::IndexList<Eigen::type2index<0>, Eigen::type2index<0>>(); in dsizes_00()
74 inline Eigen::IndexList<Eigen::type2index<1>, Eigen::type2index<0>>
76 return Eigen::IndexList<Eigen::type2index<1>, Eigen::type2index<0>>(); in dsizes_10()
Dfill_functor.cc151 Eigen::IndexList<Eigen::type2index<1> > rank1; in operator ()()
Dcwise_ops_common.h452 inline Eigen::IndexList<int, Eigen::type2index<1>> NByOne(int n) {
453 Eigen::IndexList<int, Eigen::type2index<1>> ret;
457 inline Eigen::IndexList<Eigen::type2index<1>, int> OneByM(int m) {
458 Eigen::IndexList<Eigen::type2index<1>, int> ret;
Dpooling_ops_3d.cc260 Eigen::IndexList<Eigen::type2index<1>, int, int, int, in launch()
261 Eigen::type2index<1>> in launch()
433 Eigen::IndexList<Eigen::type2index<1>, int, int, int, in launch()
434 Eigen::type2index<1>> in launch()
/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