Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/core/kernels/
Ddepthwise_conv_ops_test.cc59 const int batch_count = 1; in Run() local
/external/tensorflow/tensorflow/stream_executor/cuda/
Dcuda_fft.cc80 fft::Type type, int batch_count, ScratchAllocator *scratch_allocator) { in Initialize()
392 bool in_place_fft, int batch_count) { in CreateBatchedPlan()
419 bool in_place_fft, int batch_count, ScratchAllocator *scratch_allocator) { in CreateBatchedPlanWithScratchAllocator()
Dcuda_blas.cc2150 int ldc, int batch_count, ScratchAllocator *scratch_allocator) { in DoBlasGemmBatchedInternal()
2278 int ldc, int batch_count, ScratchAllocator *scratch_allocator) { in DoBlasGemmBatched()
2296 int batch_count, ScratchAllocator *scratch_allocator) { in DoBlasGemmBatched()
2312 int ldc, int batch_count, ScratchAllocator *scratch_allocator) { in DoBlasGemmBatched()
2330 int ldc, int batch_count, ScratchAllocator *scratch_allocator) { in DoBlasGemmBatched()
2348 int ldc, int batch_count, ScratchAllocator *scratch_allocator) { in DoBlasGemmBatched()
2363 int64 stride_c, int batch_count) { in DoBlasGemmStridedBatched()
2420 int batch_count) { in DoBlasGemmStridedBatched()
2433 int batch_count) { in DoBlasGemmStridedBatched()
2447 int64 stride_c, int batch_count) { in DoBlasGemmStridedBatched()
[all …]
Dcuda_dnn.cc4162 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()
476 bool in_place_fft, int batch_count) { in CreateBatchedPlan()
494 bool in_place_fft, int batch_count, ScratchAllocator *scratch_allocator) { in CreateBatchedPlanWithScratchAllocator()
Drocm_blas.cc1820 int batch_count, uint64_t batch_stride, ScratchAllocator *scratch_allocator, in AllocateStridedBuffer()
1885 int ldc, int batch_count, ScratchAllocator *scratch_allocator) { in DoBlasGemmBatchedInternal()
1977 int batch_count, ScratchAllocator *scratch_allocator) { in DoBlasGemmBatched()
1998 int batch_count, ScratchAllocator *scratch_allocator) { in DoBlasGemmBatched()
2015 int ldc, int batch_count, ScratchAllocator *scratch_allocator) { in DoBlasGemmBatched()
2034 int ldc, int batch_count, ScratchAllocator *scratch_allocator) { in DoBlasGemmBatched()
2048 int ldc, int batch_count, ScratchAllocator *scratch_allocator) { in DoBlasGemmBatched()
2344 int64 stride_c, int batch_count) { in DoBlasGemmStridedBatched()
2365 int batch_count) { in DoBlasGemmStridedBatched()
2378 int batch_count) { in DoBlasGemmStridedBatched()
[all …]
Drocm_dnn.cc3669 const int batch_count = output_dimensions.NodesPerFeatureMap(); in DoMatMul() local
/external/tensorflow/tensorflow/stream_executor/
Dstream.cc4525 int batch_count) { in ThenBlasGemmBatched()
4537 int batch_count, ScratchAllocator *scratch_allocator) { in ThenBlasGemmBatchedWithScratch()
4558 int batch_count) { in ThenBlasGemmBatched()
4569 int batch_count, ScratchAllocator *scratch_allocator) { in ThenBlasGemmBatchedWithScratch()
4590 int batch_count) { in ThenBlasGemmBatched()
4601 int batch_count, ScratchAllocator *scratch_allocator) { in ThenBlasGemmBatchedWithScratch()
4624 int batch_count) { in ThenBlasGemmBatched()
4637 int batch_count, ScratchAllocator *scratch_allocator) { in ThenBlasGemmBatchedWithScratch()
4663 int batch_count) { in ThenBlasGemmBatched()
4676 int batch_count, ScratchAllocator *scratch_allocator) { in ThenBlasGemmBatchedWithScratch()
[all …]
/external/tensorflow/tensorflow/compiler/xla/service/cpu/
Ddot_op_emitter.cc1005 int64 batch_count = lhs_array_reshaped.GetShape().dimensions(0); in EmitBatchDotOperation() local
/external/blktrace/iowatcher/
Dmain.c1104 int batch_count; in plot_io_movie() local