Home
last modified time | relevance | path

Searched defs:kCustomCallTarget (Results 1 – 1 of 1) sorted by relevance

/aosp_15_r20/external/tensorflow/tensorflow/compiler/mlir/tfrt/transforms/lmhlo_to_gpu/
H A Dlmhlo_to_jitrt.cc287 static constexpr const char kCustomCallTarget[] = "xla.gpu.memset"; member in tensorflow::MemsetOpLowering
325 static constexpr const char kCustomCallTarget[] = "xla.gpu.func.launch"; member in tensorflow::LaunchFuncOpLowering
404 static constexpr const char kCustomCallTarget[] = "xla.gpu.gemm"; member in tensorflow::GemmOpLowering
464 static constexpr const char kCustomCallTarget[] = "xla.gpu.cublas.lt.matmul"; member in tensorflow::CublasLtMatmulOpLowering
821 static constexpr const char kCustomCallTarget[] = "xla.gpu.custom_call"; member in tensorflow::CustomCallOpLowering
977 static constexpr const char kCustomCallTarget[] = "xla.gpu.fft"; member in tensorflow::FftOpLowering
1024 static constexpr const char kCustomCallTarget[] = "xla.gpu.cholesky"; member in tensorflow::CholeskyOpLowering
1428 static constexpr const char kCustomCallTarget[] = "xla.gpu.all_reduce_done"; member in tensorflow::AllReduceDoneOpLowering