Searched refs:CustomCallWithComputation (Results 1 – 5 of 5) sorted by relevance
188 TEST_F(XlaBuilderTest, CustomCallWithComputation) { in TEST_F() argument202 auto custom_call = CustomCallWithComputation( in TEST_F()
237 auto approx_topk = CustomCallWithComputation( in ApproxTopK()
1267 friend XlaOp CustomCallWithComputation(2142 XlaOp CustomCallWithComputation(
4435 XlaOp CustomCallWithComputation( in CustomCallWithComputation() function
1120 value_map[result] = xla::CustomCallWithComputation( in ExportXlaOp()