Searched refs:SetFusedOpAttributes (Results 1 – 1 of 1) sorted by relevance
| /external/tensorflow/tensorflow/core/grappler/optimizers/ |
| D | remapper.cc | 2198 void SetFusedOpAttributes(NodeDef* fused, in SetFusedOpAttributes() function
2240 SetFusedOpAttributes(&fused_op, {"BiasAdd"}); in AddFusedContractionNode()
2291 SetFusedOpAttributes(&fused_op, {"BiasAdd", activation.op()}); in AddFusedContractionNode()
2338 SetFusedOpAttributes(&fused_conv, {"BiasAdd"}); in AddFusedConvNode()
2383 SetFusedOpAttributes(&fused_conv2d, {"FusedBatchNorm"}, in AddFusedConv2DNode()
2426 SetFusedOpAttributes(&fused_conv2d, {"FusedBatchNorm", activation.op()}, in AddFusedConv2DNode()
2479 SetFusedOpAttributes(&contraction_node, {"BiasAdd", "Add"}, 2); in AddFusedContractionNode()
2581 SetFusedOpAttributes(&fused_conv, {"BiasAdd", "Add", activation.op()}, 2); in AddFusedContractionNode()
2623 SetFusedOpAttributes(&fused_node, {"BiasAdd", "GeluApproximate"}); in AddFusedMatMulBiasAddAndGelu()
2625 SetFusedOpAttributes(&fused_node, {"BiasAdd", "GeluExact"}); in AddFusedMatMulBiasAddAndGelu()
[all …]
|