Searched refs:excluded_ops (Results 1 – 11 of 11) sorted by relevance
| /external/tensorflow/tensorflow/core/api_def/ |
| D | api_test.cc | 219 auto* excluded_ops = GetExcludedOps(); in TEST_F() local
221 if (excluded_ops->find(op.name()) != excluded_ops->end()) { in TEST_F()
248 auto* excluded_ops = GetExcludedOps(); in TEST_F() local
250 if (excluded_ops->find(op.name()) != excluded_ops->end()) { in TEST_F()
|
| D | excluded_ops.cc | 21 static std::unordered_set<std::string>* excluded_ops = in GetExcludedOps() local
52 return excluded_ops; in GetExcludedOps()
|
| D | update_api_def.cc | 247 auto* excluded_ops = GetExcludedOps(); in CreateApiDefs() local
251 if (excluded_ops->find(op.name()) != excluded_ops->end()) { in CreateApiDefs()
|
| D | BUILD | 49 srcs = ["excluded_ops.cc"],
50 hdrs = ["excluded_ops.h"],
|
| /external/tensorflow/tensorflow/tools/graph_transforms/ |
| D | fold_constants_lib.cc | 301 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()
|
| D | fold_constants_test.cc | 175 std::vector<string> excluded_ops, in TestConstantFolding() argument
189 context.params["exclude_op"] = std::move(excluded_ops); in TestConstantFolding()
|
| /external/tensorflow/tensorflow/lite/delegates/gpu/common/ |
| D | model_builder_internal.h | 31 const absl::flat_hash_set<TfLiteBuiltinOperator>* excluded_ops = nullptr);
|
| D | model_builder.h | 43 const absl::flat_hash_set<TfLiteBuiltinOperator>* excluded_ops = nullptr);
|
| D | model_builder.cc | 2907 const absl::flat_hash_set<TfLiteBuiltinOperator>* excluded_ops = nullptr) { in IsSupported() argument
2908 return NewOperationParser(registration, allow_quant_ops, excluded_ops) in IsSupported()
2939 const absl::flat_hash_set<TfLiteBuiltinOperator>* excluded_ops) { in NewOperationParser() argument
2941 if (excluded_ops != nullptr && in NewOperationParser()
2942 excluded_ops->contains( in NewOperationParser()
3135 const absl::flat_hash_set<TfLiteBuiltinOperator>* excluded_ops) { in GetOpsToReplace() argument
3141 IsSupported(context, node, registration, allow_quant_ops, excluded_ops); in GetOpsToReplace()
|
| D | model_builder_test.cc | 1353 absl::flat_hash_set<TfLiteBuiltinOperator> excluded_ops = { in TEST() local
1357 /*max_delegated_partitions=*/1, &excluded_ops); in TEST()
|
| /external/tensorflow/tensorflow/lite/delegates/gpu/ |
| D | delegate.cc | 568 absl::flat_hash_set<TfLiteBuiltinOperator> excluded_ops; in DelegatePrepare() local
570 excluded_ops.insert(kTfLiteBuiltinSplit); in DelegatePrepare()
571 excluded_ops.insert(kTfLiteBuiltinSplitV); in DelegatePrepare()
575 gpu_delegate->MaxDelegatedPartitions(), &excluded_ops); in DelegatePrepare()
|