Home
last modified time | relevance | path

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

/external/pthreadpool/src/
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 …]
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 …]
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()
/external/tensorflow/tensorflow/compiler/xla/
Dindex_util.cc76 int64 linear_index = 0; in MultidimensionalIndexToLinearIndex() local
93 const Shape& shape, int64 linear_index) { in LinearIndexToMultidimensionalIndex()
Dindex_util_test.cc132 for (auto linear_index : linear_indexes) { in TEST() local
Dliteral_comparison.cc301 int64 linear_index; member
389 void CompareValues(T expected, T actual, int64 linear_index) { in CompareValues()
478 void CompareValues(complex64 expected, complex64 actual, int64 linear_index) { in CompareValues()
492 int64 linear_index) { in CompareValues()
Dliteral.cc275 auto linear_index = [](const Shape& shape, in CopySliceFromInternal() local
/external/tensorflow/tensorflow/compiler/xla/service/gpu/
Dparallel_loop_emitter.cc135 llvm::Value* linear_index = in EmitIndexAndSetExitBasicBlock() local
Dir_emitter_unnested.cc3982 int64 linear_index = j * vector_size + i; in UnrollInnerTileLoop() local
/external/tensorflow/tensorflow/compiler/xla/service/llvm_ir/
Dir_array.cc323 llvm::Value* linear_index = GetConstantWithIndexType(0); in SourceIndexOfBitcast() local
/external/tensorflow/tensorflow/compiler/xla/service/cpu/
Dir_emitter.cc2176 llvm::Value* linear_index = array_index.Linearize(dynamic_dims, &b_); in HandleSliceToDynamic() local
2240 llvm::Value* linear_index = array_index.Linearize(dynamic_dims, &b_); in HandlePadToStatic() local
/external/tensorflow/tensorflow/compiler/xla/service/
Dhlo_parser.cc3125 int64 linear_index = 0; in ParseDenseLiteral() local