Searched refs:CusolverContext (Results 1 – 5 of 5) sorted by relevance
37 class CusolverContext {41 static StatusOr<CusolverContext> Create(se::Stream* stream);42 CusolverContext() = default;43 ~CusolverContext();45 CusolverContext(const CusolverContext&) = delete;46 CusolverContext(CusolverContext&&);47 CusolverContext& operator=(const CusolverContext&) = delete;48 CusolverContext& operator=(CusolverContext&&);68 CusolverContext(se::Stream* stream, cusolverDnHandle_t handle);80 Status CusolverContext::Potrf<T>( \[all …]
89 StatusOr<CusolverContext> CusolverContext::Create(se::Stream* stream) { in Create()92 CusolverContext context(stream, handle); in Create()106 CusolverContext::CusolverContext(se::Stream* stream, cusolverDnHandle_t handle) in CusolverContext() function in xla::gpu::CusolverContext109 CusolverContext::CusolverContext(CusolverContext&& other) { in CusolverContext() function in xla::gpu::CusolverContext116 CusolverContext& CusolverContext::operator=(CusolverContext&& other) { in operator =()122 CusolverContext::~CusolverContext() { in ~CusolverContext()139 StatusOr<int64> CusolverContext::PotrfBufferSize(PrimitiveType type, in PotrfBufferSize()173 Status CusolverContext::Potrf<T>( \
49 StatusOr<HloInstruction*> CreateCholesky(CusolverContext* context, in CreateCholesky()132 StatusOr<bool> RunOnInstruction(CusolverContext* context, in RunOnInstruction()165 TF_ASSIGN_OR_RETURN(CusolverContext context, in RunOnComputation()166 CusolverContext::Create(/*stream=*/nullptr)); in RunOnComputation()
59 CusolverContext* context; in ExecuteOnStream()62 auto result = contexts_.emplace(params.stream, CusolverContext()); in ExecuteOnStream()65 CusolverContext::Create(params.stream)); in ExecuteOnStream()
68 absl::flat_hash_map<se::Stream*, CusolverContext> contexts_