Searched defs:CustomCall (Results 1 – 5 of 5) sorted by relevance
327 inline ::testing::Matcher<const ::xla::HloInstruction*> CustomCall( in CustomCall() function339 inline ::testing::Matcher<const ::xla::HloInstruction*> CustomCall( in CustomCall() function344 inline ::testing::Matcher<const ::xla::HloInstruction*> CustomCall() { in CustomCall() function
816 ## CustomCall section in Operation Semantics
618 LocalOp ComputationBuilder::CustomCall( in CustomCall() function in xla::swig::ComputationBuilder
1418 def CustomCall(self, member in ComputationBuilder
1493 XlaOp XlaBuilder::CustomCall( in CustomCall() function in xla::XlaBuilder3081 XlaOp CustomCall(XlaBuilder* builder, const string& call_target_name, in CustomCall() function