Searched refs:index_j (Results 1 – 4 of 4) sorted by relevance
| /external/pthreadpool/src/ |
| D | threadpool-legacy.c | 94 const size_t index_j = tile_index_i_j.remainder * max_tile_j; in compute_3d_tiled() local
97 const size_t tile_j = min(max_tile_j, context->range_j - index_j); in compute_3d_tiled()
99 context->function(context->argument, index_i, index_j, index_k, tile_i, tile_j, tile_k); in compute_3d_tiled()
174 const size_t index_j = tile_index_i_j.remainder * max_tile_j; in compute_4d_tiled() local
178 const size_t tile_j = min(max_tile_j, context->range_j - index_j); in compute_4d_tiled()
181 …context->function(context->argument, index_i, index_j, index_k, index_l, tile_i, tile_j, tile_k, t… in compute_4d_tiled()
|
| D | threadpool-pthreads.c | 706 const size_t index_j = tile_index.remainder * max_tile_j; in compute_2d_tile_1d() local
707 const size_t tile_j = min(max_tile_j, context->range_j - index_j); in compute_2d_tile_1d()
708 context->task(context->argument, index_i, index_j, tile_j); in compute_2d_tile_1d()
766 const size_t index_j = tile_index.remainder * max_tile_j; in compute_2d_tile_2d() local
768 const size_t tile_j = min(max_tile_j, context->range_j - index_j); in compute_2d_tile_2d()
769 context->task(context->argument, index_i, index_j, tile_i, tile_j); in compute_2d_tile_2d()
833 const size_t index_j = tile_index_i_j.remainder * max_tile_j; in compute_3d_tile_2d() local
835 const size_t tile_j = min(max_tile_j, context->range_j - index_j); in compute_3d_tile_2d()
837 context->task(context->argument, index_i, index_j, index_k, tile_j, tile_k); in compute_3d_tile_2d()
910 const size_t index_j = tile_index_i_j.remainder; in compute_4d_tile_2d() local
[all …]
|
| /external/eigen/bench/btl/generic_bench/init/ |
| 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()
|
| /external/tensorflow/tensorflow/core/kernels/ |
| D | gather_nd_op_gpu.cu.cc | 44 const Index index_j = ldg(indices_i + j); in GatherSliceOpKernel() local
45 out_of_bounds |= !FastBoundsCheck(index_j, batch_indices[j]); in GatherSliceOpKernel()
46 offset += batch_strides[j] * index_j; in GatherSliceOpKernel()
|