Searched defs:CustomCall (Results 1 – 4 of 4) sorted by relevance
355 inline ::testing::Matcher<const ::xla::HloInstruction*> CustomCall( in CustomCall() function367 inline ::testing::Matcher<const ::xla::HloInstruction*> CustomCall( in CustomCall() function372 inline ::testing::Matcher<const ::xla::HloInstruction*> CustomCall() { in CustomCall() function
852 ## CustomCall section in Operation Semantics
1195 def CustomCall(self, member in ComputationBuilder
1549 XlaOp XlaBuilder::CustomCall( in CustomCall() function in xla::XlaBuilder3146 XlaOp CustomCall(XlaBuilder* builder, const string& call_target_name, in CustomCall() function