| /aosp_15_r20/external/tensorflow/tensorflow/stream_executor/rocm/ |
| H A D | rocm_blas.cc | 646 int batch_count, blas::ComputationType computation_type, in DoBlasGemmStridedBatchedWithAlgorithm()
666 int batch_count, uint64_t batch_stride, in ReorganizeMemory()
712 int batch_count, uint64_t batch_stride, ScratchAllocator *scratch_allocator, in AllocateStridedBuffer()
771 int ldc, int batch_count, ScratchAllocator *scratch_allocator) { in DoBlasGemmBatchedInternal()
865 int batch_count, ScratchAllocator *scratch_allocator) { in DoBlasGemmBatched()
887 int batch_count, ScratchAllocator *scratch_allocator) { in DoBlasGemmBatched()
905 int ldc, int batch_count, ScratchAllocator *scratch_allocator) { in DoBlasGemmBatched()
925 int ldc, int batch_count, ScratchAllocator *scratch_allocator) { in DoBlasGemmBatched()
945 int ldc, int batch_count, ScratchAllocator *scratch_allocator) { in DoBlasGemmBatched()
1014 int batch_count) { in DoBlasTrsmBatched()
[all …]
|
| H A D | rocm_fft.cc | 161 fft::Type type, int batch_count, ScratchAllocator *scratch_allocator) { in Initialize()
467 bool in_place_fft, int batch_count) { in CreateBatchedPlan()
485 bool in_place_fft, int batch_count, ScratchAllocator *scratch_allocator) { in CreateBatchedPlanWithScratchAllocator()
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/stream_executor/cuda/ |
| H A D | cuda_blas.cc | 1050 int batch_count, blas::ComputationType computation_type, in DoBlasInternalImpl()
1192 const DeviceMemorySlice<T> &c_ptrs_to_wrappers, int ldc, int batch_count, in DoBlasInternalImpl()
1329 const DeviceMemorySlice<Eigen::half> &c_array, int ldc, int batch_count, in DoBlasInternalImpl()
1346 const DeviceMemorySlice<float> &c_array, int ldc, int batch_count, in DoBlasInternalImpl()
1362 const DeviceMemorySlice<double> &c_array, int ldc, int batch_count, in DoBlasInternalImpl()
1380 int batch_count, ScratchAllocator *scratch_allocator) { in DoBlasInternalImpl()
1397 int batch_count, ScratchAllocator *scratch_allocator) { in DoBlasInternalImpl()
1412 DeviceMemoryBase *c, int ldc, int64_t stride_c, int batch_count) { in DoBlasInternalImpl()
1603 int batch_count) { in DoBlasInternalImpl()
1616 int batch_count) { in DoBlasInternalImpl()
[all …]
|
| H A D | cuda_fft.cc | 100 fft::Type type, int batch_count, ScratchAllocator *scratch_allocator) { in Initialize()
420 bool in_place_fft, int batch_count) { in CreateBatchedPlan()
448 bool in_place_fft, int batch_count, ScratchAllocator *scratch_allocator) { in CreateBatchedPlanWithScratchAllocator()
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/stream_executor/ |
| H A D | stream.cc | 1627 int batch_count) { in ThenBlasTrsmBatched()
1645 int batch_count) { in ThenBlasTrsmBatched()
1662 DeviceMemory<std::complex<float> *> *bs, int ldb, int batch_count) { in ThenBlasTrsmBatched()
1680 DeviceMemory<std::complex<double> *> *bs, int ldb, int batch_count) { in ThenBlasTrsmBatched()
1702 int ldc, int batch_count) { in ThenBlasGemmBatched()
1716 int ldc, int batch_count, ScratchAllocator *scratch_allocator) { in ThenBlasGemmBatchedWithScratch()
1742 int ldc, int batch_count) { in ThenBlasGemmBatched()
1755 int ldc, int batch_count, ScratchAllocator *scratch_allocator) { in ThenBlasGemmBatchedWithScratch()
1779 int ldc, int batch_count) { in ThenBlasGemmBatched()
1792 int ldc, int batch_count, ScratchAllocator *scratch_allocator) { in ThenBlasGemmBatchedWithScratch()
[all …]
|
| H A D | stream.h | 923 int64_t stride_c, int batch_count, blas::ComputationType computation_type, in ThenBlasGemmStridedBatchedWithAlgorithm()
1038 int64_t stride_c, int batch_count) { in ThenBlasGemmStridedBatched()
|
| /aosp_15_r20/external/pytorch/aten/src/ATen/cuda/ |
| H A D | CUDASparseDescriptors.cpp | 107 int batch_count = in createRawDnMatDescriptor() local
194 int batch_count = in CuSparseSpMatCsrDescriptor() local
|
| /aosp_15_r20/external/libopus/dnn/torch/fargan/ |
| H A D | adv_train_fargan.py | 138 batch_count = 0 variable
263 batch_count = batch_count + 1 variable
|
| /aosp_15_r20/external/pytorch/aten/src/ATen/native/cuda/ |
| H A D | PersistentSoftmax.cuh | 303 …nst input_t *src, int softmax_elements, int softmax_elements_stride, int batch_count, const bool *… in dispatch_softmax_forward()
353 … input_t *output, int softmax_elements, int softmax_elements_stride, int batch_count, const bool *… in dispatch_softmax_backward()
|
| H A D | SoftMax.cu | 1165 int batch_count = input.numel() / softmax_elements; in masked_softmax_cuda() local
1234 int64_t batch_count = grad.numel() / softmax_elements; in masked_softmax_backward_cuda() local
|
| /aosp_15_r20/external/mesa3d/src/intel/common/ |
| H A D | intel_measure.h | 124 unsigned frame, batch_count, event_index, primary_renderpass; member
157 unsigned frame, batch_count, event_count; member
|
| H A D | intel_measure.c | 645 unsigned int batch_count, in intel_measure_print_cpu_result()
|
| /aosp_15_r20/external/tensorflow/tensorflow/core/kernels/ |
| H A D | matmul_util.h | 39 size_t batch_count = 1; member
|
| H A D | depthwise_conv_ops_test.cc | 59 const int batch_count = 1; in Run() local
|
| /aosp_15_r20/external/mesa3d/src/gallium/drivers/d3d12/ |
| H A D | d3d12_residency.cpp | 187 unsigned batch_count = 0; in d3d12_process_batch_residency() local
|
| /aosp_15_r20/external/pytorch/aten/src/ATen/native/sparse/ |
| H A D | ValidateCompressedIndicesCommon.h | 291 const auto batch_count = indexCount(batch_dims); in _validate_compressed_sparse_indices_kernel() local
|
| H A D | SparseTensorMath.cpp | 940 int64_t batch_count = 1; in intersection_binary_op_sparse_dense_out() local
943 int64_t batch_count = 1; in intersection_binary_op_sparse_dense_out() local
|
| /aosp_15_r20/external/mesa3d/src/gallium/drivers/iris/ |
| H A D | iris_measure.c | 313 static unsigned batch_count = 0; in _iris_measure_snapshot() local
|
| /aosp_15_r20/external/vboot_reference/firmware/2lib/ |
| H A D | 2load_kernel.c | 554 const uint64_t batch_count = VB2_MIN(disk_count_half, batch_count_1); in try_minios_sector_region() local
|
| /aosp_15_r20/external/mesa3d/src/gallium/drivers/freedreno/ |
| H A D | freedreno_batch_cache.c | 252 uint32_t batch_count = 0; in fd_bc_flush_readers() local
|
| /aosp_15_r20/external/cronet/base/strings/ |
| H A D | string_util_impl_helpers.h | 179 constexpr int batch_count = 16; in DoIsStringASCII() local
|
| /aosp_15_r20/external/pytorch/aten/src/ATen/native/cuda/cutlass_extensions/gemm/kernel/ |
| H A D | fpA_intB_gemm.h | 110 int batch_count; member
|
| /aosp_15_r20/external/mesa3d/src/virtio/vulkan/ |
| H A D | vn_renderer.h | 107 uint32_t batch_count; member
|
| H A D | vn_renderer_vtest.c | 532 const uint32_t batch_count = submit->batch_count; in vtest_vcmd_submit_cmd2() local
|
| /aosp_15_r20/external/pytorch/aten/src/ATen/native/sparse/cuda/ |
| H A D | SparseCsrTensorMath.cu | 230 auto batch_count = in add_out_dense_sparse_compressed_cuda() local
|