Home
last modified time | relevance | path

Searched refs:op_names_ (Results 1 – 2 of 2) sorted by relevance

/external/tensorflow/tensorflow/core/framework/
Ddataset_stateful_op_whitelist.h29 op_names_.insert(std::move(op_name)); in Add()
33 bool Contains(const string& op_name) { return op_names_.count(op_name); } in Contains()
47 std::unordered_set<string> op_names_; variable
/external/tensorflow/tensorflow/core/kernels/
Dunary_ops_composition.cc94 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: [" << str_util::Join(op_names_, ", ") in UnaryOpsComposition()
153 std::vector<string> op_names_; member in tensorflow::UnaryOpsComposition