Searched defs:broadcast_helper (Results 1 – 3 of 3) sorted by relevance
75 xla::XlaOp y, const BCast& broadcast_helper) { in DivNoNanImpl()93 xla::XlaOp y, const BCast& broadcast_helper) { in MulNoNanImpl()115 xla::XlaOp y, const BCast& broadcast_helper) { in FloorDivImpl()144 const BCast& broadcast_helper) { in XlogyImpl()153 const BCast& broadcast_helper) { in Xlog1pyImpl()163 const BCast& broadcast_helper) { in XdivyImpl()176 xla::XlaOp y, const BCast& broadcast_helper) { in FloorModImpl()260 const BCast& broadcast_helper) { in IgammaImpl()268 const BCast& broadcast_helper) { in IgammaGradAImpl()276 const BCast& broadcast_helper) { in RandomGammaGradImpl()[all …]
84 const MatMulBCast& broadcast_helper) { in Broadcast()
149 xla::XlaOp lhs, xla::XlaOp rhs, const BCast& broadcast_helper) { in Broadcast()