Searched defs:fused_node (Results 1 – 10 of 10) sorted by relevance
47 NodeDef* fused_node) { in FuseShuffleV1AndRepeat()81 NodeDef* fused_node) { in FuseShuffleV2AndRepeat()121 NodeDef* fused_node) { in FuseShuffleV3AndRepeat()172 NodeDef fused_node; in OptimizeAndCollectStats() local
40 NodeDef fused_node; in MakeFusedFilterNode() local
42 NodeDef fused_node; in MakeFusedNode() local
74 for (const auto &fused_node : fused_function->node_def()) { in TEST() local
48 NodeDef fused_node; in MakeFusedNode() local
394 NodeDef* fused_node) { in MaybeSetFusedMetadata()
145 auto fused_node = g.node(fused_id); in fuse_convolution_with_batch_normalization() local209 auto fused_node = g.node(fused_id); in fuse_depthwise_convolution_with_batch_normalization() local379 void fuse_convolution_with_post_op(Graph &g, INode *fused_node, std::list<INode *> post_op_node_lis… in fuse_convolution_with_post_op()624 auto fused_node = g.node(fused_id); in fuse_convolution_with_post_ops() local743 auto fused_node = g.node(fused_id); in fuse_convolution_batch_normalization_with_post_ops() local
2505 NodeDef fused_node; in AddFusedConv3DNode() local2607 NodeDef fused_node; in AddFusedMatMulBiasAddAndGelu() local2652 NodeDef fused_node; in AddMklLayerNorm() local2908 const NodeDef& fused_node = graph->node(matched.fused_batch_norm); in AddBatchNormNodes() local3153 NodeDef fused_node; in AddFusedBatchMatMul() local3314 NodeDef fused_node; in ReplaceSoftplusTanhAndMulWithMish() local
207 FusionCandidate& fused_node = candidates_[get_candidate_id(fused)]; in GetNewFusibles() local
453 NodeDef fused_node; in TEST_F() local