Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/contrib/lite/toco/graph_transformations/
Dunroll_batch_matmul.cc75 int batch_count = input_array_a.shape().dims(0); in Run() local
/external/tensorflow/tensorflow/stream_executor/cuda/
Dcuda_fft.cc115 fft::Type type, int batch_count, ScratchAllocator *scratch_allocator) { in Initialize()
420 bool in_place_fft, int batch_count) { in CreateBatchedPlan()
438 bool in_place_fft, int batch_count, ScratchAllocator *scratch_allocator) { in CreateBatchedPlanWithScratchAllocator()
Dcuda_blas.cc2319 int ldc, int batch_count, ScratchAllocator *scratch_allocator) { in DoBlasGemmBatchedInternal()
2399 int batch_count, ScratchAllocator *scratch_allocator) { in DoBlasGemmBatched()
2415 int ldc, int batch_count, ScratchAllocator *scratch_allocator) { in DoBlasGemmBatched()
2433 int ldc, int batch_count, ScratchAllocator *scratch_allocator) { in DoBlasGemmBatched()
2451 int ldc, int batch_count, ScratchAllocator *scratch_allocator) { in DoBlasGemmBatched()
Dcuda_dnn.cc3787 const int batch_count = output_dimensions.NodesPerFeatureMap(); in DoMatMul() local
/external/drm_gralloc/
Dgralloc_drm_intel.c90 batch_count(struct intel_info *info) in batch_count() function
/external/tensorflow/tensorflow/stream_executor/
Dstream.cc4367 int batch_count) { in ThenBlasGemmBatched()
4378 int batch_count, ScratchAllocator *scratch_allocator) { in ThenBlasGemmBatchedWithScratch()
4399 int batch_count) { in ThenBlasGemmBatched()
4410 int batch_count, ScratchAllocator *scratch_allocator) { in ThenBlasGemmBatchedWithScratch()
4433 int batch_count) { in ThenBlasGemmBatched()
4446 int batch_count, ScratchAllocator *scratch_allocator) { in ThenBlasGemmBatchedWithScratch()
4472 int batch_count) { in ThenBlasGemmBatched()
4485 int batch_count, ScratchAllocator *scratch_allocator) { in ThenBlasGemmBatchedWithScratch()