Searched refs:IsAddN (Results 1 - 8 of 8) sorted by relevance

/external/tensorflow/tensorflow/core/grappler/optimizers/
H A Dmodel_pruner.cc44 if (IsAddN(node) && NumNonControlInputs(node) <= 1) {
H A Dlayout_optimizer.cc358 if (IsShapeN(node) || IsIdentityN(node) || IsAddN(node)) {
1979 if (IsAddN(*node)) {
H A Dmemory_optimizer.cc498 if (!IsAddN(node) && node.op() != "AccumulateNV2") {
/external/tensorflow/tensorflow/c/
H A Dc_test_util.h96 bool IsAddN(const tensorflow::NodeDef& node_def, int n);
H A Dc_test_util.cc310 bool IsAddN(const tensorflow::NodeDef& node_def, int n) { function
H A Dc_api_test.cc534 } else if (IsAddN(n, 2)) {
/external/tensorflow/tensorflow/core/grappler/
H A Dop_types.h26 bool IsAddN(const NodeDef& node);
H A Dop_types.cc36 bool IsAddN(const NodeDef& node) { return node.op() == "AddN"; } function in namespace:tensorflow::grappler

Completed in 504 milliseconds