Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/compiler/tf2xla/
Dtf2xla_util.h66 class AssociatedFunctionInfo {
75 static AssociatedFunctionInfo FunctionAttr(const string& func_name, in FunctionAttr()
78 return AssociatedFunctionInfo(kFunctionAttr, func_name, attrs, attr_name); in FunctionAttr()
82 static AssociatedFunctionInfo FunctionCall(const string& func_name, in FunctionCall()
85 return AssociatedFunctionInfo(kFunctionCallNode, func_name, attrs, in FunctionCall()
90 static AssociatedFunctionInfo SymbolicGradient(const string& func_name, in SymbolicGradient()
93 return AssociatedFunctionInfo(kSymbolicGradient, func_name, attrs, in SymbolicGradient()
106 AssociatedFunctionInfo(AssociatedFunctionType type, const string& func_name, in AssociatedFunctionInfo() function
131 std::vector<AssociatedFunctionInfo> GetAssociatedFunctions(
142 const AssociatedFunctionInfo& associated_function,
Dtf2xla_util.cc89 case AssociatedFunctionInfo::kFunctionCallNode: { in CopyAssociatedFunctions()
100 case AssociatedFunctionInfo::kSymbolicGradient: in CopyAssociatedFunctions()
101 case AssociatedFunctionInfo::kFunctionAttr: in CopyAssociatedFunctions()
592 std::vector<AssociatedFunctionInfo> GetAssociatedFunctions( in GetAssociatedFunctions()
594 std::vector<AssociatedFunctionInfo> results; in GetAssociatedFunctions()
599 results.emplace_back(AssociatedFunctionInfo::FunctionCall(op, attrs)); in GetAssociatedFunctions()
603 results.emplace_back(AssociatedFunctionInfo::SymbolicGradient(op, attrs)); in GetAssociatedFunctions()
613 results.emplace_back(AssociatedFunctionInfo::FunctionAttr( in GetAssociatedFunctions()
623 const AssociatedFunctionInfo& associated_function, in RewriteAssociatedFunction()
626 case AssociatedFunctionInfo::kFunctionCallNode: { in RewriteAssociatedFunction()
[all …]
Dfunctionalize_control_flow.cc133 std::vector<std::pair<Node*, std::vector<AssociatedFunctionInfo>>> in FunctionalizeControlFlowForFunction()
161 AssociatedFunctionInfo::AssociatedFunctionType::kSymbolicGradient) { in FunctionalizeControlFlowForFunction()