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
|