Searched defs:CustomCallTarget (Results 1 – 2 of 2) sorted by relevance
183 static StringRef CustomCallTarget(InfeedOp) { return "xla.gpu.infeed"; } in CustomCallTarget() function in tensorflow::IoFeedOpLowering184 static StringRef CustomCallTarget(OutfeedOp) { return "xla.gpu.outfeed"; } in CustomCallTarget() function in tensorflow::IoFeedOpLowering549 static StringRef CustomCallTarget(ConvForwardOp) { in CustomCallTarget() function in tensorflow::ConvOpLowering552 static StringRef CustomCallTarget(ConvForwardFusedOp) { in CustomCallTarget() function in tensorflow::ConvOpLowering555 static StringRef CustomCallTarget(ConvForwardFusedSideInputOp) { in CustomCallTarget() function in tensorflow::ConvOpLowering558 static StringRef CustomCallTarget(ConvBackwardFilterOp) { in CustomCallTarget() function in tensorflow::ConvOpLowering561 static StringRef CustomCallTarget(ConvBackwardInputOp) { in CustomCallTarget() function in tensorflow::ConvOpLowering1130 static StringRef CustomCallTarget(AllGatherOp) { in CustomCallTarget() function in tensorflow::CollectiveOpLowering1133 static StringRef CustomCallTarget(AllReduceOp) { in CustomCallTarget() function in tensorflow::CollectiveOpLowering1136 static StringRef CustomCallTarget(AllReduceStartOp) { in CustomCallTarget() function in tensorflow::CollectiveOpLowering[all …]
51 using CustomCallTarget = std::function<void(Stream, void**, const char*, variable