Searched refs:branch_func (Results 1 – 3 of 3) sorted by relevance
| /external/tensorflow/tensorflow/compiler/tf2xla/ |
| D | side_effect_util.cc | 45 NameAttrList branch_func; in SetDeviceOrdinalAttributeForNode() local
46 TF_RETURN_IF_ERROR(GetNodeAttr(node->attrs(), attr_name, &branch_func)); in SetDeviceOrdinalAttributeForNode()
47 (*branch_func.mutable_attr())["device_ordinal"] = device_ordinal_value; in SetDeviceOrdinalAttributeForNode()
49 node->AddAttr(attr_name, branch_func); in SetDeviceOrdinalAttributeForNode()
55 NameAttrList branch_func; in SetDeviceOrdinalAttributeForNode() local
56 TF_RETURN_IF_ERROR(GetNodeAttr(node->attrs(), attr_name, &branch_func)); in SetDeviceOrdinalAttributeForNode()
57 (*branch_func.mutable_attr())["device_ordinal"] = device_ordinal_value; in SetDeviceOrdinalAttributeForNode()
59 node->AddAttr(attr_name, branch_func); in SetDeviceOrdinalAttributeForNode()
|
| /external/tensorflow/tensorflow/core/grappler/optimizers/data/ |
| D | map_vectorization.cc | 325 FunctionDef* branch_func = library->add_function(); in AddBranch() local
326 auto* signature = branch_func->mutable_signature(); in AddBranch()
327 graph_utils::SetUniqueGraphFunctionName("branch", library, branch_func); in AddBranch()
347 auto function_node = branch_func->add_node_def(); in AddBranch()
349 function_utils::SetUniqueFunctionNodeName(node->name(), branch_func, in AddBranch()
383 (*branch_func->mutable_ret())["output"] = prev_node_output; in AddBranch()
387 func_attr.set_name(branch_func->signature().name()); in AddBranch()
|
| /external/tensorflow/tensorflow/compiler/jit/ |
| D | extract_outside_compilation_pass.cc | 423 NameAttrList branch_func; in ResetDeviceOrdinalToPlaceholderValue() local
424 TF_RETURN_IF_ERROR(GetNodeAttr(n->attrs(), attr_name, &branch_func)); in ResetDeviceOrdinalToPlaceholderValue()
425 (*branch_func.mutable_attr())["device_ordinal"] = device_ordinal_value; in ResetDeviceOrdinalToPlaceholderValue()
427 n->AddAttr(attr_name, branch_func); in ResetDeviceOrdinalToPlaceholderValue()
431 NameAttrList branch_func; in ResetDeviceOrdinalToPlaceholderValue() local
432 TF_RETURN_IF_ERROR(GetNodeAttr(n->attrs(), attr_name, &branch_func)); in ResetDeviceOrdinalToPlaceholderValue()
433 (*branch_func.mutable_attr())["device_ordinal"] = device_ordinal_value; in ResetDeviceOrdinalToPlaceholderValue()
435 n->AddAttr(attr_name, branch_func); in ResetDeviceOrdinalToPlaceholderValue()
|