Home
last modified time | relevance | path

Searched refs:kCusolverCholeskyCallTarget (Results 1 – 4 of 4) sorted by relevance

/external/tensorflow/tensorflow/compiler/xla/service/gpu/
Dir_emission_utils.h157 extern const char* const kCusolverCholeskyCallTarget;
Dcusolver_rewriter.cc93 call_shape, {operand}, kCusolverCholeskyCallTarget, {a_shape})); in CreateCholesky()
Dnvptx_compiler.cc196 user->custom_call_target() == kCusolverCholeskyCallTarget) { in CanShareBufferHint()
Dir_emission_utils.cc201 const char* const kCusolverCholeskyCallTarget = "__cusolver$cholesky"; variable
208 return target == kCusolverCholeskyCallTarget; in IsCustomCallToCusolver()