Searched refs:HasNodeAttr (Results 1 – 17 of 17) sorted by relevance
| /external/tensorflow/tensorflow/compiler/tf2xla/ |
| D | side_effect_util.cc | 30 if (!HasNodeAttr(node->def(), kXlaHasHostTransferAttrName)) { in SetDeviceOrdinalAttributeForNode()
61 } else if (HasNodeAttr(node->def(), "device_ordinal")) { in SetDeviceOrdinalAttributeForNode()
|
| D | sharding_util.cc | 31 if (!HasNodeAttr(node_def, kShardingAttribute)) { in GetShardingFromNodeDef()
|
| /external/tensorflow/tensorflow/compiler/jit/ |
| D | encapsulate_util.cc | 199 HasNodeAttr(n->def(), kOutsideCompilationOriginalNodeAttrName)) { in PostprocessDataEdgesBetweenOutsideCompilations()
342 if (HasNodeAttr(e->dst()->def(), outside_compilation_attr_name)) { in PreprocessEdgesBetweenOutsideCompilations()
347 if (HasNodeAttr(e->src()->def(), outside_compilation_attr_name)) { in PreprocessEdgesBetweenOutsideCompilations()
|
| D | extract_outside_compilation_pass.cc | 412 if (!HasNodeAttr(n->def(), kXlaHasHostTransferAttrName)) { in ResetDeviceOrdinalToPlaceholderValue()
437 } else if (HasNodeAttr(n->def(), "device_ordinal")) { in ResetDeviceOrdinalToPlaceholderValue()
554 if (HasNodeAttr(copy->def(), kXlaHasHostTransferAttrName)) { in ConstructHostGraph()
669 HasNodeAttr(copy->def(), "_xla_host_transfer_sequencer")) { in ExpandHostGraphIntoMainGraph()
1535 if (HasNodeAttr(n->def(), kXlaConnectedToXlaComputationAttrName)) { in operator ()()
1539 if (HasNodeAttr(n->def(), kXlaConnectedFromXlaComputationAttrName)) { in operator ()()
1609 if (HasNodeAttr(n->def(), outside_compilation_attr_name)) { in ExtractOutsideCompilationForFunction()
1639 if (HasNodeAttr(n->def(), "_outside_compilation_subgraph")) { in ExtractOutsideCompilationForFunction()
|
| 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()
365 } else if (HasNodeAttr(n->def(), "_xla_host_transfer_sequencer")) { in TEST_F()
|
| /external/tensorflow/tensorflow/core/grappler/optimizers/data/vectorization/ |
| D | unpack_vectorizer.cc | 33 if (HasNodeAttr(node.def(), "axis")) { in Vectorize()
|
| /external/tensorflow/tensorflow/tools/graph_transforms/ |
| D | fuse_quantized_convolution.cc | 123 if (HasNodeAttr(quantized_conv2D_node, "dilations")) in FuseQuantizedConvolutionAndRequantize()
|
| /external/tensorflow/tensorflow/core/framework/ |
| D | node_def_util.h | 176 bool HasNodeAttr(const NodeDef& node_def, StringPiece attr_name);
|
| D | node_def_util.cc | 298 bool HasNodeAttr(const NodeDef& node_def, StringPiece attr_name) { in HasNodeAttr() function
|
| D | op_kernel.cc | 222 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/grappler/optimizers/ |
| D | meta_optimizer.cc | 110 HasNodeAttr(*node, "value")) { in CompressConstants()
|
| D | scoped_allocator_optimizer.cc | 204 if (HasNodeAttr(*node_def, name)) { in ExtendNodeAttr()
|
| D | constant_folding.cc | 3413 HasNodeAttr(*node, "value")) { in CompressConstants()
|
| D | arithmetic_optimizer.cc | 421 return HasNodeAttr(node, tag); in IsMarkedWithTag()
|
| /external/tensorflow/tensorflow/core/common_runtime/ |
| D | graph_execution_state.cc | 426 if (node.op() == "Const" && HasNodeAttr(node, "value")) { in GetFeedShapeAndTypeFromAttribute()
|
| /external/tensorflow/tensorflow/core/graph/ |
| D | mkl_layout_pass.cc | 2392 bool has_padding_list = HasNodeAttr(orig_node->def(), "padding_list"); in CopyAttrsQuantizedConv2D()
|