Home
last modified time | relevance | path

Searched defs:linear_index (Results 1 – 21 of 21) sorted by relevance

/aosp_15_r20/external/pthreadpool/src/
H A Dfastpath.c184 const size_t linear_index = pthreadpool_decrement_fetch_relaxed_size_t(&other_thread->range_end); in pthreadpool_thread_parallelize_2d_fastpath() local
233 const size_t linear_index = pthreadpool_decrement_fetch_relaxed_size_t(&other_thread->range_end); in pthreadpool_thread_parallelize_2d_tile_1d_fastpath() local
285 const size_t linear_index = pthreadpool_decrement_fetch_relaxed_size_t(&other_thread->range_end); in pthreadpool_thread_parallelize_2d_tile_2d_fastpath() local
347 const size_t linear_index = pthreadpool_decrement_fetch_relaxed_size_t(&other_thread->range_end); in pthreadpool_thread_parallelize_2d_tile_2d_with_uarch_fastpath() local
401 const size_t linear_index = pthreadpool_decrement_fetch_relaxed_size_t(&other_thread->range_end); in pthreadpool_thread_parallelize_3d_fastpath() local
457 const size_t linear_index = pthreadpool_decrement_fetch_relaxed_size_t(&other_thread->range_end); in pthreadpool_thread_parallelize_3d_tile_1d_fastpath() local
517 const size_t linear_index = pthreadpool_decrement_fetch_relaxed_size_t(&other_thread->range_end); in pthreadpool_thread_parallelize_3d_tile_2d_fastpath() local
587 const size_t linear_index = pthreadpool_decrement_fetch_relaxed_size_t(&other_thread->range_end); in pthreadpool_thread_parallelize_3d_tile_2d_with_uarch_fastpath() local
649 const size_t linear_index = pthreadpool_decrement_fetch_relaxed_size_t(&other_thread->range_end); in pthreadpool_thread_parallelize_4d_fastpath() local
713 const size_t linear_index = pthreadpool_decrement_fetch_relaxed_size_t(&other_thread->range_end); in pthreadpool_thread_parallelize_4d_tile_1d_fastpath() local
[all …]
H A Dportable-api.c171 const size_t linear_index = pthreadpool_decrement_fetch_relaxed_size_t(&other_thread->range_end); in thread_parallelize_2d() local
215 const size_t linear_index = pthreadpool_decrement_fetch_relaxed_size_t(&other_thread->range_end); in thread_parallelize_2d_tile_1d() local
262 const size_t linear_index = pthreadpool_decrement_fetch_relaxed_size_t(&other_thread->range_end); in thread_parallelize_2d_tile_2d() local
319 const size_t linear_index = pthreadpool_decrement_fetch_relaxed_size_t(&other_thread->range_end); in thread_parallelize_2d_tile_2d_with_uarch() local
368 const size_t linear_index = pthreadpool_decrement_fetch_relaxed_size_t(&other_thread->range_end); in thread_parallelize_3d() local
419 const size_t linear_index = pthreadpool_decrement_fetch_relaxed_size_t(&other_thread->range_end); in thread_parallelize_3d_tile_1d() local
474 const size_t linear_index = pthreadpool_decrement_fetch_relaxed_size_t(&other_thread->range_end); in thread_parallelize_3d_tile_2d() local
539 const size_t linear_index = pthreadpool_decrement_fetch_relaxed_size_t(&other_thread->range_end); in thread_parallelize_3d_tile_2d_with_uarch() local
596 const size_t linear_index = pthreadpool_decrement_fetch_relaxed_size_t(&other_thread->range_end); in thread_parallelize_4d() local
655 const size_t linear_index = pthreadpool_decrement_fetch_relaxed_size_t(&other_thread->range_end); in thread_parallelize_4d_tile_1d() local
[all …]
H A Dlegacy-api.c76 static void compute_3d_tiled(const struct compute_3d_tiled_context* context, size_t linear_index) { in compute_3d_tiled()
153 static void compute_4d_tiled(const struct compute_4d_tiled_context* context, size_t linear_index) { in compute_4d_tiled()
/aosp_15_r20/external/executorch/kernels/portable/cpu/util/test/
H A Dbroadcast_test.cpp135 size_t linear_index = 0; in linearize_indexes() local
155 auto linear_index = linearize_indexes(indexes, DIMS, t); in TEST() local
177 size_t linear_index = linearize_access_indexes( in TEST() local
187 size_t linear_index = linearize_access_indexes( in TEST() local
/aosp_15_r20/external/pytorch/aten/src/ATen/native/cuda/
H A DTensorTransformations.cu33 …for (IndexType linear_index = blockIdx.x * blockDim.x + threadIdx.x; linear_index < N; linear_inde… in C10_LAUNCH_BOUNDS_2() local
60 int64_t linear_index = blockIdx.x * blockDim.x + threadIdx.x; in C10_LAUNCH_BOUNDS_1() local
93 int64_t linear_index = blockIdx.x * blockDim.x + threadIdx.x; in C10_LAUNCH_BOUNDS_1() local
H A DTensorFactories.cu251 int64_t linear_index = blockIdx.x * blockDim.x + threadIdx.x; in tril_indices_kernel() local
327 int64_t linear_index = blockIdx.x * blockDim.x + threadIdx.x; in triu_indices_kernel() local
H A DActivationGluKernel.cu86 const uint32_t linear_index = blockIdx.x * blockDim.x + threadIdx.x; in glu_backward_kernel() local
H A DRreluWithNoise.cu44 for (int linear_index = idx; linear_index < rounded_size; linear_index += grid_stride) { in rrelu_with_noise_cuda_kernel() local
H A DDistributionTemplates.h80 …for(int linear_index = idx; linear_index < rounded_size; linear_index += blockDim.x * gridDim.x * … in C10_LAUNCH_BOUNDS_2() local
/aosp_15_r20/external/pytorch/caffe2/utils/threadpool/
H A Dpthreadpool.cc42 static void compute_1d_tiled(void* context_, size_t linear_index) { in compute_1d_tiled()
79 static void compute_2d(void* context_, size_t linear_index) { in compute_2d()
124 static void compute_2d_tiled(void* context_, size_t linear_index) { in compute_2d_tiled()
189 size_t linear_index) { in compute_3d_tiled()
281 size_t linear_index) { in compute_4d_tiled()
/aosp_15_r20/external/executorch/kernels/portable/cpu/util/
H A Dbroadcast_util.cpp262 size_t linear_index, in delinearize_index()
276 size_t linear_index, in delinearize_index()
294 size_t linear_index = 0; in linearize_access_indexes() local
/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/
H A Dindex_util.cc76 int64_t linear_index = 0; in MultidimensionalIndexToLinearIndex() local
93 const Shape& shape, int64_t linear_index) { in LinearIndexToMultidimensionalIndex()
H A Dliteral_comparison.cc304 int64_t linear_index; member
394 void CompareValues(T expected, T actual, int64_t linear_index) { in CompareValues()
484 int64_t linear_index) { in CompareValues()
498 int64_t linear_index) { in CompareValues()
H A Dindex_util_test.cc132 for (auto linear_index : linear_indexes) { in TEST() local
H A Dlayout_util.cc544 int64_t linear_index = 0; in LinearIndex() local
H A Dliteral.cc380 auto linear_index = [](const Shape& shape, in CopySliceFromInternal() local
/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/gpu/
H A Dparallel_loop_emitter.cc267 llvm::Value* linear_index = in EmitIndexAndSetExitBasicBlock() local
H A Dir_emitter_unnested.cc3403 llvm::Value* linear_index = in EmitXTileLoop() local
/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/llvm_ir/
H A Dir_array.cc352 llvm::Value* linear_index = GetConstantWithIndexType(0); in SourceIndexOfBitcast() local
/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/cpu/
H A Dir_emitter.cc2304 llvm::Value* linear_index = array_index.Linearize(dynamic_dims, &b_); in HandleSliceToDynamic() local
2371 llvm::Value* linear_index = array_index.Linearize(dynamic_dims, &b_); in HandlePadToStatic() local
/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/
H A Dhlo_parser.cc3581 int64_t linear_index = 0; in ParseDenseLiteral() local