Searched defs:ProdOp (Results 1 – 2 of 2) sorted by relevance
47 class ProdOp : public XlaReductionOp { class49 explicit ProdOp(OpKernelConstruction* ctx) in ProdOp() function in tensorflow::__anon98361f2a0111::ProdOp
278 struct ProdOp { struct279 inline T operator()(const T& a) const { return a; } in operator()280 inline T operator()(const T& a, const T& b) const { return a * b; } in operator()281 static constexpr T kNeutralElement = T(1);