Searched refs:SetFusedOpAttributes (Results 1 – 1 of 1) sorted by relevance
| /external/tensorflow/tensorflow/core/grappler/optimizers/ |
| D | remapper.cc | 1050 void SetFusedOpAttributes(NodeDef* fused, in IsCpuCompatibleDataType() function
1090 SetFusedOpAttributes(&fused_op, {"BiasAdd"}); in IsCpuCompatibleDataType()
1139 SetFusedOpAttributes(&fused_op, {"BiasAdd", activation.op()}); in IsCpuCompatibleDataType()
1181 SetFusedOpAttributes(&fused_conv2d, {"BiasAdd"}); in IsCpuCompatibleDataType()
1226 SetFusedOpAttributes(&fused_conv2d, {"FusedBatchNorm"}, in IsCpuCompatibleDataType()
1269 SetFusedOpAttributes(&fused_conv2d, {"FusedBatchNorm", activation.op()}, in IsCpuCompatibleDataType()
1319 SetFusedOpAttributes(&contraction_node, {"BiasAdd", "Add"}, 2); in IsCpuCompatibleDataType()
1358 SetFusedOpAttributes(&fused_conv2d, {"BiasAdd", "Add", activation.op()}, 2); in IsCpuCompatibleDataType()
|