Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/core/common_runtime/
Dlower_case_op.cc41 CaseBuilder(Node* case_op, const std::vector<string>& branch_fn_names,
99 const std::vector<string>& branch_fn_names, in CaseBuilder() argument
102 num_branches_(branch_fn_names.size()), in CaseBuilder()
110 branch_fn_names[b], graph->op_registry(), in CaseBuilder()
279 std::vector<string> branch_fn_names; in RewriteCaseNode() local
280 branch_fn_names.reserve(num_branches); in RewriteCaseNode()
282 branch_fn_names.emplace_back(branches_attr->list().func(b).name()); in RewriteCaseNode()
284 CaseBuilder cb(n, branch_fn_names, keep_node_fetchable, g); in RewriteCaseNode()