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.h145 extern const char* const kCusolverCholeskyCallTarget;
Dir_emission_utils.cc145 const char* const kCusolverCholeskyCallTarget = "__cusolver$cholesky"; variable
152 return target == kCusolverCholeskyCallTarget; in IsCustomCallToCusolver()
Dcusolver_rewriter.cc124 call_shape, {operand}, kCusolverCholeskyCallTarget, {a_shape})); in CreateCholesky()
Dir_emitter_unnested.cc484 if (custom_call->custom_call_target() == kCusolverCholeskyCallTarget) { in HandleCustomCall()