| /aosp_15_r20/external/pytorch/caffe2/utils/threadpool/ |
| H A D | pthreadpool.cc | 132 const size_t index_i = q * max_tile_i; in compute_2d_tiled() local
134 const size_t tile_i = min(max_tile_i, context->range_i - index_i); in compute_2d_tiled()
136 context->function(context->argument, index_i, index_j, tile_i, tile_j); in compute_2d_tiled()
199 const size_t index_i = static_cast<uint32_t>(tile_index_i) * max_tile_i; in compute_3d_tiled() local
202 const size_t tile_i = min(max_tile_i, context->range_i - index_i); in compute_3d_tiled()
206 context->argument, index_i, index_j, index_k, tile_i, tile_j, tile_k); in compute_3d_tiled()
294 const size_t index_i = static_cast<uint32_t>(tile_index_i) * max_tile_i; in compute_4d_tiled() local
298 const size_t tile_i = min(max_tile_i, context->range_i - index_i); in compute_4d_tiled()
304 index_i, in compute_4d_tiled()
|
| /aosp_15_r20/external/pytorch/torch/csrc/jit/passes/onnx/pattern_conversion/ |
| H A D | pattern_conversion.cpp | 197 auto index_i = dim_index_map.find(i); in ReshapeToAdvancedIndexingFormat() local
198 TORCH_INTERNAL_ASSERT(index_i != dim_index_map.end()); in ReshapeToAdvancedIndexingFormat()
199 if (index_i->second.orig_node_kind == aten::index) { in ReshapeToAdvancedIndexingFormat()
220 auto index_i = dim_index_map.find(i); in ReshapeToAdvancedIndexingFormat() local
221 TORCH_INTERNAL_ASSERT(index_i != dim_index_map.end()); in ReshapeToAdvancedIndexingFormat()
222 Value* index = index_i->second.index; in ReshapeToAdvancedIndexingFormat()
223 switch (index_i->second.orig_node_kind) { in ReshapeToAdvancedIndexingFormat()
240 false, "Unexpected node kind ", index_i->second.orig_node_kind); in ReshapeToAdvancedIndexingFormat()
|
| /aosp_15_r20/external/pthreadpool/src/ |
| H A D | legacy-api.c | 84 const size_t index_i = tile_index_i_j.quotient * max_tile_i; in compute_3d_tiled() local
87 const size_t tile_i = min(max_tile_i, context->range_i - index_i); in compute_3d_tiled()
90 context->function(context->argument, index_i, index_j, index_k, tile_i, tile_j, tile_k); in compute_3d_tiled()
164 const size_t index_i = tile_index_i_j.quotient * max_tile_i; in compute_4d_tiled() local
168 const size_t tile_i = min(max_tile_i, context->range_i - index_i); in compute_4d_tiled()
172 …context->function(context->argument, index_i, index_j, index_k, index_l, tile_i, tile_j, tile_k, t… in compute_4d_tiled()
|
| /aosp_15_r20/external/eigen/bench/btl/generic_bench/init/ |
| H A D | init_function.hh | 28 double simple_function(int index_i, int index_j) in simple_function() argument
30 return index_i+index_j; in simple_function()
|
| /aosp_15_r20/external/sdv/vsomeip/third_party/boost/numeric/ublas/include/boost/numeric/ublas/ |
| D | matrix_sparse.hpp | 567 index1 = layout_type::index_i ((*it).first, size1_, size2_); in find1()
596 index1 = layout_type::index_i ((*it).first, size1_, size2_); in find1()
625 index1 = layout_type::index_i ((*it).first, size1_, size2_); in find2()
654 index1 = layout_type::index_i ((*it).first, size1_, size2_); in find2()
799 …BOOST_UBLAS_CHECK (layout_type::index_i ((*it_).first, m.size1 (), m.size2 ()) < (*this) ().size1 … in index1()
800 return layout_type::index_i ((*it_).first, m.size1 (), m.size2 ()); in index1()
953 …BOOST_UBLAS_CHECK (layout_type::index_i ((*it_).first, m.size1 (), m.size2 ()) < (*this) ().size1 … in index1()
954 return layout_type::index_i ((*it_).first, m.size1 (), m.size2 ()); in index1()
1131 …BOOST_UBLAS_CHECK (layout_type::index_i ((*it_).first, m.size1 (), m.size2 ()) < (*this) ().size1 … in index1()
1132 return layout_type::index_i ((*it_).first, m.size1 (), m.size2 ()); in index1()
[all …]
|
| D | matrix.hpp | 612 return layout_type::index_i (it_ - m.begin1 ().it_, m.size1 (), m.size2 ()); in index1()
762 return layout_type::index_i (it_ - m.begin1 ().it_, m.size1 (), m.size2 ()); in index1()
935 return layout_type::index_i (it_ - m.begin2 ().it_, m.size1 (), m.size2 ()); in index1()
1085 return layout_type::index_i (it_ - m.begin2 ().it_, m.size1 (), m.size2 ()); in index1()
1695 return layout_type::index_i (it_ - m.begin1 ().it_, m.size1 (), m.size2 ()); in index1()
1845 return layout_type::index_i (it_ - m.begin1 ().it_, m.size1 (), m.size2 ()); in index1()
2018 return layout_type::index_i (it_ - m.begin2 ().it_, m.size1 (), m.size2 ()); in index1()
2168 return layout_type::index_i (it_ - m.begin2 ().it_, m.size1 (), m.size2 ()); in index1()
|
| D | functional.hpp | 1449 size_type index_i (difference_type k, size_type /* size_i */, size_type size_j) { in index_i() function
1627 size_type index_i (difference_type k, size_type size_i, size_type /* size_j */) { in index_i() function
|