Searched refs:kGemmCallTarget (Results 1 – 3 of 3) sorted by relevance
62 kGemmCallTarget); in HandleDot()82 m::Op(&existing_gemm).WithCustomCallTarget(kGemmCallTarget), in HandleMultiply()103 m::Op(&existing_gemm).WithCustomCallTarget(kGemmCallTarget), in HandleAdd()117 kGemmCallTarget); in HandleAdd()
80 extern const char* const kGemmCallTarget;
130 hlo.custom_call_target() == kGemmCallTarget; in IsCublasGemm()181 const char* const kGemmCallTarget = "__cublas$gemm"; variable