Home
last modified time | relevance | path

Searched refs:IsProd (Results 1 – 7 of 7) sorted by relevance

/external/tensorflow/tensorflow/core/kernels/
Dscan_ops_gpu.h125 struct IsProd { struct
139 static_assert(IsSum<T, Op>::value || IsProd<T, Op>::value || argument
151 typename std::enable_if<IsProd<U, OpCopy>::value, U>::type t = U(1)) { in operator()
Dreduction_gpu_kernels.cu.h1019 struct IsProd {
1028 IsMin<T, Op>::value || IsProd<T, Op>::value ||
1052 typename std::enable_if<IsProd<U, OpCopy>::value, U>::type t = U(1)) {
/external/tensorflow/tensorflow/core/grappler/optimizers/
Dshape_optimizer.cc52 } else if (IsProd(node) && is_int(node)) { in Optimize()
Dlayout_optimizer.cc320 return IsSum(node) || IsMean(node) || IsProd(node) || IsMax(node) || in IsReduceOp()
Dgeneric_layout_optimizer_transposer.cc2102 return IsSum(node) || IsMean(node) || IsProd(node) || IsMax(node) || in IsReduceOp()
/external/tensorflow/tensorflow/core/grappler/
Dop_types.h139 bool IsProd(const NodeDef& node);
Dop_types.cc434 bool IsProd(const NodeDef& node) { return node.op() == "Prod"; } in IsProd() function