Searched refs:tf_node (Results 1 – 2 of 2) sorted by relevance
109 const Node* tf_node() const { return node_; } in tf_node() function in tensorflow::tensorrt::segment::__anon8f933a440111::SimpleNode674 node == nullptr ? nullptr : node->tf_node(), in AddSegmentForNode()739 << "(Op type: " << node->tf_node()->type_string() << "), " in SegmentGraph()742 unsupported_ops.emplace(node->tf_node()->type_string()); in SegmentGraph()747 GetDeviceParsedName(node->tf_node()); in SegmentGraph()756 node->tf_node())) { in SegmentGraph()762 node->tf_node())) { in SegmentGraph()765 const Status status = candidate_fn(node->tf_node()); in SegmentGraph()768 } else if (tftrt_op_denylist.count(node->tf_node()->type_string())) { in SegmentGraph()772 << "(Op type: " << node->tf_node()->type_string() << "), " in SegmentGraph()[all …]
175 explicit TFAttrs(const NodeDef& tf_node) { in TFAttrs() argument176 for (const auto& attr : tf_node.attr()) { in TFAttrs()