Searched defs:CustomCall (Results 1 – 6 of 6) sorted by relevance
46 struct CustomCall { in ConvertCustomCallOperations() struct47 func::CallOp call; in ConvertCustomCallOperations()48 func::FuncOp callee; in ConvertCustomCallOperations()49 llvm::StringRef target; in ConvertCustomCallOperations()50 bool direct; in ConvertCustomCallOperations()
409 inline ::testing::Matcher<const ::xla::HloInstruction*> CustomCall( in CustomCall() function422 inline ::testing::Matcher<const ::xla::HloInstruction*> CustomCall( in CustomCall() function427 inline ::testing::Matcher<const ::xla::HloInstruction*> CustomCall() { in CustomCall() function
937 ## CustomCall section in Operation Semantics
468 bool CustomCall(KernelContext* ctx, const char* target, void** args, in CustomCall() function
1965 XlaOp XlaBuilder::CustomCall( in CustomCall() function in xla::XlaBuilder2073 XlaOp XlaBuilder::CustomCall( in CustomCall() function in xla::XlaBuilder4420 XlaOp CustomCall( in CustomCall() function
1576 static bool CustomCall(runtime::KernelContext* ctx, void** args, void** attrs) { in CustomCall() function