| /external/XNNPACK/test/ |
| D | gemm-microkernel-tester.h | 249 for (size_t n_index = 0; n_index < n(); n_index++) { variable
251 acc[m_index * n() + n_index] +=
253 (int32_t(b[n_index * k() + k_index]) - int32_t(b_zero_point()));
255 acc[m_index * n() + n_index] += bias[n_index];
289 for (size_t n_index = 0; n_index < n(); n_index++) { variable
290 …c_ref[m_index * n() + n_index] = xnn_qu8_requantize_q31(acc[m_index * n() + n_index], scalar_requa…
366 for (size_t n_index = 0; n_index < n(); n_index++) { variable
370 acc[m_index * n() + n_index] +=
372 … (int32_t(b[(n_index * ks() + ks_index) * k() + k_index]) - int32_t(b_zero_point()));
374 acc[m_index * n() + n_index] +=
[all …]
|
| /external/llvm-project/polly/lib/External/ppcg/ |
| D | gpu_array_tile.c | 28 struct gpu_array_tile *gpu_array_tile_create(isl_ctx *ctx, int n_index) in gpu_array_tile_create() argument
38 tile->bound = isl_alloc_array(ctx, struct gpu_array_bound, n_index); in gpu_array_tile_create()
42 tile->n = n_index; in gpu_array_tile_create()
44 for (i = 0; i < n_index; ++i) { in gpu_array_tile_create()
|
| D | gpu.h | 29 int n_index; member
76 unsigned n_index; member
141 unsigned n_index; member
|
| D | gpu_array_tile.h | 54 struct gpu_array_tile *gpu_array_tile_create(isl_ctx *ctx, int n_index);
|
| D | gpu.c | 114 int n_index; in compute_extent() local
121 n_index = isl_set_dim(accessed, isl_dim_set); in compute_extent()
122 if (n_index == 0) in compute_extent()
127 outer = isl_set_project_out(outer, isl_dim_set, 1, n_index - 1); in compute_extent()
149 if (array->n_index != 0) in is_read_only_scalar()
191 int n_index; in extract_array_info() local
195 n_index = isl_set_dim(pa->extent, isl_dim_set); in extract_array_info()
200 info->n_index = n_index; in extract_array_info()
405 return array->n_index == 0; in gpu_array_is_scalar()
458 for (i = 0; i < array->n_index; ++i) { in gpu_array_positive_size_guard()
[all …]
|
| D | gpu_print.c | 49 for (i = 0; i < array->n_index; ++i) { in gpu_array_info_print_size()
97 if (array->n_index != 0 && !array->linearize) in gpu_array_info_print_declaration_argument()
|
| D | cuda.c | 52 if (!array->linearize && array->n_index > 1) in declare_device_array()
56 if (!array->linearize && array->n_index > 1) { in declare_device_array()
58 for (i = 1; i < array->n_index; i++) { in declare_device_array()
414 for (j = 0; j < var->array->n_index; ++j) { in print_kernel_var()
|
| D | gpu_group.c | 853 group->slice = access->n_index < local->array->n_index; in populate_array_references()
1096 int n_index = group->array->n_index; in compute_group_bounds_core() local
1131 group->array->n_index); in compute_group_bounds_core()
1168 group->private_tile = gpu_array_tile_create(ctx, n_index); in compute_group_bounds_core()
|
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | qr_expander.cc | 301 int64 n_index = batch_dims.size() + 1; in CompactWYRepresentation() local
321 ConcatVectors(batch_dim_indices, {n_index})); in CompactWYRepresentation()
|
| /external/deqp/external/openglcts/modules/glesext/tessellation_shader/ |
| D | esextcTessellationShaderTessellation.cpp | 1265 for (unsigned int n_index = interleave_rate; n_index < bo_size; n_index += interleave_rate) in initRun() local
1267 bo_contents[n_index] = restart_index; in initRun()
1270 n_index++; in initRun()
|
| /external/llvm-project/polly/lib/CodeGen/ |
| D | PPCGCodeGeneration.cpp | 1107 for (unsigned int i = 1; i < Array->n_index; i++) { in getArraySize()
2015 for (long j = 1, n = Kernel->array[i].array->n_index; j < n; j++) { in createKernelFunctionDecl()
2228 for (unsigned int j = 1; j < Var.array->n_index; ++j) { in createKernelVariables()
2235 for (int j = Var.array->n_index - 1; j >= 0; --j) { in createKernelVariables()
2788 Access->n_index = Acc->getScopArrayInfo()->getNumberOfDimensions(); in getStmtAccesses()
2903 if (PPCGArray.n_index > 0) { in setArrayBounds()
2913 Dom = isl_set_project_out(Dom, isl_dim_set, 1, PPCGArray.n_index - 1); in setArrayBounds()
2927 for (unsigned i = 1; i < PPCGArray.n_index; ++i) { in setArrayBounds()
3012 PPCGArray.n_index = Array->getNumberOfDimensions(); in createArrays()
|
| /external/deqp/external/openglcts/modules/gl/ |
| D | gl4cVertexAttrib64BitTest.cpp | 3521 for (unsigned short n_index = 0; n_index < (unsigned short)m_n_batches; ++n_index) in initBuffers() local
3523 m_bo_index_data[n_index] = (unsigned short)((unsigned short)(m_n_batches - 1) - n_index); in initBuffers()
|
| D | gl4cSparseBufferTests.cpp | 6028 for (unsigned int n_index = 0; n_index < m_n_vertices_per_instance + 1; ++n_index) in initTestData() local
6030 m_index_data[n_index] = m_n_vertices_per_instance - n_index; in initTestData()
|
| /external/tensorflow/tensorflow/compiler/mlir/xla/transforms/ |
| D | legalize_tf.cc | 5996 int64_t n_index = batch_dims.size() + 1; in ComputeWYRepresentation() local
6021 builder->getI64IntegerAttr(n_index)); in ComputeWYRepresentation()
6047 beta_broadcast_dims.push_back(n_index); in ComputeWYRepresentation()
6067 beta_broadcast_dims.push_back(n_index); in ComputeWYRepresentation()
|