Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/core/grappler/optimizers/
Dremapper.cc234 if (IsConv2D(*contraction) || IsDepthwiseConv2dNative(*contraction) || in IsCpuCompatibleDataType()
238 if (IsConv2D(*contraction) || IsDepthwiseConv2dNative(*contraction) || in IsCpuCompatibleDataType()
297 DCHECK(IsDepthwiseConv2dNative(*dw_conv2d)) in IsCpuCompatibleDataType()
308 } else if (IsDepthwiseConv2dNative(node)) { in IsCpuCompatibleDataType()
421 IsDepthwiseConv2dNative(*contraction_node_def); in IsCpuCompatibleDataType()
989 DCHECK(IsDepthwiseConv2dNative(dw_conv2d)) in IsCpuCompatibleDataType()
1082 } else if (IsDepthwiseConv2dNative(contraction)) { in IsCpuCompatibleDataType()
1131 } else if (IsDepthwiseConv2dNative(contraction)) { in IsCpuCompatibleDataType()
Dlayout_optimizer.cc2083 } else if (IsDepthwiseConv2dNative(*node)) { in Expand()
/external/tensorflow/tensorflow/core/grappler/
Dop_types.h69 bool IsDepthwiseConv2dNative(const NodeDef& node);
Dop_types.cc199 bool IsDepthwiseConv2dNative(const NodeDef& node) { in IsDepthwiseConv2dNative() function