Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/compiler/tf2xla/
Dside_effect_util.cc47 if (!HasNodeAttr(node->def(), kXlaHasHostTransferAttrName)) { in SetDeviceOrdinalAttributeForNode()
78 } else if (HasNodeAttr(node->def(), "_device_ordinal")) { in SetDeviceOrdinalAttributeForNode()
Dsharding_util.cc31 if (!HasNodeAttr(node_def, kShardingAttribute)) { in GetShardingFromNodeDef()
/external/tensorflow/tensorflow/core/grappler/optimizers/data/vectorization/
Dunpack_vectorizer.cc33 if (HasNodeAttr(node.def(), "axis")) { in Vectorize()
/external/tensorflow/tensorflow/compiler/jit/
Dencapsulate_util.cc205 HasNodeAttr(n->def(), kOutsideCompilationOriginalNodeAttrName)) { in PostprocessDataEdgesBetweenOutsideCompilations()
389 if (HasNodeAttr(e->dst()->def(), outside_compilation_attr_name)) { in PreprocessEdgesBetweenOutsideCompilations()
394 if (HasNodeAttr(e->src()->def(), outside_compilation_attr_name)) { in PreprocessEdgesBetweenOutsideCompilations()
Dextract_outside_compilation_pass.cc46 if (HasNodeAttr(n->def(), kXlaHasHostTransferAttrName)) { in HostGraphControlRetMapping()
399 if (!HasNodeAttr(n->def(), kXlaHasHostTransferAttrName)) { in ResetDeviceOrdinalToPlaceholderValue()
424 } else if (HasNodeAttr(n->def(), "_device_ordinal")) { in ResetDeviceOrdinalToPlaceholderValue()
907 if (!HasNodeAttr(n->def(), kXlaHasHostTransferAttrName)) { in PostprocessLiftedArgs()
1040 if (HasNodeAttr(copy->def(), kXlaHasHostTransferAttrName)) { in ConstructHostGraph()
1151 HasNodeAttr(copy->def(), "_xla_host_transfer_sequencer")) { in ExpandHostGraphIntoMainGraph()
2104 !HasNodeAttr(n->def(), outside_compilation_attr_name)) { in CopyOutsideCompilationConstNodes()
2113 !HasNodeAttr(e->dst()->def(), outside_compilation_attr_name)) { in CopyOutsideCompilationConstNodes()
2135 !HasNodeAttr(e->dst()->def(), outside_compilation_attr_name)) { in CopyOutsideCompilationConstNodes()
2204 if (HasNodeAttr(n->def(), kXlaConnectedToXlaComputationAttrName)) { in operator ()()
[all …]
Dextract_outside_compilation_pass_test.cc99 EXPECT_TRUE(HasNodeAttr(add_node->def(), "_xla")); in TEST()
100 EXPECT_TRUE(HasNodeAttr(add_node->def(), "_oc")); in TEST()
356 } else if (HasNodeAttr(n->def(), "_xla_host_transfer_sequencer")) { in TEST_F()
/external/tensorflow/tensorflow/core/grappler/utils/
Dcanonicalizer.cc59 HasNodeAttr(*node, "value")) { in CompressConstants()
/external/tensorflow/tensorflow/core/grappler/optimizers/data/
Dshuffle_and_repeat_fusion.cc91 if (HasNodeAttr(shuffle_node, "reshuffle_each_iteration") && in OptimizeAndCollectStats()
Dslack.cc83 if (HasNodeAttr(*dataset_node, "slack_period")) { in RecursivelyHandleOp()
/external/tensorflow/tensorflow/core/framework/
Dnode_def_util.h185 bool HasNodeAttr(const NodeDef& node_def, StringPiece attr_name);
Dnode_def_util.cc313 bool HasNodeAttr(const NodeDef& node_def, StringPiece attr_name) { in HasNodeAttr() function
Dop_kernel.cc234 return HasNodeAttr(def(), attr_name); in HasAttr()
/external/tensorflow/tensorflow/core/grappler/
Dutils.cc550 if (!HasNodeAttr(node, key)) { in CheckAttrExists()
/external/tensorflow/tensorflow/core/common_runtime/
Dgraph_execution_state.cc477 if (node.op() == "Const" && HasNodeAttr(node, "value")) { in GetFeedShapeAndTypeFromAttribute()
/external/tensorflow/tensorflow/core/grappler/optimizers/
Dscoped_allocator_optimizer.cc344 if (HasNodeAttr(*node_def, name)) { in ExtendNodeAttr()
Darithmetic_optimizer.cc444 return HasNodeAttr(node, tag); in IsMarkedWithTag()
Dconstant_folding.cc3734 HasNodeAttr(*node, "value")) { in CompressConstants()
/external/tensorflow/tensorflow/core/graph/
Dmkl_layout_pass.cc2676 bool has_padding_list = HasNodeAttr(orig_node->def(), "padding_list"); in CopyAttrsQuantizedConv2D()