Home
last modified time | relevance | path

Searched refs:group_index (Results 1 – 20 of 20) sorted by relevance

/external/tensorflow/tensorflow/core/kernels/
Drandom_op_cpu.h127 int64 group_index;
128 for (group_index = start_group; group_index < limit_group_full;
129 ++group_index) {
134 gen.Skip(group_index * kGeneratorSkipPerOutputGroup);
145 gen.Skip(group_index * kGeneratorSkipPerOutputGroup);
Drandom_op_gpu.h190 int64 group_index = thread_id;
191 int64 offset = group_index * kGroupSize;
200 gen.Skip(group_index * kGeneratorSkipPerOutputGroup);
214 group_index += total_thread_count;
Dbias_op_gpu.cu.cc166 int32 group_index = blockIdx.x / bias_size; in BiasGradNCHW_SharedAtomics() local
169 for (int32 index = group_index * blockDim.x + threadIdx.x; in BiasGradNCHW_SharedAtomics()
/external/mesa3d/src/gallium/auxiliary/pipebuffer/
Dpb_slab.c64 struct pb_slab_group *group = &slabs->groups[entry->group_index]; in pb_slab_reclaim()
101 unsigned group_index; in pb_slab_alloc() local
109 group_index = heap * slabs->num_orders + (order - slabs->min_order); in pb_slab_alloc()
110 group = &slabs->groups[group_index]; in pb_slab_alloc()
139 slab = slabs->slab_alloc(slabs->priv, heap, 1 << order, group_index); in pb_slab_alloc()
Dpb_slab.h64 unsigned group_index; /* index into pb_slabs::groups */ member
94 unsigned group_index);
/external/XNNPACK/src/
Doperator-run.c26 size_t group_index, in xnn_compute_grouped_gemm() argument
40 (const void*) ((uintptr_t) context->a + mr_block_start * a_stride + group_index * k_scaled), in xnn_compute_grouped_gemm()
42 … ((uintptr_t) context->packed_w + nr_block_start * context->w_stride + group_index * context->wg_s… in xnn_compute_grouped_gemm()
43 …mr_block_start * cm_stride + (nr_block_start << context->log2_csize) + group_index * context->cg_s… in xnn_compute_grouped_gemm()
93 size_t group_index, in xnn_compute_grouped_batch_igemm() argument
108 … ((uintptr_t) context->packed_w + nr_block_start * context->w_stride + group_index * context->gw_s… in xnn_compute_grouped_batch_igemm()
109 …(void*) ((uintptr_t) context->c + group_index * context->gc_stride + batch_index * context->bc_str… in xnn_compute_grouped_batch_igemm()
112 context->a_offset + group_index * context->ga_stride + batch_index * context->ba_stride, in xnn_compute_grouped_batch_igemm()
119 size_t group_index, in xnn_compute_grouped_igemm() argument
134 … ((uintptr_t) context->packed_w + nr_block_start * context->w_stride + group_index * context->gw_s… in xnn_compute_grouped_igemm()
[all …]
/external/llvm-project/lldb/packages/Python/lldbsuite/test/tools/lldb-server/
Dlldbgdbserverutils.py518 for group_index, var_name in list(self.capture.items()):
519 capture_text = match.group(group_index)
527 for group_index, var_name in list(self.expect_captures.items()):
528 capture_text = match.group(group_index)
531 "No content to expect for group index {}".format(group_index))
777 for group_index, var_name in list(self._capture.items()):
778 capture_text = match.group(group_index)
781 "No content for group index {}".format(group_index))
/external/mesa3d/src/gallium/winsys/amdgpu/drm/
Damdgpu_bo.h143 unsigned group_index);
146 unsigned group_index);
Damdgpu_bo.c666 unsigned group_index, in amdgpu_bo_slab_alloc() argument
733 bo->u.slab.entry.group_index = group_index; in amdgpu_bo_slab_alloc()
758 unsigned group_index) in amdgpu_bo_slab_alloc_encrypted() argument
760 return amdgpu_bo_slab_alloc(priv, heap, entry_size, group_index, true); in amdgpu_bo_slab_alloc_encrypted()
765 unsigned group_index) in amdgpu_bo_slab_alloc_normal() argument
767 return amdgpu_bo_slab_alloc(priv, heap, entry_size, group_index, false); in amdgpu_bo_slab_alloc_normal()
/external/XNNPACK/src/xnnpack/
Dcompute.h88 size_t group_index,
105 size_t group_index,
191 size_t group_index,
200 size_t group_index,
225 size_t group_index,
235 size_t group_index,
288 size_t group_index,
334 size_t group_index,
/external/mesa3d/src/gallium/drivers/iris/
Diris_monitor.c112 unsigned group_index, in iris_get_monitor_group_info() argument
128 if (group_index >= perf_cfg->n_queries) { in iris_get_monitor_group_info()
133 struct gen_perf_query_info *query = &perf_cfg->queries[group_index]; in iris_get_monitor_group_info()
/external/mesa3d/src/gallium/winsys/radeon/drm/
Dradeon_drm_bo.h86 unsigned group_index);
Dradeon_drm_bo.c774 unsigned group_index) in radeon_bo_slab_alloc() argument
815 bo->u.slab.entry.group_index = group_index; in radeon_bo_slab_alloc()
/external/libaom/libaom/av1/encoder/
Dsvc_layercontext.h43 unsigned char group_index; member
Dsvc_layercontext.c160 gf_group->index = lc->group_index; in av1_restore_layer_context()
200 lc->group_index = gf_group->index; in av1_save_layer_context()
Dratectrl.c2088 lc->group_index = 0; in av1_get_one_pass_rt_params()
/external/tensorflow/tensorflow/compiler/xla/service/
Dhlo_sharding.cc59 std::vector<int64> group_index(indices.begin(), indices.end()); in PartialTile() local
60 group_index.pop_back(); in PartialTile()
61 int64 group = group_tile_assignment(group_index); in PartialTile()
/external/mesa3d/src/gallium/drivers/r600/sb/
Dsb_bc_dump.cpp73 sblog.print_w(++group_index, 5); in visit()
469 new_group(), group_index() { in bc_dump()
Dsb_pass.h95 unsigned new_group, group_index; variable
102 vpass(s), bc_data(bc_ptr), ndw(ndw), id(), new_group(), group_index() {} in bc_dump()
/external/mesa3d/src/gallium/drivers/radeonsi/
Dsi_state.h524 unsigned group_index);