Searched refs:GetWorkspaceSize (Results 1 – 9 of 9) sorted by relevance
138 auto workspace_size = selected_runner->GetWorkspaceSize(); in AllocateScratchOrFallback()146 if (selected_runner->GetWorkspaceSize() > 0) { in AllocateScratchOrFallback()
148 runner->GetWorkspaceSize())); in AutotuneFusedConv()327 runner->GetWorkspaceSize())); in AutotuneUnfusedConv()
119 csr_geam.GetWorkspaceSize(a_comp, b_comp, &thisWorkspaceSize)); in operator ()()293 Status GetWorkspaceSize(const ConstCSRComponent<T>& a, in GetWorkspaceSize() function
170 virtual Status GetWorkspaceSize(const ConstCSRComponent<T>& a,
434 ctx, csr_gemm.GetWorkspaceSize(a_comp, b_comp, &thisWorkspaceSize)); in Compute()599 Status GetWorkspaceSize(const ConstCSRComponent<T>& a, in CSRSparseSparseMatrixMatMul() function
844 result.set_scratch_bytes(runners[0]->GetWorkspaceSize()); in PickBestAlgorithmNoCacheRocm()866 scratch_allocator.AllocateBytes(runner->GetWorkspaceSize())); in PickBestAlgorithmNoCacheRocm()
885 virtual size_t GetWorkspaceSize() const = 0;
4074 size_t GetWorkspaceSize() const override { return workspace_size_; } in GetWorkspaceSize() function in stream_executor::gpu::CudnnLegacyConvRunner4444 size_t GetWorkspaceSize() const override { return workspace_size_; } in GetWorkspaceSize() function in stream_executor::gpu::CudnnExecutionPlanRunner4912 uint64_t GetWorkspaceSize() const override { return workspace_size_; } in GetWorkspaceSize() function in stream_executor::gpu::CudnnLegacyFusedConvRunner
2975 size_t GetWorkspaceSize() const override { return workspace_size_; } in GetWorkspaceSize() function in stream_executor::gpu::RocmConvRunner