Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/core/grappler/optimizers/
Dremapper.cc1459 string x_format = fused_node.attr().at(kDataFormat).s(); in IsCpuCompatibleDataType() local
1460 if (x_format == "NCHW" || x_format == "NCDHW") { in IsCpuCompatibleDataType()
1464 AddPrefixToNodeName(x_format + "Shape", fused_node.name()); in IsCpuCompatibleDataType()
1470 if (x_format == "NCHW") { in IsCpuCompatibleDataType()
1493 AddPrefixToNodeName(x_format + "ShapedScale", fused_node.name())); in IsCpuCompatibleDataType()
1506 AddPrefixToNodeName(x_format + "ShapedOffset", fused_node.name())); in IsCpuCompatibleDataType()
1519 AddPrefixToNodeName(x_format + "ShapedMean", fused_node.name())); in IsCpuCompatibleDataType()
1532 AddPrefixToNodeName(x_format + "ShapedVariance", fused_node.name())); in IsCpuCompatibleDataType()