Searched refs:IsFusedBatchNorm (Results 1 – 4 of 4) sorted by relevance
550 if (!IsFusedBatchNorm(*node_def)) return false; in IsCpuCompatibleDataType()776 if (!IsFusedBatchNorm(*node_def)) return false; in IsCpuCompatibleDataType()844 if (!IsFusedBatchNorm(*fused_batch_norm_node_def)) return false; in IsCpuCompatibleDataType()1009 DCHECK(IsFusedBatchNorm(fused_batch_norm)) in IsCpuCompatibleDataType()1696 if (!IsFusedBatchNorm(*node_def)) return false; in IsCpuCompatibleDataType()1740 if (IsFusedBatchNorm(*relu_fanin_0_node_def)) { in IsCpuCompatibleDataType()1750 if (IsFusedBatchNorm(*add_regular_fanin_0.node_view()->node())) in IsCpuCompatibleDataType()1754 if (IsFusedBatchNorm(*add_regular_fanin_1.node_view()->node())) in IsCpuCompatibleDataType()
1663 if (IsFusedBatchNorm(*operand) || IsBiasAdd(*operand)) { in OpsAreSafeToHoist()2903 if (IsFusedBatchNorm(producer) || IsBiasAdd(producer)) { in TrySimplify()
87 bool IsFusedBatchNorm(const NodeDef& node);
282 bool IsFusedBatchNorm(const NodeDef& node) { in IsFusedBatchNorm() function