Home
last modified time | relevance | path

Searched defs:batch_count (Results 1 – 19 of 19) sorted by relevance

/external/tensorflow/tensorflow/stream_executor/cuda/
Dcuda_fft.cc81 fft::Type type, int batch_count, ScratchAllocator *scratch_allocator) { in Initialize()
396 bool in_place_fft, int batch_count) { in CreateBatchedPlan()
423 bool in_place_fft, int batch_count, ScratchAllocator *scratch_allocator) { in CreateBatchedPlanWithScratchAllocator()
Dcuda_blas.cc2182 int batch_count, blas::ComputationType computation_type, in DoBlasInternalImpl()
2314 int ldc, int batch_count, ScratchAllocator *scratch_allocator) { in DoBlasInternalImpl()
2450 int ldc, int batch_count, ScratchAllocator *scratch_allocator) { in DoBlasInternalImpl()
2468 int batch_count, ScratchAllocator *scratch_allocator) { in DoBlasInternalImpl()
2484 int ldc, int batch_count, ScratchAllocator *scratch_allocator) { in DoBlasInternalImpl()
2502 int ldc, int batch_count, ScratchAllocator *scratch_allocator) { in DoBlasInternalImpl()
2520 int ldc, int batch_count, ScratchAllocator *scratch_allocator) { in DoBlasInternalImpl()
2535 DeviceMemoryBase *c, int ldc, int64_t stride_c, int batch_count) { in DoBlasInternalImpl()
3160 int64_t stride, int batch_count) { in DoBlasInternalImpl()
Dcuda_dnn.cc5361 const int batch_count = output_dimensions.NodesPerFeatureMap(); in DoMatMul() local
/external/tensorflow/tensorflow/stream_executor/rocm/
Drocm_fft.cc161 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()
Drocm_blas.cc1605 int batch_count, blas::ComputationType computation_type, in DoBlasGemmStridedBatchedWithAlgorithm()
1625 int batch_count, uint64_t batch_stride, in ReorganizeMemory()
1671 int batch_count, uint64_t batch_stride, ScratchAllocator *scratch_allocator, in AllocateStridedBuffer()
1730 int ldc, int batch_count, ScratchAllocator *scratch_allocator) { in DoBlasGemmBatchedInternal()
1824 int batch_count, ScratchAllocator *scratch_allocator) { in DoBlasGemmBatched()
1846 int batch_count, ScratchAllocator *scratch_allocator) { in DoBlasGemmBatched()
1864 int ldc, int batch_count, ScratchAllocator *scratch_allocator) { in DoBlasGemmBatched()
1884 int ldc, int batch_count, ScratchAllocator *scratch_allocator) { in DoBlasGemmBatched()
1904 int ldc, int batch_count, ScratchAllocator *scratch_allocator) { in DoBlasGemmBatched()
2251 DeviceMemoryBase *c, int ldc, int64 stride_c, int batch_count) { in DoBlasGemmStridedBatched()
Drocm_dnn.cc3797 const int batch_count = output_dimensions.NodesPerFeatureMap(); in DoMatMul() local
/external/tensorflow/tensorflow/core/kernels/
Ddepthwise_conv_ops_test.cc59 const int batch_count = 1; in Run() local
/external/tensorflow/tensorflow/stream_executor/
Dstream.h1482 int64_t stride_c, int batch_count, blas::ComputationType computation_type, in ThenBlasGemmStridedBatchedWithAlgorithm()
1595 int64_t stride_c, int batch_count) { in ThenBlasGemmStridedBatched()
Dstream.cc3598 int batch_count) { in ThenBlasGemmBatched()
3610 int batch_count, ScratchAllocator *scratch_allocator) { in ThenBlasGemmBatchedWithScratch()
3631 int batch_count) { in ThenBlasGemmBatched()
3642 int batch_count, ScratchAllocator *scratch_allocator) { in ThenBlasGemmBatchedWithScratch()
3663 int batch_count) { in ThenBlasGemmBatched()
3674 int batch_count, ScratchAllocator *scratch_allocator) { in ThenBlasGemmBatchedWithScratch()
3697 int batch_count) { in ThenBlasGemmBatched()
3710 int batch_count, ScratchAllocator *scratch_allocator) { in ThenBlasGemmBatchedWithScratch()
3736 int batch_count) { in ThenBlasGemmBatched()
3749 int batch_count, ScratchAllocator *scratch_allocator) { in ThenBlasGemmBatchedWithScratch()
/external/blktrace/iowatcher/
Dmain.c1104 int batch_count; in plot_io_movie() local
/external/tensorflow/tensorflow/compiler/xla/service/cpu/
Ddot_op_emitter.cc1168 int64_t batch_count = lhs_array_reshaped.GetShape().dimensions(0); in EmitBatchDotOperation() local
/external/virglrenderer/vtest/
Dvtest_renderer.c2078 uint32_t batch_count; in vtest_submit_cmd2() local
/external/bcc/libbpf-tools/powerpc/
Dvmlinux_510.h19863 unsigned int batch_count; member
Dvmlinux.h19863 unsigned int batch_count; member
/external/libbpf/.github/actions/build-selftests/
Dvmlinux.h21257 unsigned int batch_count; member
/external/bcc/libbpf-tools/arm64/
Dvmlinux.h34977 unsigned int batch_count; member
Dvmlinux_510.h34977 unsigned int batch_count; member
/external/bcc/libbpf-tools/x86/
Dvmlinux.h21377 unsigned int batch_count; member
Dvmlinux_505.h21377 unsigned int batch_count; member