Searched refs:AssociatedFunctionInfo (Results 1 – 4 of 4) sorted by relevance
| /external/tensorflow/tensorflow/compiler/tf2xla/ |
| D | tf2xla_util.h | 67 class AssociatedFunctionInfo {
76 static AssociatedFunctionInfo FunctionAttr(const string& func_name, in FunctionAttr()
79 return AssociatedFunctionInfo(kFunctionAttr, func_name, attrs, attr_name); in FunctionAttr()
83 static AssociatedFunctionInfo FunctionCall(const string& func_name, in FunctionCall()
86 return AssociatedFunctionInfo(kFunctionCallNode, func_name, attrs, in FunctionCall()
91 static AssociatedFunctionInfo SymbolicGradient(const string& func_name, in SymbolicGradient()
94 return AssociatedFunctionInfo(kSymbolicGradient, func_name, attrs, in SymbolicGradient()
107 AssociatedFunctionInfo(AssociatedFunctionType type, const string& func_name, in AssociatedFunctionInfo() function
132 std::vector<AssociatedFunctionInfo> GetAssociatedFunctions(
143 const AssociatedFunctionInfo& associated_function,
|
| D | functionalize_control_flow.cc | 70 AssociatedFunctionInfo::AssociatedFunctionType func_type, in GetNewFunctionName()
76 AssociatedFunctionInfo::AssociatedFunctionType::kSymbolicGradient in GetNewFunctionName()
98 const string& func_name, const AssociatedFunctionInfo& associated_function, in AddFunctionDefToGraphLibrary()
144 std::vector<std::pair<Node*, std::vector<AssociatedFunctionInfo>>> in FunctionalizeControlFlowForNodeAssociatedFunctions()
164 AssociatedFunctionInfo::kFunctionCallNode || in FunctionalizeControlFlowForNodeAssociatedFunctions()
|
| D | tf2xla_util.cc | 89 case AssociatedFunctionInfo::kFunctionCallNode: { in CopyAssociatedFunctions()
100 case AssociatedFunctionInfo::kSymbolicGradient: in CopyAssociatedFunctions()
101 case AssociatedFunctionInfo::kFunctionAttr: in CopyAssociatedFunctions()
591 std::vector<AssociatedFunctionInfo> GetAssociatedFunctions( in GetAssociatedFunctions()
593 std::vector<AssociatedFunctionInfo> results; in GetAssociatedFunctions()
598 results.emplace_back(AssociatedFunctionInfo::FunctionCall(op, attrs)); in GetAssociatedFunctions()
602 results.emplace_back(AssociatedFunctionInfo::SymbolicGradient(op, attrs)); in GetAssociatedFunctions()
612 results.emplace_back(AssociatedFunctionInfo::FunctionAttr( in GetAssociatedFunctions()
622 const AssociatedFunctionInfo& associated_function, in RewriteAssociatedFunction()
625 case AssociatedFunctionInfo::kFunctionCallNode: { in RewriteAssociatedFunction()
[all …]
|
| /external/tensorflow/tensorflow/core/tpu/graph_rewrite/ |
| D | host_training_loop_optimization_util.cc | 394 std::vector<AssociatedFunctionInfo> associated_function_list; in DetectHostTrainingLoop()
406 if (function.type() != AssociatedFunctionInfo::kFunctionAttr) continue; in DetectHostTrainingLoop()
|