| /external/tensorflow/tensorflow/stream_executor/rocm/ |
| D | rocm_fft.cc | 161 fft::Type type, int batch_count, ScratchAllocator *scratch_allocator) { in Initialize()
465 bool in_place_fft, int batch_count) { in CreateBatchedPlan()
483 bool in_place_fft, int batch_count, ScratchAllocator *scratch_allocator) { in CreateBatchedPlanWithScratchAllocator()
|
| D | rocm_blas.cc | 1940 int batch_count, uint64_t batch_stride, in ReorganizeMemory()
1986 int batch_count, uint64_t batch_stride, ScratchAllocator *scratch_allocator, in AllocateStridedBuffer()
2045 int ldc, int batch_count, ScratchAllocator *scratch_allocator) { in DoBlasGemmBatchedInternal()
2139 int batch_count, ScratchAllocator *scratch_allocator) { in DoBlasGemmBatched()
2161 int batch_count, ScratchAllocator *scratch_allocator) { in DoBlasGemmBatched()
2179 int ldc, int batch_count, ScratchAllocator *scratch_allocator) { in DoBlasGemmBatched()
2199 int ldc, int batch_count, ScratchAllocator *scratch_allocator) { in DoBlasGemmBatched()
2219 int ldc, int batch_count, ScratchAllocator *scratch_allocator) { in DoBlasGemmBatched()
2566 int64 stride_c, int batch_count) { in DoBlasGemmStridedBatched()
2588 int batch_count) { in DoBlasGemmStridedBatched()
[all …]
|
| D | rocm_dnn.cc | 3906 const int batch_count = output_dimensions.NodesPerFeatureMap(); in DoMatMul() local
|
| /external/tensorflow/tensorflow/stream_executor/cuda/ |
| D | cuda_fft.cc | 81 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()
|
| D | cuda_blas.cc | 2370 int ldc, int batch_count, ScratchAllocator *scratch_allocator) { in DoBlasInternalImpl()
2519 int ldc, int batch_count, ScratchAllocator *scratch_allocator) { in DoBlasInternalImpl()
2537 int batch_count, ScratchAllocator *scratch_allocator) { in DoBlasInternalImpl()
2553 int ldc, int batch_count, ScratchAllocator *scratch_allocator) { in DoBlasInternalImpl()
2571 int ldc, int batch_count, ScratchAllocator *scratch_allocator) { in DoBlasInternalImpl()
2589 int ldc, int batch_count, ScratchAllocator *scratch_allocator) { in DoBlasInternalImpl()
2604 int64 stride_c, int batch_count) { in DoBlasInternalImpl()
2658 int batch_count) { in DoBlasInternalImpl()
2677 int batch_count) { in DoBlasInternalImpl()
2691 int64 stride_c, int batch_count) { in DoBlasInternalImpl()
[all …]
|
| D | cuda_dnn.cc | 4153 const int batch_count = output_dimensions.NodesPerFeatureMap(); in DoMatMul() local
|
| /external/tensorflow/tensorflow/core/kernels/ |
| D | depthwise_conv_ops_test.cc | 59 const int batch_count = 1; in Run() local
|
| /external/tensorflow/tensorflow/stream_executor/ |
| D | blas.h | 228 int batch_count = 1; member
|
| D | stream.cc | 4046 int batch_count) { in ThenBlasGemmBatched()
4058 int batch_count, ScratchAllocator *scratch_allocator) { in ThenBlasGemmBatchedWithScratch()
4079 int batch_count) { in ThenBlasGemmBatched()
4090 int batch_count, ScratchAllocator *scratch_allocator) { in ThenBlasGemmBatchedWithScratch()
4111 int batch_count) { in ThenBlasGemmBatched()
4122 int batch_count, ScratchAllocator *scratch_allocator) { in ThenBlasGemmBatchedWithScratch()
4145 int batch_count) { in ThenBlasGemmBatched()
4158 int batch_count, ScratchAllocator *scratch_allocator) { in ThenBlasGemmBatchedWithScratch()
4184 int batch_count) { in ThenBlasGemmBatched()
4197 int batch_count, ScratchAllocator *scratch_allocator) { in ThenBlasGemmBatchedWithScratch()
[all …]
|
| /external/llvm-project/compiler-rt/lib/sanitizer_common/ |
| D | sanitizer_quarantine.h | 277 uptr batch_count = 0; in PrintStats() local
|
| /external/blktrace/iowatcher/ |
| D | main.c | 1104 int batch_count; in plot_io_movie() local
|
| /external/tensorflow/tensorflow/compiler/xla/service/cpu/ |
| D | dot_op_emitter.cc | 1149 int64 batch_count = lhs_array_reshaped.GetShape().dimensions(0); in EmitBatchDotOperation() local
|