| /external/zstd/contrib/linux-kernel/ |
| D | zstd_decompress_module.c | 47 zstd_dctx *zstd_init_dctx(void *workspace, size_t workspace_size) in zstd_init_dctx()
69 size_t workspace_size) in zstd_init_dstream()
|
| D | zstd_compress_module.c | 88 zstd_cctx *zstd_init_cctx(void *workspace, size_t workspace_size) in zstd_init_cctx()
111 unsigned long long pledged_src_size, void *workspace, size_t workspace_size) in zstd_init_cstream()
|
| /external/ComputeLibrary/src/cpu/operators/ |
| D | CpuPool2d.cpp | 77 const size_t workspace_size = pooling_wrapper->get_working_size(num_threads); in configure() local
|
| /external/tensorflow/tensorflow/core/kernels/ |
| D | conv_ops_gpu.h | 138 auto workspace_size = selected_runner->GetWorkspaceSize(); in AllocateScratchOrFallback() local
|
| /external/tensorflow/tensorflow/compiler/xla/service/gpu/ |
| D | cusolver_rewriter.cc | 71 int64_t workspace_size; // Number of elements of size a_shape.element_type() in CreateCholesky() local
|
| D | jitrt_custom_calls.h | 68 int64_t workspace_size; member
|
| D | gpu_conv_algorithm_picker.cc | 497 std::optional<size_t> workspace_size = in AutotuneOneConvRunner() local
|
| D | ir_emitter_unnested.cc | 1053 auto workspace_size = op.getBackendConfig().getWorkspaceSize(); in EmitConvolutionThunk() local
|
| /external/tensorflow/tensorflow/compiler/xla/stream_executor/cuda/ |
| D | cuda_blas_lt.h | 134 size_t workspace_size; member
|
| D | cuda_dnn.cc | 4020 size_t workspace_size; in Create() local
4217 size_t workspace_size, dnn::DataType input_type, in CudnnLegacyConvRunner()
4346 const cudnn_frontend::ExecutionPlan& plan, size_t workspace_size) { in ExecutionPlanToAlgorithmDesc()
4461 size_t workspace_size = plan_.getWorkspaceSize(); in operator ()() local
4528 auto workspace_size = static_cast<uint64_t>(plan.getWorkspaceSize()); in Create() local
4537 size_t workspace_size, in CudnnExecutionPlanRunner()
4887 size_t workspace_size; in Create() local
5017 size_t workspace_size, dnn::DataType input_type, in CudnnLegacyFusedConvRunner()
|
| /external/tensorflow/tensorflow/compiler/xla/stream_executor/ |
| D | dnn.proto | 134 google.protobuf.UInt64Value workspace_size = 6; field
|
| D | dnn.cc | 57 std::optional<uint64_t> workspace_size) { in AlgorithmDesc()
|
| D | dnn.h | 790 std::optional<uint64_t> workspace_size) { in AlgorithmDesc()
807 std::optional<uint64_t> workspace_size() const { in workspace_size() function
|
| /external/tensorflow/tensorflow/stream_executor/rocm/ |
| D | rocm_dnn.h | 49 size_t workspace_size; member
|
| D | rocm_dnn.cc | 2954 size_t workspace_size, dnn::ConvolutionKind kind, in RocmConvRunner()
3200 auto workspace_size = algorithm_desc.workspace_size(); in ConvolveRunnerFromDesc() local
4044 size_t workspace_size = 0; in DoPoolForward() local
|
| /external/ComputeLibrary/src/cpu/operators/internal/ |
| D | CpuGemmAssemblyDispatch.cpp | 379 const size_t workspace_size = _gemm_kernel_asm->get_working_size(); in configure() local
|
| /external/tensorflow/tensorflow/core/kernels/image/ |
| D | non_max_suppression_op.cu.cc | 276 size_t workspace_size = 0; in CountIf() local
|
| /external/pcre/src/ |
| D | pcre2_intmodedep.h | 730 PCRE2_SIZE workspace_size; /* Size of workspace */ member
|