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.cc57 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()
393 bool in_place_fft, int batch_count) { in CreateBatchedPlan()
420 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.cc3620 const int batch_count = output_dimensions.NodesPerFeatureMap(); in DoMatMul() local
/external/tensorflow/tensorflow/stream_executor/rocm/
Drocm_fft.cc149 fft::Type type, int batch_count, ScratchAllocator *scratch_allocator) { in Initialize()
466 bool in_place_fft, int batch_count) { in CreateBatchedPlan()
484 bool in_place_fft, int batch_count, ScratchAllocator *scratch_allocator) { in CreateBatchedPlanWithScratchAllocator()
Drocm_blas.cc1814 int ldc, int batch_count, ScratchAllocator *scratch_allocator) { in DoBlasGemmBatchedInternal()
1910 int batch_count, ScratchAllocator *scratch_allocator) { in DoBlasGemmBatched()
1931 int batch_count, ScratchAllocator *scratch_allocator) { in DoBlasGemmBatched()
1948 int ldc, int batch_count, ScratchAllocator *scratch_allocator) { in DoBlasGemmBatched()
1967 int ldc, int batch_count, ScratchAllocator *scratch_allocator) { in DoBlasGemmBatched()
1981 int ldc, int batch_count, ScratchAllocator *scratch_allocator) { in DoBlasGemmBatched()
2277 int64 stride_c, int batch_count) { in DoBlasGemmStridedBatched()
2288 int batch_count) { in DoBlasGemmStridedBatched()
2299 int batch_count) { in DoBlasGemmStridedBatched()
2311 int64 stride_c, int batch_count) { in DoBlasGemmStridedBatched()
[all …]
Drocm_dnn.cc3380 const int batch_count = output_dimensions.NodesPerFeatureMap(); in DoMatMul() local
/external/tensorflow/tensorflow/stream_executor/
Dstream.cc4383 int batch_count) { in ThenBlasGemmBatched()
4395 int batch_count, ScratchAllocator *scratch_allocator) { in ThenBlasGemmBatchedWithScratch()
4416 int batch_count) { in ThenBlasGemmBatched()
4427 int batch_count, ScratchAllocator *scratch_allocator) { in ThenBlasGemmBatchedWithScratch()
4448 int batch_count) { in ThenBlasGemmBatched()
4459 int batch_count, ScratchAllocator *scratch_allocator) { in ThenBlasGemmBatchedWithScratch()
4482 int batch_count) { in ThenBlasGemmBatched()
4495 int batch_count, ScratchAllocator *scratch_allocator) { in ThenBlasGemmBatchedWithScratch()
4521 int batch_count) { in ThenBlasGemmBatched()
4534 int batch_count, ScratchAllocator *scratch_allocator) { in ThenBlasGemmBatchedWithScratch()
[all …]
/external/tensorflow/tensorflow/compiler/xla/service/cpu/
Ddot_op_emitter.cc953 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