Home
last modified time | relevance | path

Searched refs:IndexList (Results 1 – 25 of 43) sorted by relevance

12

/external/eigen/unsupported/Eigen/CXX11/src/Tensor/
DTensorIndexList.h300 struct IndexList : internal::IndexTuple<FirstType, OtherTypes...> {
311 …EIGEN_DEVICE_FUNC constexpr IndexList(const internal::IndexTuple<FirstType, OtherTypes...>& other)…
312 …EIGEN_DEVICE_FUNC constexpr IndexList(FirstType& first, OtherTypes... other) : internal::IndexTupl…
313 EIGEN_DEVICE_FUNC constexpr IndexList() : internal::IndexTuple<FirstType, OtherTypes...>() { }
329 constexpr IndexList<FirstType, OtherTypes...> make_index_list(FirstType val1, OtherTypes... other_v…
330 return IndexList<FirstType, OtherTypes...>(val1, other_vals...);
353 template<typename FirstType, typename... OtherTypes> size_t array_prod(const IndexList<FirstType, O…
355 for (int i = 0; i < array_size<IndexList<FirstType, OtherTypes...> >::value; ++i) {
361 template<typename FirstType, typename... OtherTypes> struct array_size<IndexList<FirstType, OtherTy…
364 template<typename FirstType, typename... OtherTypes> struct array_size<const IndexList<FirstType, O…
[all …]
DTensorSyclTuple.h139 struct IndexList {};
156 typedef IndexList<Is...> type;
184 Tuple<Args..., T> append_base(Tuple<Args...> t, T a,IndexList<I...>) {
215 … Args2...> append_base(Tuple<Args1...> t1, Tuple<Args2...> t2, IndexList<I1...>, IndexList<I2...>)…
/external/tensorflow/tensorflow/core/kernels/image/
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()
/external/tensorflow/tensorflow/core/kernels/
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()
Dxent_op.h86 Eigen::IndexList<Eigen::type2index<kClassDim> > along_class; in Compute()
87 Eigen::IndexList<int, Eigen::type2index<1> > batch_by_one; in Compute()
89 Eigen::IndexList<int> batch_only; in Compute()
91 Eigen::IndexList<Eigen::type2index<1>, int> one_by_class; in Compute()
Deigen_pooling.h59 const Eigen::IndexList<Eigen::type2index<1>, Eigen::type2index<2> >,
60 const Eigen::IndexList<Eigen::type2index<2>,
125 const Eigen::IndexList<Eigen::type2index<1>, Eigen::type2index<2> >,
126 const Eigen::IndexList<Eigen::type2index<2>,
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>,
489 const Eigen::IndexList<Eigen::type2index<1>, Eigen::type2index<2> >,
490 const Eigen::IndexList<Eigen::type2index<2>,
[all …]
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()
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()
206 Eigen::IndexList<int> batch_only; in Compute()
208 Eigen::IndexList<Eigen::type2index<1>, int> one_by_class; in Compute()
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.cc74 Eigen::IndexList<int, int, int> bsc; in operator ()()
79 Eigen::IndexList<int, Eigen::type2index<1>, int> boc; in operator ()()
83 Eigen::IndexList<Eigen::type2index<1>, int, Eigen::type2index<1>> oso; in operator ()()
Dreduction_ops_common.h65 const Eigen::IndexList<Eigen::type2index<0>> kZero;
66 const Eigen::IndexList<Eigen::type2index<1>> kOne;
67 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()
Dfused_batch_norm_op.cu.cc73 Eigen::IndexList<Eigen::type2index<1>, Eigen::Index> one_by_depth; in operator ()()
75 Eigen::IndexList<Eigen::type2index<0> > reduction_axis; in operator ()()
76 Eigen::IndexList<Eigen::Index, Eigen::type2index<1> > rest_by_one; in operator ()()
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()
/external/llvm/lib/CodeGen/
DSlotIndexes.cpp113 for (IndexList::iterator I = indexList.begin(), E = indexList.end(); in renumberIndexes()
122 void SlotIndexes::renumberIndexes(IndexList::iterator curItr) { in renumberIndexes()
127 IndexList::iterator startItr = std::prev(curItr); in renumberIndexes()
170 IndexList::iterator ListB = startIdx.listEntry()->getIterator(); in repairIndexesInRange()
171 IndexList::iterator ListI = endIdx.listEntry()->getIterator(); in repairIndexesInRange()
213 for (IndexList::const_iterator itr = indexList.begin(); in dump()
/external/swiftshader/third_party/llvm-10.0/llvm/lib/CodeGen/
DSlotIndexes.cpp159 void SlotIndexes::renumberIndexes(IndexList::iterator curItr) { in renumberIndexes()
164 IndexList::iterator startItr = std::prev(curItr); in renumberIndexes()
207 IndexList::iterator ListB = startIdx.listEntry()->getIterator(); in repairIndexesInRange()
208 IndexList::iterator ListI = endIdx.listEntry()->getIterator(); in repairIndexesInRange()
250 for (IndexList::const_iterator itr = indexList.begin(); in dump()
/external/llvm-project/llvm/lib/CodeGen/
DSlotIndexes.cpp160 void SlotIndexes::renumberIndexes(IndexList::iterator curItr) { in renumberIndexes()
165 IndexList::iterator startItr = std::prev(curItr); in renumberIndexes()
208 IndexList::iterator ListB = startIdx.listEntry()->getIterator(); in repairIndexesInRange()
209 IndexList::iterator ListI = endIdx.listEntry()->getIterator(); in repairIndexesInRange()
251 for (IndexList::const_iterator itr = indexList.begin(); in dump()
/external/llvm/include/llvm/CodeGen/
DSlotIndexes.h343 typedef ilist<IndexListEntry> IndexList;
344 IndexList indexList;
347 IndexList graveyardList;
374 void renumberIndexes(IndexList::iterator curItr);
438 IndexList::iterator I = Index.listEntry()->getIterator();
439 IndexList::iterator E = indexList.end();
591 IndexList::iterator prevItr, nextItr;
608 IndexList::iterator newItr =
656 IndexList::iterator newItr;
/external/swiftshader/third_party/llvm-10.0/llvm/include/llvm/CodeGen/
DSlotIndexes.h319 using IndexList = ilist<IndexListEntry>;
320 IndexList indexList;
345 void renumberIndexes(IndexList::iterator curItr);
408 IndexList::iterator I = Index.listEntry()->getIterator();
409 IndexList::iterator E = indexList.end();
544 IndexList::iterator prevItr, nextItr;
561 IndexList::iterator newItr =
607 IndexList::iterator newItr;
/external/llvm-project/llvm/include/llvm/CodeGen/
DSlotIndexes.h319 using IndexList = ilist<IndexListEntry>;
320 IndexList indexList;
345 void renumberIndexes(IndexList::iterator curItr);
410 IndexList::iterator I = Index.listEntry()->getIterator();
411 IndexList::iterator E = indexList.end();
546 IndexList::iterator prevItr, nextItr;
563 IndexList::iterator newItr =
617 IndexList::iterator newItr;
/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 …typedef Eigen::IndexList<Eigen::type2index<0>, Eigen::type2index<1>, Eigen::type2index<2>, Eigen::… 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()
/external/llvm/utils/
Dllvm.grm245 | getelementptr OptInBounds "(" ^ ConstVal IndexList ^ ")"
382 IndexList ::= _ | IndexList ^ "," ResolvedVal ;
419 | getelementptr OptInBounds Types ValueRef IndexList
/external/llvm-project/llvm/utils/
Dllvm.grm246 | getelementptr OptInBounds "(" ^ ConstVal IndexList ^ ")"
383 IndexList ::= _ | IndexList ^ "," ResolvedVal ;
420 | getelementptr OptInBounds Types ValueRef IndexList
/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