Searched refs:IsInPreserveSet (Results 1 – 2 of 2) sorted by relevance
134 bool IsInPreserveSet(const RemapperContext& ctx, const NodeDef* node) { in IsInPreserveSet() function247 IsInPreserveSet(ctx, conv2d.node)) in FindConv2DWithBias()279 IsInPreserveSet(ctx, base.bias_add)) in FindConv2DWithBiasAndRelu()310 IsInPreserveSet(ctx, squeeze.node)) in FindConv2DWithSqueezeAndBias()328 IsInPreserveSet(ctx, conv2d.node)) in FindConv2DWithSqueezeAndBias()378 IsInPreserveSet(ctx, conv2d.node)) in FindConv2DWithBatchNorm()407 IsInPreserveSet(ctx, base.fused_batch_norm)) in FindConv2DWithBatchNormAndRelu()
236 bool IsInPreserveSet(const NodeDef& node) const { in IsInPreserveSet() function in tensorflow::grappler::__anonbb66c6470111::ArithmeticOptimizerStage415 bool IsInPreserveSet(const NodeDef& node) const { in IsInPreserveSet() function in tensorflow::grappler::__anonbb66c6470111::ArithmeticNodesGroupOptimizerStage503 if (IsInPreserveSet(node) || IsMarkedWithTag(node, kAddOpsRewriteTag)) { in CanOptimize()928 if (IsInPreserveSet(node)) { in IsAbsorbableByOptimizedNodesGroup()1351 return IsLogicalNot(*node) && !IsInPreserveSet(*node); in IsSupported()1358 if (IsInPreserveSet(*input) || in TrySimplify()1430 if (IsInPreserveSet(*node)) return false; in IsSupported()1581 IsInPreserveSet(*op)) { in OpsAreSafeToHoist()1735 !IsInPreserveSet(*node); in IsSupported()1767 if (IsSqrt(*y) && !IsInPreserveSet(*y) && in TrySimplify()[all …]