Searched refs:HasNodeAttr (Results 1 – 18 of 18) sorted by relevance
/external/tensorflow/tensorflow/compiler/tf2xla/ |
D | side_effect_util.cc | 47 if (!HasNodeAttr(node->def(), kXlaHasHostTransferAttrName)) { in SetDeviceOrdinalAttributeForNode() 78 } else if (HasNodeAttr(node->def(), "_device_ordinal")) { in SetDeviceOrdinalAttributeForNode()
|
D | sharding_util.cc | 31 if (!HasNodeAttr(node_def, kShardingAttribute)) { in GetShardingFromNodeDef()
|
/external/tensorflow/tensorflow/core/grappler/optimizers/data/vectorization/ |
D | unpack_vectorizer.cc | 33 if (HasNodeAttr(node.def(), "axis")) { in Vectorize()
|
/external/tensorflow/tensorflow/compiler/jit/ |
D | encapsulate_util.cc | 205 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()
|
D | extract_outside_compilation_pass.cc | 46 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 …]
|
D | extract_outside_compilation_pass_test.cc | 99 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/ |
D | canonicalizer.cc | 59 HasNodeAttr(*node, "value")) { in CompressConstants()
|
/external/tensorflow/tensorflow/core/grappler/optimizers/data/ |
D | shuffle_and_repeat_fusion.cc | 91 if (HasNodeAttr(shuffle_node, "reshuffle_each_iteration") && in OptimizeAndCollectStats()
|
D | slack.cc | 83 if (HasNodeAttr(*dataset_node, "slack_period")) { in RecursivelyHandleOp()
|
/external/tensorflow/tensorflow/core/framework/ |
D | node_def_util.h | 185 bool HasNodeAttr(const NodeDef& node_def, StringPiece attr_name);
|
D | node_def_util.cc | 313 bool HasNodeAttr(const NodeDef& node_def, StringPiece attr_name) { in HasNodeAttr() function
|
D | op_kernel.cc | 234 return HasNodeAttr(def(), attr_name); in HasAttr()
|
/external/tensorflow/tensorflow/core/grappler/ |
D | utils.cc | 550 if (!HasNodeAttr(node, key)) { in CheckAttrExists()
|
/external/tensorflow/tensorflow/core/common_runtime/ |
D | graph_execution_state.cc | 477 if (node.op() == "Const" && HasNodeAttr(node, "value")) { in GetFeedShapeAndTypeFromAttribute()
|
/external/tensorflow/tensorflow/core/grappler/optimizers/ |
D | scoped_allocator_optimizer.cc | 344 if (HasNodeAttr(*node_def, name)) { in ExtendNodeAttr()
|
D | arithmetic_optimizer.cc | 444 return HasNodeAttr(node, tag); in IsMarkedWithTag()
|
D | constant_folding.cc | 3734 HasNodeAttr(*node, "value")) { in CompressConstants()
|
/external/tensorflow/tensorflow/core/graph/ |
D | mkl_layout_pass.cc | 2676 bool has_padding_list = HasNodeAttr(orig_node->def(), "padding_list"); in CopyAttrsQuantizedConv2D()
|