| /external/tensorflow/tensorflow/stream_executor/rocm/ |
| D | rocm_blas.cc | 646 int batch_count, blas::ComputationType computation_type, in DoBlasGemmStridedBatchedWithAlgorithm()
666 int batch_count, uint64_t batch_stride, in ReorganizeMemory()
712 int batch_count, uint64_t batch_stride, ScratchAllocator *scratch_allocator, in AllocateStridedBuffer()
771 int ldc, int batch_count, ScratchAllocator *scratch_allocator) { in DoBlasGemmBatchedInternal()
865 int batch_count, ScratchAllocator *scratch_allocator) { in DoBlasGemmBatched()
887 int batch_count, ScratchAllocator *scratch_allocator) { in DoBlasGemmBatched()
905 int ldc, int batch_count, ScratchAllocator *scratch_allocator) { in DoBlasGemmBatched()
925 int ldc, int batch_count, ScratchAllocator *scratch_allocator) { in DoBlasGemmBatched()
945 int ldc, int batch_count, ScratchAllocator *scratch_allocator) { in DoBlasGemmBatched()
1014 int batch_count) { in DoBlasTrsmBatched()
[all …]
|
| D | rocm_fft.cc | 161 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()
|
| D | rocm_dnn.cc | 3923 const int batch_count = output_dimensions.NodesPerFeatureMap(); in DoMatMul() local
|
| /external/tensorflow/tensorflow/core/kernels/ |
| D | matmul_util.h | 39 size_t batch_count = 1; member
|
| D | depthwise_conv_ops_test.cc | 59 const int batch_count = 1; in Run() local
|
| /external/tensorflow/tensorflow/compiler/xla/stream_executor/cuda/ |
| D | cuda_fft.cc | 100 fft::Type type, int batch_count, ScratchAllocator *scratch_allocator) { in Initialize()
420 bool in_place_fft, int batch_count) { in CreateBatchedPlan()
448 bool in_place_fft, int batch_count, ScratchAllocator *scratch_allocator) { in CreateBatchedPlanWithScratchAllocator()
|
| D | cuda_blas.cc | 1050 int batch_count, blas::ComputationType computation_type, in DoBlasInternalImpl()
1192 const DeviceMemorySlice<T> &c_ptrs_to_wrappers, int ldc, int batch_count, in DoBlasInternalImpl()
1329 const DeviceMemorySlice<Eigen::half> &c_array, int ldc, int batch_count, in DoBlasInternalImpl()
1346 const DeviceMemorySlice<float> &c_array, int ldc, int batch_count, in DoBlasInternalImpl()
1362 const DeviceMemorySlice<double> &c_array, int ldc, int batch_count, in DoBlasInternalImpl()
1380 int batch_count, ScratchAllocator *scratch_allocator) { in DoBlasInternalImpl()
1397 int batch_count, ScratchAllocator *scratch_allocator) { in DoBlasInternalImpl()
1412 DeviceMemoryBase *c, int ldc, int64_t stride_c, int batch_count) { in DoBlasInternalImpl()
1603 int batch_count) { in DoBlasInternalImpl()
1616 int batch_count) { in DoBlasInternalImpl()
[all …]
|
| D | cuda_dnn.cc | 6052 const int batch_count = output_dimensions.NodesPerFeatureMap(); in DoMatMul() local
|
| /external/tensorflow/tensorflow/compiler/xla/stream_executor/ |
| D | stream.cc | 1627 int batch_count) { in ThenBlasTrsmBatched()
1645 int batch_count) { in ThenBlasTrsmBatched()
1662 DeviceMemory<std::complex<float> *> *bs, int ldb, int batch_count) { in ThenBlasTrsmBatched()
1680 DeviceMemory<std::complex<double> *> *bs, int ldb, int batch_count) { in ThenBlasTrsmBatched()
1702 int ldc, int batch_count) { in ThenBlasGemmBatched()
1716 int ldc, int batch_count, ScratchAllocator *scratch_allocator) { in ThenBlasGemmBatchedWithScratch()
1742 int ldc, int batch_count) { in ThenBlasGemmBatched()
1755 int ldc, int batch_count, ScratchAllocator *scratch_allocator) { in ThenBlasGemmBatchedWithScratch()
1779 int ldc, int batch_count) { in ThenBlasGemmBatched()
1792 int ldc, int batch_count, ScratchAllocator *scratch_allocator) { in ThenBlasGemmBatchedWithScratch()
[all …]
|
| D | stream.h | 923 int64_t stride_c, int batch_count, blas::ComputationType computation_type, in ThenBlasGemmStridedBatchedWithAlgorithm()
1038 int64_t stride_c, int batch_count) { in ThenBlasGemmStridedBatched()
|
| /external/cronet/base/strings/ |
| D | string_util_impl_helpers.h | 178 constexpr int batch_count = 16; in DoIsStringASCII() 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 | 1418 int64_t batch_count = lhs_array_reshaped.GetShape().dimensions(0); in EmitBatchDotOperation() local
|
| /external/virglrenderer/vtest/ |
| D | vtest_renderer.c | 2084 uint32_t batch_count; in vtest_submit_cmd2() local
|
| /external/libbpf/.github/actions/build-selftests/ |
| D | vmlinux.h | 21257 unsigned int batch_count; member
|
| /external/bcc/libbpf-tools/powerpc/ |
| D | vmlinux_510.h | 19863 unsigned int batch_count; member
|
| D | vmlinux.h | 19863 unsigned int batch_count; member
|
| /external/bcc/libbpf-tools/arm64/ |
| D | vmlinux.h | 34977 unsigned int batch_count; member
|
| D | vmlinux_510.h | 34977 unsigned int batch_count; member
|
| /external/bcc/libbpf-tools/x86/ |
| D | vmlinux_505.h | 21377 unsigned int batch_count; member
|
| D | vmlinux.h | 21377 unsigned int batch_count; member
|