Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/core/grappler/optimizers/
Dremapper.cc523 const auto* conv2d_node_view = squeeze_regular_fanin_0.node_view(); in IsCpuCompatibleDataType() local
524 const auto* conv2d_node_def = conv2d_node_view->node(); in IsCpuCompatibleDataType()
528 HasControlFaninOrFanout(*conv2d_node_view) || in IsCpuCompatibleDataType()
529 !HasAtMostOneFanoutAtPort0(*conv2d_node_view) || in IsCpuCompatibleDataType()
535 conv2d_node_view->node_index(), squeeze_node_view->node_index(), in IsCpuCompatibleDataType()
573 const auto* conv2d_node_view = regular_fanin_0.node_view(); in IsCpuCompatibleDataType() local
574 const auto* conv2d_node_def = conv2d_node_view->node(); in IsCpuCompatibleDataType()
580 HasControlFaninOrFanout(*conv2d_node_view) || in IsCpuCompatibleDataType()
581 !HasAtMostOneFanoutAtPort0(*conv2d_node_view) || in IsCpuCompatibleDataType()
586 matched->contraction = conv2d_node_view->node_index(); in IsCpuCompatibleDataType()