Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/core/grappler/optimizers/
Dremapper.cc841 const auto valid_batch_norm = in IsCpuCompatibleDataType() local
912 if (valid_batch_norm(*relu_fanin_0_node_view)) { in IsCpuCompatibleDataType()
945 if (valid_batch_norm(*add_regular_fanin_0.node_view())) { in IsCpuCompatibleDataType()
953 if (valid_batch_norm(*add_regular_fanin_1.node_view())) { in IsCpuCompatibleDataType()