| /external/tensorflow/tensorflow/compiler/tf2xla/ |
| D | side_effect_util.cc | 38 if (!HasNodeAttr(node->def(), kXlaHasHostTransferAttrName)) { in SetDeviceOrdinalAttributeForNode()
69 } else if (HasNodeAttr(node->def(), "_device_ordinal")) { in SetDeviceOrdinalAttributeForNode()
|
| D | sharding_util.cc | 152 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/core/tpu/graph_rewrite/ |
| D | encapsulate_tpu_computations_pass.cc | 297 (HasNodeAttr(n->def(), "_tpu_input_identity") || in RemoveIdentityNodesForArgRetval()
298 HasNodeAttr(n->def(), "_tpu_output_identity"))) { in RemoveIdentityNodesForArgRetval()
374 (!HasNodeAttr(e->src()->def(), outside_compilation_attr_name) || in MoveHeadOutsideCompilationToHost()
375 !HasNodeAttr(e->src()->def(), kOnlyArgOrOcInputAttrName))) { in MoveHeadOutsideCompilationToHost()
380 if (HasNodeAttr(n->def(), outside_compilation_attr_name) && in MoveHeadOutsideCompilationToHost()
382 !HasNodeAttr(n->def(), kXlaIsPlaceholderForArg)) { in MoveHeadOutsideCompilationToHost()
399 HasNodeAttr(e->dst()->def(), outside_compilation_attr_name) && in MoveHeadOutsideCompilationToHost()
400 !HasNodeAttr(e->dst()->def(), kOnlyArgOrOcInputAttrName)) { in MoveHeadOutsideCompilationToHost()
1015 (!HasNodeAttr(e->dst()->def(), outside_compilation_attr_name) || in MoveTailOutsideCompilationToHost()
1016 !HasNodeAttr(e->dst()->def(), kOnlyRetOrOcOutputAttrName))) { in MoveTailOutsideCompilationToHost()
[all …]
|
| D | distributed_tpu_rewrite_pass.cc | 300 TF_RET_CHECK(HasNodeAttr(node->def(), kXlaHasHostTransferAttrName)); in SetNodeDeviceForTPUCommunication()
3479 if (HasNodeAttr(image->def(), kXlaHasHostTransferAttrName)) { in CopyOutsideCompilationNodes()
3540 if (HasNodeAttr(node->def(), kXlaHasHostTransferAttrName)) { in ReplicateOutsideCompilationNodes()
3718 if (!HasNodeAttr(n->def(), kOutsideCompilationAttr)) { in LowerOutsideCompilationFunctionalNodes()
3770 HasNodeAttr(n->def(), kXlaHasHostTransferAttrName); in LowerOutsideCompilationFunctionalNodes()
3872 HasNodeAttr(n->def(), kXlaOutsideCompilationInputsAttrName)) { in LowerOutsideCompilationFunctionalNodes()
|
| /external/tensorflow/tensorflow/core/grappler/utils/ |
| D | canonicalizer.cc | 59 HasNodeAttr(*node, "value")) { in CompressConstants()
|
| /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()
401 if (!HasNodeAttr(n->def(), kXlaHasHostTransferAttrName)) { in ResetDeviceOrdinalToPlaceholderValue()
426 } else if (HasNodeAttr(n->def(), "_device_ordinal")) { in ResetDeviceOrdinalToPlaceholderValue()
942 if (!HasNodeAttr(n->def(), kXlaHasHostTransferAttrName)) { in PostprocessLiftedArgs()
1075 if (HasNodeAttr(copy->def(), kXlaHasHostTransferAttrName)) { in ConstructHostGraph()
1186 HasNodeAttr(copy->def(), "_xla_host_transfer_sequencer")) { in ExpandHostGraphIntoMainGraph()
2157 !HasNodeAttr(n->def(), outside_compilation_attr_name)) { in CopyOutsideCompilationConstNodes()
2166 !HasNodeAttr(e->dst()->def(), outside_compilation_attr_name)) { in CopyOutsideCompilationConstNodes()
2188 !HasNodeAttr(e->dst()->def(), outside_compilation_attr_name)) { in CopyOutsideCompilationConstNodes()
2257 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/optimizers/data/ |
| D | slack.cc | 83 if (HasNodeAttr(*dataset_node, "slack_period")) { in RecursivelyHandleOp()
|
| /external/tensorflow/tensorflow/core/framework/ |
| D | node_def_util.h | 193 bool HasNodeAttr(const NodeDef& node_def, StringPiece attr_name);
|
| D | node_def_util_test.cc | 662 EXPECT_FALSE(HasNodeAttr(node_def, kColocationAttrName)); in TEST()
|
| D | node_def_util.cc | 326 bool HasNodeAttr(const NodeDef& node_def, StringPiece attr_name) { in HasNodeAttr() function
|
| D | op_kernel.cc | 247 return HasNodeAttr(def(), attr_name); in HasAttr()
|
| /external/tensorflow/tensorflow/core/grappler/ |
| D | utils.cc | 478 if (!HasNodeAttr(node, key)) { in CheckAttrExists()
|
| /external/tensorflow/tensorflow/core/common_runtime/ |
| D | graph_execution_state.cc | 485 if (node.op() == "Const" && HasNodeAttr(node, "value")) { in GetFeedShapeAndTypeFromAttribute()
|
| D | mkl_layout_pass.cc | 2747 bool has_padding_list = HasNodeAttr(orig_node->def(), "padding_list"); in CopyAttrsQuantizedConv2D()
|
| /external/tensorflow/tensorflow/core/grappler/optimizers/ |
| D | scoped_allocator_optimizer.cc | 350 if (HasNodeAttr(*node_def, name)) { in ExtendNodeAttr()
|
| D | arithmetic_optimizer.cc | 446 return HasNodeAttr(node, tag); in IsMarkedWithTag()
|