Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/core/kernels/
Dscan_ops_gpu.h130 struct IsProd { struct
138 static_assert(IsSum<T, Op>::value || IsProd<T, Op>::value, argument
149 typename std::enable_if<IsProd<U, OpCopy>::value, U>::type t = U(1)) { in operator()
Dreduction_gpu_kernels.cu.h799 struct IsProd {
808 IsMin<T, Op>::value || IsProd<T, Op>::value ||
832 typename std::enable_if<IsProd<U, OpCopy>::value, U>::type t = U(1)) {
/external/tensorflow/tensorflow/core/grappler/
Dop_types.h121 bool IsProd(const NodeDef& node);
Dop_types.cc367 bool IsProd(const NodeDef& node) { return node.op() == "Prod"; } in IsProd() function
/external/tensorflow/tensorflow/core/grappler/optimizers/
Dlayout_optimizer.cc312 return IsSum(node) || IsMean(node) || IsProd(node) || IsMax(node) || in IsReduceOp()