Home
last modified time | relevance | path

Searched refs:node_from (Results 1 – 2 of 2) sorted by relevance

/external/tensorflow/tensorflow/compiler/jit/
Dxla_fusion_optimizer.cc252 Node* node_from = graph.FindNodeId(from); in Optimize() local
253 if (node_from->IsControlFlow()) { in Optimize()
258 node_from->type_string()); in Optimize()
273 if (node_from->def().device() != node_to->def().device()) { in Optimize()
274 VLOG(2) << "Devices " << node_from->def().device() << " " in Optimize()
276 VLOG(2) << "Device names " << node_from->assigned_device_name() << " " in Optimize()
283 if (clusters[from].Size() == 1 && IsShapeConsumerOp(*node_from)) { in Optimize()
Dmark_for_compilation_pass.cc1150 Node* node_from = graph->FindNodeId(from); in RunImpl() local
1151 if (node_from->IsControlFlow()) { in RunImpl()
1156 node_from->type_string()); in RunImpl()
1159 if (isolated_nodes.count(node_from)) { in RunImpl()
1197 GetNodeAttr(node_from->attrs(), kXlaScopeAttr, &from_scope).ok() && in RunImpl()
1205 if (clusters[from].Size() == 1 && IsShapeConsumerOp(*node_from)) { in RunImpl()