Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/core/grappler/optimizers/
Darithmetic_optimizer.cc253 bool IsInPreserveSet(const NodeDef& node) const { in IsInPreserveSet() function in tensorflow::grappler::__anon73f394250111::ArithmeticOptimizerStage
440 bool IsInPreserveSet(const NodeDef& node) const { in IsInPreserveSet() function in tensorflow::grappler::__anon73f394250111::ArithmeticNodesGroupOptimizerStage
528 if (IsInPreserveSet(node) || IsMarkedWithTag(node, kAddOpsRewriteTag)) { in CanOptimize()
956 if (IsInPreserveSet(node)) { in IsAbsorbableByOptimizedNodesGroup()
1290 if ((input_type == output_type) && !IsInPreserveSet(*node)) { in TrySimplify()
1300 if (IsBitcast(*operand) && !IsInPreserveSet(*operand)) { in TrySimplify()
1326 return IsCast(*node) && !IsInPreserveSet(*node); in IsSupported()
1353 return (IsAdd(*node) || IsSub(*node)) && !IsInPreserveSet(*node); in IsSupported()
1393 return IsLogicalNot(*node) && !IsInPreserveSet(*node); in IsSupported()
1400 if (IsInPreserveSet(*input) || in TrySimplify()
[all …]
Dremapper.cc210 bool IsInPreserveSet(const RemapperContext& ctx, const NodeDef* node) { in IsInPreserveSet() function
426 IsInPreserveSet(ctx, contraction_node_def)) in IsCpuCompatibleDataType()
463 IsInPreserveSet(ctx, bias_add_node_def)) in IsCpuCompatibleDataType()
510 IsInPreserveSet(ctx, squeeze_node_def)) in IsCpuCompatibleDataType()
530 IsInPreserveSet(ctx, conv2d_node_def)) in IsCpuCompatibleDataType()
582 IsInPreserveSet(ctx, conv2d_node_def)) in IsCpuCompatibleDataType()
618 IsInPreserveSet(ctx, fused_batch_norm_node_def)) in IsCpuCompatibleDataType()
649 IsInPreserveSet(ctx, bias_add_node_def)) in IsCpuCompatibleDataType()
900 IsInPreserveSet(ctx, fused_batch_norm_node_def)) in IsCpuCompatibleDataType()
929 IsInPreserveSet(ctx, relu_fanin_0_node_def)) in IsCpuCompatibleDataType()