Searched defs:GetWorkspaceSize (Results 1 – 4 of 4) sorted by relevance
293 Status GetWorkspaceSize(const ConstCSRComponent<T>& a, in GetWorkspaceSize() function
599 Status GetWorkspaceSize(const ConstCSRComponent<T>& a, in CSRSparseSparseMatrixMatMul() function
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