Searched refs:IsBiasAdd (Results 1 – 5 of 5) sorted by relevance
47 bool IsBiasAdd(const NodeDef& node);
112 bool IsBiasAdd(const NodeDef& node) { in IsBiasAdd() function
410 if (!IsBiasAdd(*node_def)) return false; in IsCpuCompatibleDataType()498 if (!IsBiasAdd(*node_def)) return false; in IsCpuCompatibleDataType()1652 if (IsBiasAdd(*node_view->node())) { in IsCpuCompatibleDataType()1715 if (!IsBiasAdd(*relu_fanin_0_node_def)) return false; in IsCpuCompatibleDataType()
2924 const bool is_add = IsAdd(*node) || IsBiasAdd(*node) || IsLogicalOr(*node); in SimplifyArithmeticOperations()3210 const bool parent_is_bias_add = IsBiasAdd(*node); in ConstantPushDownBiasAdd()3220 const bool child_is_bias_add = IsBiasAdd(*ctx.op_child); in ConstantPushDownBiasAdd()
1663 if (IsFusedBatchNorm(*operand) || IsBiasAdd(*operand)) { in OpsAreSafeToHoist()2903 if (IsFusedBatchNorm(producer) || IsBiasAdd(producer)) { in TrySimplify()