Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/core/kernels/
Dscan_ops_gpu.h118 struct IsSum { struct
139 static_assert(IsSum<T, Op>::value || IsProd<T, Op>::value || argument
145 typename std::enable_if<IsSum<U, OpCopy>::value, U>::type t = U(0)) { in operator()
Dreduction_gpu_kernels.cu.h993 struct IsSum {
1027 static_assert(IsSum<T, Op>::value || IsMax<T, Op>::value ||
1034 typename std::enable_if<IsSum<U, OpCopy>::value, U>::type t = U(0)) {
/external/tensorflow/tensorflow/core/grappler/
Dop_types.h191 bool IsSum(const NodeDef& node);
Dop_types.cc581 bool IsSum(const NodeDef& node) { return node.op() == "Sum"; } in IsSum() function
/external/tensorflow/tensorflow/core/grappler/optimizers/
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()