Home
last modified time | relevance | path

Searched refs:NodeDebugInfo (Results 1 – 25 of 29) sorted by relevance

12

/external/tensorflow/tensorflow/core/graph/
Dgraph_node_util.cc27 string FormatNodeForError(const NodeDebugInfo& debug_info) { in FormatNodeForError()
34 return FormatNodeForError(NodeDebugInfo(node)); in FormatNodeForError()
47 void GetMergedOriginalNodeNames(const NodeDebugInfo& from, in GetMergedOriginalNodeNames()
48 const NodeDebugInfo& to, in GetMergedOriginalNodeNames()
59 void MergeDebugInfo(const NodeDebugInfo& from, Node* to) { in MergeDebugInfo()
61 GetMergedOriginalNodeNames(from, NodeDebugInfo(*to), &names); in MergeDebugInfo()
65 void MergeDebugInfo(const NodeDebugInfo& from, NodeDef* to) { in MergeDebugInfo()
67 GetMergedOriginalNodeNames(from, NodeDebugInfo(*to), &names); in MergeDebugInfo()
76 MergeDebugInfo(NodeDebugInfo(from), to); in MergeDebugInfo()
Dgraph_node_util.h24 struct NodeDebugInfo;
41 void MergeDebugInfo(const NodeDebugInfo& from, Node* to);
42 void MergeDebugInfo(const NodeDebugInfo& from, NodeDef* to);
Dgraph.cc323 NodeDebugInfo::NodeDebugInfo(const Node& n) : NodeDebugInfo(n.def()) {} in NodeDebugInfo() function in tensorflow::NodeDebugInfo
324 NodeDebugInfo::NodeDebugInfo(const NodeDef& ndef) in NodeDebugInfo() function in tensorflow::NodeDebugInfo
325 : NodeDebugInfo(ndef.name(), ndef.has_experimental_debug_info(), in NodeDebugInfo()
327 NodeDebugInfo::NodeDebugInfo( in NodeDebugInfo() function in tensorflow::NodeDebugInfo
Dgraph.h353 struct NodeDebugInfo { struct
357 NodeDebugInfo(const Node& n); argument
358 NodeDebugInfo(const NodeDef& ndef);
359 NodeDebugInfo(StringPiece node_name, bool has_experimental_debug_info,
Dgraph_partition.cc218 NodeDebugInfo(*src)); in AddSend()
243 NodeDebugInfo(*src)); in AddSend()
302 NodeDebugInfo(*src)); in AddRecv()
315 NodeDebugInfo(*src)); in AddRecv()
326 NodeDebugInfo(*src)); in AddRecv()
1018 MergeDebugInfo(NodeDebugInfo(dst->def()), dst_def); in Partition()
Dnode_builder.h81 const NodeDebugInfo* debug = nullptr);
Dnode_builder.cc42 const NodeDebugInfo* debug) in NodeBuilder()
Doptimizer_cse.cc324 MergeDebugInfo(NodeDebugInfo(*n), *candidate); in Optimize()
/external/tensorflow/tensorflow/core/tpu/graph_rewrite/
Dincomplete_nodedef_builder.h33 const NodeDebugInfo& debug);
44 const NodeDebugInfo& debug);
47 const NodeDebugInfo& debug, int n);
50 const NodeDebugInfo& debug);
Dincomplete_nodedef_builder.cc25 const NodeDebugInfo& debug) { in IncompleteNodeDefBuilder()
56 const string& name, const DataType& type, const NodeDebugInfo& debug) { in Identity()
61 const string& name, const DataType& type, const NodeDebugInfo& debug, in Merge()
69 const string& name, const DataType& type, const NodeDebugInfo& debug) { in Switch()
Dcond_builder.h41 CondBuilder(string name, string device, const NodeDebugInfo& debug,
59 const string& device, const NodeDebugInfo& debug,
Ddistributed_tpu_configuration_rewrite_pass.cc87 MergeDebugInfo(NodeDebugInfo(configuration_node->def()), &host_config_def); in AddHostConfigNode()
110 MergeDebugInfo(NodeDebugInfo(host_configuration_nodes[0]->def()), in AddWaitNode()
133 MergeDebugInfo(NodeDebugInfo(wait_node->def()), &global_tpu_array_def); in AddGlobalTPUArrayNode()
156 MergeDebugInfo(NodeDebugInfo(sync_node_def), &sync_def); in AddSynchronizationNode()
192 MergeDebugInfo(NodeDebugInfo(shutdown_node_def), &shutdown_def); in AddShutdownNode()
219 MergeDebugInfo(NodeDebugInfo(post_disconnect_node->def()), in AddHostDisconnectNode()
Dcond_builder.cc25 CondBuilder::CondBuilder(string name, string device, const NodeDebugInfo& debug, in CondBuilder()
74 const string& device, const NodeDebugInfo& debug, in AddInput()
/external/tensorflow/tensorflow/core/framework/
Dnode_def_builder.h70 const NodeDebugInfo* debug = nullptr);
72 const NodeDebugInfo& debug);
Dnode_def_builder.cc42 const NodeDebugInfo* debug) { in NodeDefBuilder()
55 const NodeDebugInfo& debug) in NodeDefBuilder()
/external/tensorflow/tensorflow/core/common_runtime/
Daccumulate_n_optimizer.cc110 NodeDebugInfo debug_info(*n); in RewriteIntoTempVariable()
256 NodeDebugInfo debug_info(*n); in RewriteIntoAddN()
Dlower_case_op.cc94 NodeDebugInfo debug_info_;
148 NodeDebugInfo debug_info(*src); in AddInput()
Dlower_if_op.cc108 NodeDebugInfo debug_info_;
181 NodeDebugInfo debug_info(*src); in AddInput()
Dparallel_concat_optimizer.cc51 NodeDebugInfo debug_info(*n); in Run()
Dplacer_inspection_required_ops_utils.cc183 MergeDebugInfo(NodeDebugInfo(*node), &identity_def); in AddInputIdentity()
222 MergeDebugInfo(NodeDebugInfo(*node), &identity_def); in AddOutputIdentities()
Dlower_while_op.cc154 NodeDebugInfo debug_info_;
/external/tensorflow/tensorflow/compiler/jit/
Dpartially_decluster_pass.cc139 MergeDebugInfo(NodeDebugInfo(n->def()), &ndef); in PartiallyDeclusterNode()
Dencapsulate_subgraphs_pass.cc482 NodeDebugInfo(src_node->def())); in RecordArg()
527 NodeDebugInfo(src_node->def())); in RecordResult()
1018 NodeDebugInfo debug_info(*src_node); in AddDummyShapedNode()
Dencapsulate_xla_computations_pass.cc319 MergeDebugInfo(NodeDebugInfo(launch->def()), &def); in BuildXlaLaunchOps()
/external/tensorflow/tensorflow/compiler/tf2xla/
Dtf2xla_util.cc380 MergeDebugInfo(NodeDebugInfo(*existing), gd.mutable_node(0)); in AddPlaceholdersForFeeds()
627 NodeDebugInfo debug_info(*node); in RewriteAssociatedFunction()

12