| /external/zstd/contrib/linux-kernel/ |
| D | zstd_decompress_module.c | 40 zstd_dctx *zstd_init_dctx(void *workspace, size_t workspace_size) in zstd_init_dctx() argument
44 return ZSTD_initStaticDCtx(workspace, workspace_size); in zstd_init_dctx()
62 size_t workspace_size) in zstd_init_dstream() argument
67 return ZSTD_initStaticDStream(workspace, workspace_size); in zstd_init_dstream()
|
| D | linux_zstd.h | 173 zstd_cctx *zstd_init_cctx(void *workspace, size_t workspace_size);
212 zstd_dctx *zstd_init_dctx(void *workspace, size_t workspace_size);
289 unsigned long long pledged_src_size, void *workspace, size_t workspace_size);
380 size_t workspace_size);
|
| D | zstd_compress_module.c | 103 zstd_cctx *zstd_init_cctx(void *workspace, size_t workspace_size) in zstd_init_cctx() argument
107 return ZSTD_initStaticCCtx(workspace, workspace_size); in zstd_init_cctx()
132 unsigned long long pledged_src_size, void *workspace, size_t workspace_size) in zstd_init_cstream() argument
141 cstream = ZSTD_initStaticCStream(workspace, workspace_size); in zstd_init_cstream()
|
| /external/tensorflow/tensorflow/compiler/xla/service/gpu/ |
| D | cusolver_rewriter.cc | 68 int64 workspace_size; // Number of elements of size a_shape.element_type() in CreateCholesky() local
69 TF_ASSIGN_OR_RETURN(workspace_size, context->PotrfBufferSize( in CreateCholesky()
88 ShapeUtil::MakeShape(operand->shape().element_type(), {workspace_size}), in CreateCholesky()
|
| /external/tensorflow/tensorflow/core/kernels/image/ |
| D | non_max_suppression_op.cu.cc | 380 size_t workspace_size = 0; in CountIf() local
383 gpuprim::DeviceSelect::If(nullptr, workspace_size, in CountIf()
391 DataType::DT_INT8, TensorShape({(int64)workspace_size}), &workspace)); in CountIf()
398 workspace.flat<int8>().data(), workspace_size, dev_array, in CountIf()
|
| /external/tensorflow/tensorflow/stream_executor/rocm/ |
| D | rocm_dnn.cc | 510 profile_result.set_scratch_size(solution.workspace_size); in GetProfileResultFromConvSolution()
3221 << ", " << solution.workspace_size << ", " << solution.solution_id in GetMIOpenConvolveAlgorithmsImmediateMode()
3260 << ", " << solution.workspace_size << ", " << solution.solution_id in GetMIOpenConvolveAlgorithmsImmediateMode()
3298 << ", " << solution.workspace_size << ", " << solution.solution_id in GetMIOpenConvolveAlgorithmsImmediateMode()
4057 memory_used -= desc->workspace_size; in insert()
4070 desc->workspace_size = wsp_size; in insert()
4088 memory_used -= cache[x].workspace_size; in trim()
4113 size_t workspace_size = 0; in DoPoolForward() local
4118 pooling_desc.handle(), dest_desc.handle(), &workspace_size); in DoPoolForward()
4125 if (workspace_size != 0) { in DoPoolForward()
[all …]
|
| D | rocm_dnn.h | 49 size_t workspace_size; member
|
| /external/tensorflow/tensorflow/stream_executor/cuda/ |
| D | cuda_blas.cc | 3370 cublasLtMatmulAlgo_t algo, size_t workspace_size) in DoBlasInternalImpl() argument
3371 : index_(index), algo_(algo), workspace_size_(workspace_size) {} in DoBlasInternalImpl()
3375 size_t workspace_size() const override { return workspace_size_; } in DoBlasInternalImpl() function in stream_executor::gpu::__anone9de0a230311::CUDABlasLtMatmulAlgorithm
3568 if (cuda_algo.workspace_size()) { in DoBlasInternalImpl()
3570 &workspace, scratch_allocator, cuda_algo.workspace_size()); in DoBlasInternalImpl()
3576 << cuda_algo.workspace_size() << " bytes of workspace"; in DoBlasInternalImpl()
3638 cuda_algo.workspace_size(), cuda_stream); in DoBlasInternalImpl()
3655 if (remainder_algo.workspace_size()) { in DoBlasInternalImpl()
3657 &workspace, scratch_allocator, remainder_algo.workspace_size()); in DoBlasInternalImpl()
3663 << remainder_algo.workspace_size() in DoBlasInternalImpl()
[all …]
|
| /external/pcre/dist2/src/ |
| D | pcre2_intmodedep.h | 726 PCRE2_SIZE workspace_size; /* Size of workspace */ member
|
| D | pcre2_compile.c | 2566 end_nests = (nest_save *)(cb->start_workspace + cb->workspace_size); in parse_regex()
2573 ((cb->workspace_size * sizeof(PCRE2_UCHAR)) % sizeof(nest_save))); in parse_regex()
5397 if (code > cb->start_workspace + cb->workspace_size - in compile_branch()
5400 *errorcodeptr = (code >= cb->start_workspace + cb->workspace_size)? in compile_branch()
9781 cb.workspace_size = COMPILE_WORK_SIZE; in pcre2_compile()
|
| /external/tensorflow/tensorflow/stream_executor/ |
| D | blas.h | 210 virtual size_t workspace_size() const = 0;
|