Searched refs:XlaBinaryOp (Results 1 – 3 of 3) sorted by relevance
| /external/tensorflow/tensorflow/compiler/tf2xla/kernels/ |
| D | cwise_ops.h | 36 class XlaBinaryOp : public XlaOpKernel {
38 explicit XlaBinaryOp(OpKernelConstruction* ctx) : XlaOpKernel(ctx) { in XlaBinaryOp() function
44 ~XlaBinaryOp() override {} in ~XlaBinaryOp()
|
| D | binary_ops.cc | 37 class NAME##Op : public XlaBinaryOp { \
39 explicit NAME##Op(OpKernelConstruction* ctx) : XlaBinaryOp(ctx) {} \
72 std::tie(x, y) = XlaBinaryOp::Broadcast(x, y, broadcast_helper); in DivNoNanImpl()
90 std::tie(x, y) = XlaBinaryOp::Broadcast(x, y, broadcast_helper); in MulNoNanImpl()
113 std::tie(x, y) = XlaBinaryOp::Broadcast(x, y, broadcast_helper); in FloorDivImpl()
133 std::tie(x, y) = XlaBinaryOp::Broadcast(x, y, broadcast_helper); in XlogyImpl()
142 std::tie(x, y) = XlaBinaryOp::Broadcast(x, y, broadcast_helper); in XdivyImpl()
154 std::tie(x, y) = XlaBinaryOp::Broadcast(x, y, broadcast_helper); in FloorModImpl()
|
| D | cwise_ops.cc | 33 void XlaBinaryOp::Compile(XlaOpKernelContext* ctx) { in Compile()
90 /* static */ std::pair<xla::XlaOp, xla::XlaOp> XlaBinaryOp::Broadcast( in Broadcast()
|