| /external/tensorflow/tensorflow/core/grappler/optimizers/ |
| D | generic_layout_optimizer_transposer.h | 100 const utils::MutableNodeView& node) const;
106 utils::MutableNodeView* node) = 0;
132 utils::MutableNodeView* dst_node,
139 utils::MutableNodeView* src_node,
152 int GetFanoutPortRank(const utils::MutableNodeView& node, int port) const;
153 bool IsFanoutPortRankN(const utils::MutableNodeView& node, int port,
155 bool IsFanoutPortsRankN(const utils::MutableNodeView& node,
157 int GetFaninPortRank(const utils::MutableNodeView& node, int port) const;
158 bool IsFaninPortRankN(const utils::MutableNodeView& node, int port,
164 bool IsFaninPortDimsNIfConst(const utils::MutableNodeView& node, int port,
[all …]
|
| D | generic_layout_optimizer_transposer.cc | 73 inline bool AttrDataFormatMatch(const utils::MutableNodeView& node, in AttrDataFormatMatch()
84 inline bool AttrDataFormatMatch(const utils::MutableNodeView& node, in AttrDataFormatMatch()
90 bool IsNonFloatingConv2D(const utils::MutableNodeView& node) { in IsNonFloatingConv2D()
109 std::vector<int> GetRegularFaninPorts(const utils::MutableNodeView& node) { in GetRegularFaninPorts()
116 std::vector<int> GetConcatDataFaninPorts(const utils::MutableNodeView& node) { in GetConcatDataFaninPorts()
265 const utils::MutableNodeView& node) const { in ShouldProcess()
377 utils::MutableNodeView* dst_node, in UpdateFaninEdgesWithOp()
397 utils::MutableNodeView* src_node, in UpdateFanoutEdgesWithOp()
490 utils::MutableNodeView* src_node, utils::MutableNodeView* dst_node) { in UpdateEdge()
559 int Transposer::GetFanoutPortRank(const utils::MutableNodeView& node, in GetFanoutPortRank()
[all …]
|
| D | generic_layout_optimizer.cc | 44 void operator()(std::string* out, utils::MutableNodeView* node_view) const { in operator ()()
164 const utils::MutableNodeView& fanout_transpose, in IsCancellableConstPermTransposeNodePair()
165 const utils::MutableNodeView& fanin_transpose) { in IsCancellableConstPermTransposeNodePair()
194 const utils::MutableNodeView& fanout_transpose, in IsCancellableDataFormatNodePair()
195 const utils::MutableNodeView& fanin_transpose) { in IsCancellableDataFormatNodePair()
200 auto src_dst_match = [](const utils::MutableNodeView& src, in IsCancellableDataFormatNodePair()
201 const utils::MutableNodeView& dst) { in IsCancellableDataFormatNodePair()
220 const utils::MutableNodeView& fanout_transpose, in IsCancellableNodePair()
221 const utils::MutableNodeView& fanin_transpose) { in IsCancellableNodePair()
278 absl::flat_hash_set<utils::MutableNodeView*> cancelled_transposes; in EraseCancellableNodesAroundPad()
[all …]
|
| D | implementation_selector.cc | 102 string FindForwardNode(utils::MutableNodeView* backward_node) { in FindForwardNode()
131 void UpdateForwardIdentityNodeDtype(utils::MutableNodeView* forward_node, in UpdateForwardIdentityNodeDtype()
147 Status UpdateNodeDef(utils::MutableNodeView* node_view, const string& funcName, in UpdateNodeDef()
239 utils::MutableNodeView* node_view) const { in MaybeOptimizeFunctionCall()
305 Status FindDeviceIndex(const utils::MutableNodeView* device_index_node, in FindDeviceIndex()
326 void RewriteDeviceIndexOp(utils::MutableNodeView* device_index_node, in RewriteDeviceIndexOp()
|
| D | implementation_selector.h | 117 Status MaybeOptimizeFunctionCall(utils::MutableNodeView* node_view) const;
|
| D | remapper.cc | 380 inline bool HasControlFaninOrFanout(const utils::MutableNodeView& node_view) { in IsCpuCompatibleDataType()
386 inline bool HasAtMostOneFanoutAtPort0(const utils::MutableNodeView& node_view) { in IsCpuCompatibleDataType()
393 const utils::MutableNodeView& node_view) { in IsCpuCompatibleDataType()
634 const utils::MutableNodeView& add_node_view, in IsCpuCompatibleDataType()
667 const utils::MutableNodeView& node_view, in IsCpuCompatibleDataType()
842 [&](const utils::MutableNodeView& fused_batch_norm) -> bool { in IsCpuCompatibleDataType()
|
| D | generic_layout_optimizer_transposer_test.cc | 69 Status TransposeNode(TransposeContext*, utils::MutableNodeView*) override { in TransposeNode() argument
74 void VerifyRegularFaninMatch(const utils::MutableNodeView* node, int port, in VerifyRegularFaninMatch()
82 void VerifyShapeAttributeMatch(const utils::MutableNodeView* node, in VerifyShapeAttributeMatch()
89 void VerifyShapeAttributeMatch(const utils::MutableNodeView* node, in VerifyShapeAttributeMatch()
97 void VerifyDataFormatAttributeMatch(const utils::MutableNodeView* node, in VerifyDataFormatAttributeMatch()
526 utils::MutableNodeView* const_perm_node = in TEST_F()
|
| /external/tensorflow/tensorflow/core/grappler/utils/ |
| D | graph_view.h | 128 class MutableNodeView; variable
138 : public internal::NodeIndexAndPortIndex<MutableNodeView,
154 MutableFaninView(MutableNodeView* node_view, int index);
164 friend class MutableNodeView; variable
173 : public internal::NodeIndexAndPortIndex<MutableNodeView,
187 MutableFanoutView(MutableNodeView* node_view, int index);
197 friend class MutableNodeView; variable
207 class MutableNodeView
211 explicit MutableNodeView(MutableGraphView* graph_view, int node_index) in MutableNodeView() function
214 MutableNodeView() : NodeViewInternal() {} in MutableNodeView() function
[all …]
|
| D | graph_view.cc | 171 MutableFaninView::MutableFaninView(MutableNodeView* node_view, int index) in MutableFaninView()
175 MutableFanoutView::MutableFanoutView(MutableNodeView* node_view, int index) in MutableFanoutView()
179 NodeDef* MutableNodeView::node() const { in node()
183 bool MutableNodeView::HasFanin(const MutableFanoutView& fanin) const { in HasFanin()
191 bool MutableNodeView::HasFanout(const MutableFaninView& fanout) const { in HasFanout()
196 MutableNodeView* view = fanout.node_view(); in HasFanout()
207 const MutableFanoutView& MutableNodeView::GetMissingFanin() const { in GetMissingFanin()
211 const std::vector<MutableFaninView>& MutableNodeView::GetMissingFanout() const { in GetMissingFanout()
265 MutableNodeView* node, in AddMutation()
280 void Mutation::RemoveNode(MutableNodeView* node) { in RemoveNode()
[all …]
|
| D | graph_view_internal_test.cc | 63 MutableNodeView* d_node = graph_view.GetNode("d"); in TEST()
87 MutableNodeView* d_node = graph_view.GetNode("d"); in TEST()
111 MutableNodeView* d_node = graph_view.GetNode("d"); in TEST()
135 MutableNodeView* d_node = graph_view.GetNode("d"); in TEST()
181 MutableNodeView* d_node = graph_view.GetNode("d"); in TEST()
220 MutableNodeView* d_node = graph_view.GetNode("d"); in TEST()
304 MutableNodeView* d_node = graph_view.GetNode("d"); in TEST()
339 MutableNodeView* d_node = graph_view.GetNode("d"); in TEST()
363 MutableNodeView* d_node = graph_view.GetNode("d"); in TEST()
395 MutableNodeView* d_node = graph_view.GetNode("d"); in TEST()
[all …]
|
| D | pattern_utils.h | 148 MutableNodeView* node_view = nullptr;
185 bool GetMatchedNodes(const OpTypePattern& pattern, MutableNodeView* node_view,
197 MutableNodeView* node_view, NodeViewMatch* match);
|
| D | pattern_utils.cc | 28 const OpTypePattern& pattern, MutableNodeView* node_view, in DoesOpTypePatternMatch()
86 MutableNodeView* child_node_view = in DoesOpTypePatternMatch()
104 const OpTypePattern& pattern, MutableNodeView* node_view, in GetMatchedNodes()
|
| D | graph_view_test.cc | 805 MutableNodeView* node_view = graph_view->GetNode(node_name); in CompareGraphViewWithGraph()
1064 MutableNodeView* d_node = graph_view.GetNode("d"); in TEST_F()
1085 MutableNodeView* d_node = graph_view.GetNode("d"); in TEST_F()
1107 MutableNodeView* a_node = graph_view.GetNode("a"); in TEST_F()
1111 MutableNodeView* b_node = graph_view.GetNode("b"); in TEST_F()
1137 MutableNodeView* a_node = graph_view.GetNode("a"); in TEST_F()
1163 MutableNodeView* d_node = graph_view.GetNode("d"); in TEST_F()
1185 MutableNodeView* b_node = graph_view.GetNode("b"); in TEST_F()
1196 MutableNodeView* d_node = graph_view.GetNode("d"); in TEST_F()
1216 MutableNodeView* d_node = graph_view.GetNode("d"); in TEST_F()
[all …]
|