Home
last modified time | relevance | path

Searched refs:index_j (Results 1 – 3 of 3) sorted by relevance

/external/eigen/bench/btl/generic_bench/init/
Dinit_function.hh28 double simple_function(int index_i, int index_j) in simple_function() argument
30 return index_i+index_j; in simple_function()
/external/pthreadpool/src/
Dlegacy-api.c85 const size_t index_j = tile_index_i_j.remainder * max_tile_j; in compute_3d_tiled() local
88 const size_t tile_j = min(max_tile_j, context->range_j - index_j); 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()
165 const size_t index_j = tile_index_i_j.remainder * max_tile_j; in compute_4d_tiled() local
169 const size_t tile_j = min(max_tile_j, context->range_j - index_j); 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()
/external/tensorflow/tensorflow/core/kernels/
Dgather_nd_op_gpu.cu.cc44 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()