Home
last modified time | relevance | path

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

12

/external/tensorflow/tensorflow/core/graph/
Dgraph_node_util.cc52 void MergeDebugInfo(const NodeDebugInfo& from, Node* to_node) { in MergeDebugInfo()
53 NodeDebugInfo to = NodeDebugInfo(*to_node); in MergeDebugInfo()
66 void MergeDebugInfo(const NodeDebugInfo& from, NodeDef* to_node_def) { in MergeDebugInfo()
67 NodeDebugInfo to = NodeDebugInfo(*to_node_def); in MergeDebugInfo()
87 MergeDebugInfo(NodeDebugInfo(from_node_def), to_node_def); 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.cc347 NodeDebugInfo::NodeDebugInfo(const Node& n) : NodeDebugInfo(n.def()) {} in NodeDebugInfo() function in tensorflow::NodeDebugInfo
348 NodeDebugInfo::NodeDebugInfo(const NodeDef& ndef) in NodeDebugInfo() function in tensorflow::NodeDebugInfo
349 : NodeDebugInfo(ndef.name(), ndef.has_experimental_debug_info(), in NodeDebugInfo()
351 NodeDebugInfo::NodeDebugInfo( in NodeDebugInfo() function in tensorflow::NodeDebugInfo
Dgraph.h372 struct NodeDebugInfo { struct
377 NodeDebugInfo(const Node& n); argument
378 NodeDebugInfo(const NodeDef& ndef);
379 NodeDebugInfo(StringPiece node_name, bool has_experimental_debug_info,
Dgraph_partition.cc217 NodeDebugInfo(*src)); in AddSend()
242 NodeDebugInfo(*src)); in AddSend()
301 NodeDebugInfo(*src)); in AddRecv()
314 NodeDebugInfo(*src)); in AddRecv()
325 NodeDebugInfo(*src)); in AddRecv()
1019 MergeDebugInfo(NodeDebugInfo(dst->def()), dst_def); in Partition()
Dnode_builder.h81 const NodeDebugInfo* debug = nullptr);
Dnode_builder.cc47 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()
Ddistributed_tpu_configuration_rewrite_pass.cc89 MergeDebugInfo(NodeDebugInfo(configuration_node->def()), &host_config_def); in AddHostConfigNode()
109 MergeDebugInfo(NodeDebugInfo(host_configuration_nodes[0]->def()), in AddWaitNode()
128 MergeDebugInfo(NodeDebugInfo(wait_node->def()), &global_tpu_array_def); in AddGlobalTPUArrayNode()
148 MergeDebugInfo(NodeDebugInfo(sync_node_def), &sync_def); in AddSynchronizationNode()
180 MergeDebugInfo(NodeDebugInfo(shutdown_node_def), &shutdown_def); in AddShutdownNode()
203 MergeDebugInfo(NodeDebugInfo(post_disconnect_node->def()), in AddHostDisconnectNode()
Dcond_builder.h41 CondBuilder(string name, string device, const NodeDebugInfo& debug,
59 const string& device, const NodeDebugInfo& debug,
Dcond_builder.cc25 CondBuilder::CondBuilder(string name, string device, const NodeDebugInfo& debug, in CondBuilder()
74 const string& device, const NodeDebugInfo& debug, in AddInput()
Dconfigure_tpu_embedding_rewrite_pass.cc57 MergeDebugInfo(NodeDebugInfo(sync_node_def), &sync_def); in AddSynchronizationNode()
90 MergeDebugInfo(NodeDebugInfo(input_nodes[0]->def()), &node_def); in AddSetupPropagationEmbeddingNode()
/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/
Dlower_case_op.cc94 NodeDebugInfo debug_info_;
148 NodeDebugInfo debug_info(*src); in AddInput()
Daccumulate_n_optimizer.cc110 NodeDebugInfo debug_info(*n); in RewriteIntoTempVariable()
256 NodeDebugInfo debug_info(*n); in RewriteIntoAddN()
Dlower_if_op.cc108 NodeDebugInfo debug_info_;
181 NodeDebugInfo debug_info(*src); in AddInput()
Dplacer_inspection_required_ops_utils.cc179 MergeDebugInfo(NodeDebugInfo(*node), &identity_def); in AddInputIdentity()
214 MergeDebugInfo(NodeDebugInfo(*node), &identity_def); in AddOutputIdentities()
Dparallel_concat_optimizer.cc51 NodeDebugInfo debug_info(*n); in Run()
Dlower_while_op.cc157 NodeDebugInfo debug_info_;
/external/tensorflow/tensorflow/compiler/jit/
Dpartially_decluster_pass.cc139 MergeDebugInfo(NodeDebugInfo(n->def()), &ndef); in PartiallyDeclusterNode()
Dencapsulate_subgraphs_pass.cc463 NodeDebugInfo(src_node->def())); in RecordArg()
506 NodeDebugInfo(src_node->def())); in RecordResult()
992 NodeDebugInfo debug_info(*src_node); in AddDummyShapedNode()
Dencapsulate_xla_computations_pass.cc318 MergeDebugInfo(NodeDebugInfo(launch->def()), &def); in BuildXlaLaunchOps()

12