Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/core/api_def/
Dapi_test.cc211 auto* excluded_ops = GetExcludedOps(); in TEST_F() local
213 if (excluded_ops->find(op.name()) != excluded_ops->end()) { in TEST_F()
240 auto* excluded_ops = GetExcludedOps(); in TEST_F() local
242 if (excluded_ops->find(op.name()) != excluded_ops->end()) { in TEST_F()
Dexcluded_ops.cc21 static std::unordered_set<std::string>* excluded_ops = in GetExcludedOps() local
37 return excluded_ops; in GetExcludedOps()
DBUILD45 srcs = ["excluded_ops.cc"],
46 hdrs = ["excluded_ops.h"],
Dupdate_api_def.cc243 auto* excluded_ops = GetExcludedOps(); in CreateApiDefs() local
247 if (excluded_ops->find(op.name()) != excluded_ops->end()) { in CreateApiDefs()
/external/tensorflow/tensorflow/tools/graph_transforms/
Dfold_constants_lib.cc301 std::set<string> excluded_ops, excluded_nodes; in FoldConstants() local
304 excluded_ops = std::set<string>(ops.begin(), ops.end()); in FoldConstants()
310 if (!excluded_ops.empty() || !excluded_nodes.empty()) { in FoldConstants()
311 cf_opts.consider = [excluded_ops, excluded_nodes](const Node* n) { in FoldConstants()
312 return excluded_ops.find(n->op_def().name()) == excluded_ops.end() && in FoldConstants()
Dfold_constants_test.cc175 std::vector<string> excluded_ops, in TestConstantFolding() argument
189 context.params["exclude_op"] = std::move(excluded_ops); in TestConstantFolding()