[update]
removing commented code & unused code
This commit is contained in:
parent
97aa1856bc
commit
af4782345c
@ -19,7 +19,6 @@ limitations under the License.
|
||||
#include "cuda/include/cuda_runtime_api.h"
|
||||
#include "tensorflow/contrib/tensorrt/log/trt_logger.h"
|
||||
#include "tensorflow/core/platform/logging.h"
|
||||
//#include "tensorflow/core/framework/device_base.h"
|
||||
#include "tensorflow/core/platform/stream_executor.h"
|
||||
|
||||
namespace tensorflow {
|
||||
|
@ -223,15 +223,6 @@ tensorflow::Status SegmentGraph(
|
||||
}
|
||||
}
|
||||
|
||||
// Cleanup the graph to remove disconnected nodes before outputting
|
||||
if (VLOG_IS_ON(2)) {
|
||||
for (tensorflow::Node* node : graph.nodes()) {
|
||||
if ((node->in_edges().size() == 0) && (node->out_edges().size() == 0)) {
|
||||
graph.RemoveNode(node);
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
// Convert the segments into the expected return format
|
||||
for (const auto& itr : sg_map) {
|
||||
const auto& segment_node_names = itr.second;
|
||||
|
Loading…
Reference in New Issue
Block a user