| /aosp_15_r20/external/zstd/contrib/linux-kernel/ |
| H A 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()
|
| H A 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()
|
| /aosp_15_r20/external/pytorch/aten/src/ATen/native/cudnn/ |
| H A D | LossCTC.cpp | 216 size_t workspace_size; in _cudnn_ctc_loss() local
305 size_t workspace_size; in _cudnn_ctc_loss_tensor() local
|
| H A D | BatchNorm.cpp | 195 size_t workspace_size; in cudnn_batch_norm() local
346 size_t workspace_size; in cudnn_batch_norm_backward() local
|
| H A D | MHA.cpp | 601 auto workspace_size = mha_graph->get_workspace_size(); in run_cudnn_SDP_fprop() local
694 auto workspace_size = mha_graph->get_workspace_size(); in run_cudnn_SDP_bprop() local
|
| H A D | RNN.cpp | 1518 size_t workspace_size; in _cudnn_rnn() local
1813 size_t workspace_size; in _cudnn_rnn_backward_input() local
2001 size_t workspace_size; in _cudnn_rnn_backward_weight() local
|
| /aosp_15_r20/external/pytorch/aten/src/ATen/cuda/tunable/ |
| H A D | GemmHipblaslt.h | 341 size_t workspace_size = 32*1024; // going with 32MB in GetHipblasltWorkspaceSize() local
484 size_t workspace_size = GetHipblasltWorkspaceSize(); in Call() local
|
| /aosp_15_r20/external/pytorch/aten/src/ATen/native/xnnpack/ |
| H A D | AveragePooling.cpp | 47 size_t workspace_size = 0; in global_average_pool() local
|
| H A D | Convolution.cpp | 359 size_t workspace_size = SIZE_MAX; in run() local
|
| /aosp_15_r20/external/ComputeLibrary/src/cpu/operators/ |
| H A D | CpuPool2d.cpp | 77 const size_t workspace_size = pooling_wrapper->get_working_size(num_threads); in configure() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/core/kernels/ |
| H A D | conv_ops_gpu.h | 138 auto workspace_size = selected_runner->GetWorkspaceSize(); in AllocateScratchOrFallback() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/gpu/ |
| H A D | cusolver_rewriter.cc | 71 int64_t workspace_size; // Number of elements of size a_shape.element_type() in CreateCholesky() local
|
| H A D | jitrt_custom_calls.h | 68 int64_t workspace_size; member
|
| /aosp_15_r20/out/soong/.intermediates/external/tensorflow/tensorflow_core_proto_cpp_lite/android_arm64_armv8-2a_cortex-a55_static_apex33/gen/proto/tensorflow/compiler/xla/stream_executor/ |
| D | dnn.pb.h | 1448 inline const ::PROTOBUF_NAMESPACE_ID::UInt64Value& AlgorithmProto::workspace_size() const { in workspace_size() function
1453 ::PROTOBUF_NAMESPACE_ID::UInt64Value* workspace_size) { in unsafe_arena_set_allocated_workspace_size()
1495 …void AlgorithmProto::set_allocated_workspace_size(::PROTOBUF_NAMESPACE_ID::UInt64Value* workspace_… in set_allocated_workspace_size()
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/stream_executor/cuda/ |
| H A D | cuda_blas_lt.h | 134 size_t workspace_size; member
|
| H A 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()
|
| /aosp_15_r20/external/pytorch/aten/src/ATen/native/quantized/cpu/ |
| H A D | XnnpackUtils.h | 199 size_t workspace_size = SIZE_MAX; variable
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/stream_executor/ |
| H A D | dnn.proto | 134 google.protobuf.UInt64Value workspace_size = 6; field
|
| /aosp_15_r20/external/tensorflow/tensorflow/stream_executor/rocm/ |
| H A D | rocm_dnn.h | 49 size_t workspace_size; member
|
| /aosp_15_r20/external/pytorch/aten/src/ATen/native/quantized/cudnn/ |
| H A D | BinaryOps.cpp | 138 auto workspace_size = 0; in add() local
|
| H A D | Conv.cpp | 118 auto workspace_size = plan_desc.getWorkspaceSize(); in apply_impl_helper() local
|
| H A D | Linear.cpp | 146 auto workspace_size = plan_desc.getWorkspaceSize(); in apply_impl_helper() local
|
| /aosp_15_r20/external/pytorch/aten/src/ATen/native/sparse/cuda/ |
| H A D | cuSPARSELtOps.cpp | 286 size_t workspace_size; in _cslt_sparse_mm_impl() local
|
| /aosp_15_r20/external/pytorch/aten/src/ATen/native/cuda/ |
| H A D | CuFFTPlanCache.h | 347 int64_t workspace_size() const { return ws_size; } in workspace_size() function
|
| /aosp_15_r20/external/pytorch/aten/src/ATen/native/miopen/ |
| H A D | RNN_miopen.cpp | 513 size_t workspace_size; in miopen_rnn() local
634 size_t workspace_size; in miopen_rnn_backward_input() local
|