Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/core/grappler/optimizers/
Dremapper.cc455 const auto* bias_add_node_view = regular_fanin_0.node_view(); in IsCpuCompatibleDataType() local
456 const auto* bias_add_node_def = bias_add_node_view->node(); in IsCpuCompatibleDataType()
459 if (!FindContractionWithBias(ctx, bias_add_node_view->node_index(), &base, in IsCpuCompatibleDataType()
461 !HasAtMostOneFanoutAtPort0(*bias_add_node_view) || in IsCpuCompatibleDataType()
468 bias_add_node_view->GetRegularFanin(0).node_view(); in IsCpuCompatibleDataType()
639 const auto& bias_add_node_view = in IsCpuCompatibleDataType() local
641 if (bias_add_node_view == nullptr) return false; in IsCpuCompatibleDataType()
642 const auto* bias_add_node_def = bias_add_node_view->node(); in IsCpuCompatibleDataType()
644 if (!FindContractionWithBias(ctx, bias_add_node_view->node_index(), base, in IsCpuCompatibleDataType()
647 if (!HasAtMostOneFanoutAtPort0(*bias_add_node_view) || in IsCpuCompatibleDataType()
[all …]