Home
last modified time | relevance | path

Searched refs:kTriangularSolveCallTarget (Results 1 – 5 of 5) sorted by relevance

/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/gpu/
H A Dcublas_cudnn.h69 extern const char* const kTriangularSolveCallTarget;
H A Dcublas_cudnn.cc33 const char* const kTriangularSolveCallTarget = "__cublas$triangularSolve"; variable
H A Dtriangular_solve_rewriter.cc55 new_shape, instr->operands(), kTriangularSolveCallTarget)); in Run()
H A Djitrt_custom_calls.cc1511 if (call_target_name == kTriangularSolveCallTarget) in operator ()()
H A Dir_emitter_unnested.cc5316 if (call.getCallTargetName() == kTriangularSolveCallTarget) { in EmitOp()