Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/core/kernels/
Dscan_ops_gpu.h123 struct IsSum { struct
138 static_assert(IsSum<T, Op>::value || IsProd<T, Op>::value, argument
143 typename std::enable_if<IsSum<U, OpCopy>::value, U>::type t = U(0)) { in operator()
Dreduction_gpu_kernels.cu.h777 struct IsSum {
807 static_assert(IsSum<T, Op>::value || IsMax<T, Op>::value ||
814 typename std::enable_if<IsSum<U, OpCopy>::value, U>::type t = U(0)) {
/external/tensorflow/tensorflow/core/grappler/
Dop_types.h170 bool IsSum(const NodeDef& node);
Dop_types.cc502 bool IsSum(const NodeDef& node) { return node.op() == "Sum"; } in IsSum() function
/external/tensorflow/tensorflow/core/grappler/optimizers/
Dlayout_optimizer.cc312 return IsSum(node) || IsMean(node) || IsProd(node) || IsMax(node) || in IsReduceOp()