Searched refs:op_names_ (Results 1 – 3 of 3) sorted by relevance
| /external/tensorflow/tensorflow/core/framework/ |
| D | dataset_stateful_op_allowlist.h | 29 op_names_.insert(std::move(op_name)); in Add()
34 op_names_.erase(op_name); in Remove()
38 bool Contains(const string& op_name) { return op_names_.count(op_name); } in Contains()
52 std::unordered_set<string> op_names_; variable
|
| /external/tensorflow/tensorflow/compiler/tf2xla/kernels/ |
| D | unary_ops_composition.cc | 96 OP_REQUIRES_OK(ctx, ctx->GetAttr("op_names", &op_names_)); in UnaryOpsCompositionOp()
99 for (absl::string_view op_name : op_names_) { in UnaryOpsCompositionOp()
109 for (absl::string_view op_name : op_names_) { in Compile()
116 std::vector<string> op_names_; member in tensorflow::__anonfa2f6eb00111::UnaryOpsCompositionOp
|
| /external/tensorflow/tensorflow/core/kernels/ |
| D | unary_ops_composition.cc | 94 OP_REQUIRES_OK(context, context->GetAttr("op_names", &op_names_)); in UnaryOpsComposition()
96 OP_REQUIRES(context, !op_names_.empty(), in UnaryOpsComposition()
101 support_.ExportComputeFns(op_names_, &fns_, &cost_)); in UnaryOpsComposition()
103 VLOG(2) << "Composed unary op: [" << absl::StrJoin(op_names_, ", ") in UnaryOpsComposition()
154 std::vector<string> op_names_; member in tensorflow::UnaryOpsComposition
|