Searched defs:XlaCustomCall (Results 1 – 2 of 2) sorted by relevance
1493 struct XlaCustomCall { struct1494 using Stream = se::gpu::GpuStreamHandle;1500 static XlaCustomCall Handler() { return XlaCustomCall(); } in Handler()
56081 func XlaCustomCall(scope *Scope, args []tf.Output, target_name string, backend_config string, dtype… func