Searched refs:tf_node (Results 1 – 2 of 2) sorted by relevance
102 const Node* tf_node() const { return node_; } in tf_node() function in tensorflow::tensorrt::segment::SimpleNode430 << "(Op type: " << node->tf_node()->type_string() << "), " in SegmentGraph()433 unsupported_ops.emplace(node->tf_node()->type_string()); in SegmentGraph()437 const Status status = candidate_fn(node->tf_node()); in SegmentGraph()440 << "(Op type: " << node->tf_node()->type_string() << "), " in SegmentGraph()443 unsupported_ops.emplace(node->tf_node()->type_string()); in SegmentGraph()558 sg_map[u.ParentValue()->name()].insert(u.Value()->tf_node()); in SegmentGraph()559 auto tf_node = u.Value()->tf_node(); in SegmentGraph() local565 if (tf_node->has_assigned_device_name()) { in SegmentGraph()567 tf_node->assigned_device_name()); in SegmentGraph()[all …]
121 explicit TFAttrs(const NodeDef& tf_node) { in TFAttrs() argument122 for (const auto& attr : tf_node.attr()) { in TFAttrs()