Home
last modified time | relevance | path

Searched refs:thread_depth (Results 1 – 2 of 2) sorted by relevance

/external/tensorflow/tensorflow/core/kernels/
Ddepthwise_conv_op_gpu.h235 const int thread_depth = threadIdx.x;
241 const int thread_idx = thread_pix * kBlockDepth + thread_depth;
250 const int tensor_idx = thread_pix * in_depth + thread_depth;
254 const int data_idx = data_pix * kBlockDepth + thread_depth;
258 const int tile_idx = tile_pix * kBlockDepth + thread_depth;
260 const int max_channel = in_depth - thread_depth;
264 tile_size + thread_depth +
527 const int thread_depth = threadIdx.z;
531 const int thread_idx = thread_depth * block_pixels + thread_pix;
540 const int tensor_idx = thread_depth * in_pixels + thread_pix;
[all …]
/external/llvm-project/polly/lib/External/ppcg/
Dgpu_group.c471 int thread_depth; member
556 next_thread_x = next(space, data->thread_depth + data->n_thread - 1); in access_is_coalesced()
613 ids = ppcg_scop_generate_names(data->scop, data->thread_depth, "s"); in access_is_bijective()
706 if (data->thread_depth == data->kernel_depth) in compute_accessed_by_single_thread_depth()
707 return data->thread_depth; in compute_accessed_by_single_thread_depth()
714 space = isl_space_add_dims(space, isl_dim_set, data->thread_depth); in compute_accessed_by_single_thread_depth()
727 for (i = data->thread_depth - 1; i >= data->kernel_depth; --i) { in compute_accessed_by_single_thread_depth()
794 group->min_depth = data->thread_depth; in set_depth()
1012 if (data->shared_depth == data->thread_depth) in shared_access()
1159 acc = isl_map_project_out(acc, isl_dim_in, data->thread_depth, in compute_group_bounds_core()
[all …]