Home
last modified time | relevance | path

Searched refs:PotrfBufferSize (Results 1 – 3 of 3) sorted by relevance

/external/tensorflow/tensorflow/compiler/xla/service/gpu/
Dcusolver_context.h64 StatusOr<int64> PotrfBufferSize(PrimitiveType type, se::blas::UpperLower uplo,
109 StatusOr<int64> PotrfBufferSize(PrimitiveType type, se::blas::UpperLower uplo,
Dcusolver_context.cc139 StatusOr<int64> CusolverContext::PotrfBufferSize(PrimitiveType type, in PotrfBufferSize() function in xla::gpu::CusolverContext
Dcusolver_rewriter.cc69 TF_ASSIGN_OR_RETURN(workspace_size, context->PotrfBufferSize( in CreateCholesky()