Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/compiler/tf2xla/
Dfunctionalize_cond.h92 using CondState = std::map<OutputTensor, BranchType, OutputTensorLess>; variable
95 using CondId = const CondState*;
107 CondId GetCondId(const CondState& state);
143 size_t operator()(const CondState& map) const;
150 std::unordered_set<CondState, Hash> condstate_set_;
Dfunctionalize_cond_test.cc40 StateMap::CondId GetUniqueId(const StateMap::StateMap::CondState& state) { in GetUniqueId()
78 StateMap::CondState ss; in TEST_F()
84 StateMap::CondState ss; in TEST_F()
110 StateMap::CondState cond_state; in TEST_F()
Dfunctionalize_cond.cc103 using value_type = StateMap::CondState::value_type; in DebugString()
146 bool operator()(const StateMap::CondState::value_type& lhs, in operator ()()
147 const StateMap::CondState::value_type& rhs) const { in operator ()()
169 size_t StateMap::Hash::operator()(const StateMap::CondState& map) const { in operator ()()
229 StateMap::CondId StateMap::GetCondId(const StateMap::CondState& state) { in GetCondId()
632 std::vector<StateMap::CondState::value_type> diff; in ExtractBodies()
657 StateMap::CondState state = *dst_id; in ExtractBodies()
939 StateMap::CondState state = *id; in AddIfNode()
997 const CondState& nodes = *id; in FindBranchOf()
1014 StateMap::CondState both = *src; in JoinCondStatesNonMerge()
[all …]
/external/tensorflow/tensorflow/python/autograph/pyct/
Dtransformer_test.py107 class CondState(object): class
114 anno.setanno(node, 'cond_state', self.state[CondState].value)
124 self.state[CondState].enter()
126 self.state[CondState].exit()