HomeSort by relevance Sort by last modified time
    Searched refs:IsAddN (Results 1 - 8 of 8) sorted by null

  /external/tensorflow/tensorflow/core/grappler/optimizers/
model_pruner.cc 44 if (IsAddN(node) && NumNonControlInputs(node) <= 1) {
layout_optimizer.cc 358 if (IsShapeN(node) || IsIdentityN(node) || IsAddN(node)) {
    [all...]
memory_optimizer.cc 498 if (!IsAddN(node) && node.op() != "AccumulateNV2") {
    [all...]
  /external/tensorflow/tensorflow/c/
c_test_util.h 96 bool IsAddN(const tensorflow::NodeDef& node_def, int n);
c_test_util.cc 310 bool IsAddN(const tensorflow::NodeDef& node_def, int n) {
c_api_test.cc 534 } else if (IsAddN(n, 2)) {
    [all...]
  /external/tensorflow/tensorflow/core/grappler/
op_types.h 26 bool IsAddN(const NodeDef& node);
op_types.cc 36 bool IsAddN(const NodeDef& node) { return node.op() == "AddN"; }

Completed in 90 milliseconds