| /aosp_15_r20/external/pytorch/aten/src/ATen/native/cuda/ |
| H A D | TensorShape.cu | 163 int64_t num_chunks) { in split_with_sizes_copy_out_contiguous_no_cast_kernel()
238 int64_t num_chunks = tensor.numel(); in get_num_chunks() local
279 int64_t num_chunks, in get_chunk_cat_out_sizes()
446 int64_t num_chunks) { in get_pad_size()
461 int64_t num_chunks, in get_chunk_size()
491 int64_t num_chunks, in get_chunk_cat_metadata()
559 int64_t num_chunks, in _chunk_cat_out_cuda_contiguous()
630 const auto num_chunks = detail::get_num_chunks(self, dim); in split_with_sizes_copy_out_cuda_contiguous_no_cast() local
778 Tensor _chunk_cat_cuda(TensorList tensors, int64_t dim, int64_t num_chunks) { in _chunk_cat_cuda()
804 int64_t num_chunks, in _chunk_cat_out_cuda()
|
| /aosp_15_r20/external/perfetto/src/tracing/core/ |
| H A D | shared_memory_abi.cc | 56 size_t num_chunks = SharedMemoryABI::kNumChunksForLayout[i]; in InitChunkSizes() local
143 const size_t num_chunks = GetNumChunksFromHeaderBitmap(header_bitmap); in GetChunkUnchecked() local
166 const size_t num_chunks = GetNumChunksFromHeaderBitmap(header_bitmap); in TryAcquireChunk() local
219 const uint32_t num_chunks = GetNumChunksFromHeaderBitmap(bitmap); in GetFreeChunks() local
|
| H A D | shared_memory_abi_unittest.cc | 84 const size_t num_chunks = SharedMemoryABI::GetNumChunksFromHeaderBitmap( in TEST_P() local
210 const size_t num_chunks = in TEST_P() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/core/common_runtime/ |
| H A D | base_collective_executor.cc | 58 int64_t num_chunks) { in AlignedChunkElts()
94 CollectiveAdapterImpl(Tensor* output, int64_t num_chunks, in CollectiveAdapterImpl()
195 CollectiveAdapter* MakeCollectiveAdapter(Tensor* output, int num_chunks, in MakeCollectiveAdapter()
|
| /aosp_15_r20/external/webrtc/modules/audio_processing/agc2/rnn_vad/ |
| H A D | test_utils.cc | 102 const int num_chunks = rtc::CheckedDivExact(reader->size(), kBufSize24kHz); in CreatePitchBuffer24kHzReader() local
112 const int num_chunks = rtc::CheckedDivExact(reader->size(), kChunkSize); in CreateLpResidualAndPitchInfoReader() local
|
| H A D | test_utils.h | 68 const int num_chunks; member
|
| /aosp_15_r20/external/tensorflow/tensorflow/core/debug/ |
| H A D | debug_io_utils.cc | 62 const uint64 wall_time_us, const size_t num_chunks, in PrepareChunkEventProto()
156 const size_t num_chunks = cutoffs.size(); in WrapStringTensorAsEvents() local
205 const size_t num_chunks = in WrapTensorAsEvents() local
253 const size_t num_chunks = in PublishEncodedGraphDefInChunks() local
|
| H A D | debugger_event_metadata.proto | 9 int32 num_chunks = 3; field
|
| /aosp_15_r20/external/webrtc/modules/audio_processing/vad/ |
| H A D | voice_activity_detector_unittest.cc | 72 size_t num_chunks = 0; in TEST() local
107 size_t num_chunks = 0; in TEST() local
|
| /aosp_15_r20/external/pytorch/aten/src/ATen/native/cpu/ |
| H A D | SoftMaxKernel.cpp | 278 int64_t num_chunks = divup(inner_size, CHUNK_SIZE); in _vec_softmax_backward() local
363 int64_t num_chunks = divup(inner_size, CHUNK_SIZE); in _vec_softmax_backward() local
489 int64_t num_chunks = divup(inner_size, CHUNK_SIZE); in _vec_log_softmax_backward() local
573 int64_t num_chunks = divup(inner_size, CHUNK_SIZE); in _vec_log_softmax_backward() local
904 int64_t num_chunks = divup(inner_size, CHUNK_SIZE); in _vec_logsoftmax() local
1010 int64_t num_chunks = divup(inner_size, CHUNK_SIZE); in _vec_logsoftmax() local
|
| /aosp_15_r20/external/sdv/vsomeip/third_party/boost/pool/include/boost/pool/ |
| D | pool.hpp | 485 const size_type num_chunks = total_req_size / partition_size + in BOOST_PREVENT_MACRO_SUBSTITUTION() local
502 const size_type num_chunks = total_req_size / partition_size + in ordered_free() local
798 const size_type num_chunks = total_req_size / partition_size + in ordered_malloc() local
|
| /aosp_15_r20/external/sdv/vsomeip/third_party/boost/container/test/ |
| D | pool_resource_test.hpp | 310 const std::size_t num_chunks = bi::detail::floor_log2(opts.max_blocks_per_chunk)+1u; in test_do_allocate_deallocate() local
341 const std::size_t num_chunks = bi::detail::floor_log2(opts.max_blocks_per_chunk)+1u; in test_do_allocate_deallocate() local
|
| /aosp_15_r20/external/libdrm/amdgpu/ |
| H A D | amdgpu_cs.c | 264 uint32_t i, size, num_chunks, bo_list_handle = 0, sem_count = 0; in amdgpu_cs_submit_one() local
899 int num_chunks, in amdgpu_cs_submit_raw()
930 int num_chunks, in amdgpu_cs_submit_raw2()
|
| /aosp_15_r20/prebuilts/runtime/mainline/platform/sdk/include/external/perfetto/include/perfetto/ext/tracing/core/ |
| H A D | shared_memory_abi.h | 496 const uint32_t num_chunks = GetNumChunksForLayout(layout); in is_page_complete() local
598 const uint32_t num_chunks = GetNumChunksForLayout(page_layout); in GetUsedChunks() local
|
| /aosp_15_r20/external/perfetto/include/perfetto/ext/tracing/core/ |
| H A D | shared_memory_abi.h | 496 const uint32_t num_chunks = GetNumChunksFromHeaderBitmap(bitmap); in is_page_complete() local
606 const uint32_t num_chunks = GetNumChunksFromHeaderBitmap(header_bitmap); in GetUsedChunks() local
|
| /aosp_15_r20/external/cronet/third_party/rust/chromium_crates_io/vendor/base64-0.13.1/src/ |
| H A D | decode.rs | 153 let num_chunks = num_chunks(input_bytes); in decode_config_buf() localVariable
191 fn num_chunks(input: &[u8]) -> usize { in num_chunks() function
|
| /aosp_15_r20/external/pytorch/torch/csrc/api/src/nn/options/ |
| H A D | rnn.cpp | 31 int64_t num_chunks) in RNNCellOptionsBase()
|
| /aosp_15_r20/external/cronet/third_party/brotli/enc/ |
| H A D | compound_dictionary.h | 45 size_t num_chunks; member
|
| /aosp_15_r20/external/pytorch/torch/testing/_internal/ |
| H A D | autocast_test_lists.py | 11 def _rnn_cell_args(self, n, num_chunks, is_lstm, dev, dtype): argument
|
| /aosp_15_r20/external/mesa3d/src/intel/vulkan/tests/ |
| H A D | state_pool_test_helper.h | 49 const unsigned num_chunks = ctx->states_per_thread / chunk_size; in alloc_states() local
|
| /aosp_15_r20/external/mesa3d/src/intel/vulkan_hasvk/tests/ |
| H A D | state_pool_test_helper.h | 49 const unsigned num_chunks = ctx->states_per_thread / chunk_size; in alloc_states() local
|
| /aosp_15_r20/external/bsdiff/ |
| H A D | split_patch_writer_unittest.cc | 19 void SetUpForSize(size_t num_chunks, in SetUpForSize()
|
| /aosp_15_r20/external/pytorch/torch/distributed/pipelining/ |
| H A D | microbatch.py | 120 num_chunks, argument
|
| /aosp_15_r20/external/pytorch/aten/src/ATen/native/ |
| H A D | TensorShape.h | 82 inline int64_t preprocess_chunk_cat_inputs(TensorList tensors, int64_t dim, int64_t num_chunks) { in preprocess_chunk_cat_inputs()
|
| /aosp_15_r20/external/deqp-deps/SPIRV-Tools/source/fuzz/ |
| D | shrinker.cpp | 168 const uint32_t num_chunks = in Run() local
|