segment.h
imports tensorflow/core/platform/types.hC declares a character or byte buffer
SHA-25604f0358b45888dc2de0585906c533371555acb113b24d2734fc18ff3d2b3cc6f
MaleculeH(Db)Md(In)
Evidence
1/* Copyright 2018 The TensorFlow Authors. All Rights Reserved.
2
3Licensed under the Apache License, Version 2.0 (the "License");
4you may not use this file except in compliance with the License.
5You may obtain a copy of the License at
6
7 http://www.apache.org/licenses/LICENSE-2.0
8
9Unless required by applicable law or agreed to in writing, software
10distributed under the License is …
11:42… IND, either express or implied.
12See the License for the specific language governing permissions and
13limitations under the License.
14================================================================ …
17:12… SORFLOW_COMPILER_TF2TENSORRT_SEGMENT_SEGMENT_H_
18
19#include <set>
20#include <vector>
21
22#include "absl/types/optional.h"
23#include "tensorflow/compiler/tf2tensorrt/segment/union_find.h"
24#include "tensorflow/core/framework/graph.pb.h"
25#include "tensorflow/core/graph/graph.h"
26#include "tensorflow/core/grappler/costs/graph_properties.h"
27#include "tensorflow/core/lib/core/status.h"
28#include "tensorflow/core/platform/types.h"
29
30#if GOOGLE_CUDA && GOOGLE_TENSORRT
31
32namespace tensorflow {
33namespace tensorrt {
34namespace segment {
35
36constexpr char kTftrtOpMaxBatchSizeAttr[] = "_tftrt_op_max_batch_size";
37
38struct SegmentOptions {
39 // This struct holds per graph segmenting parameters.
40 // Segment mu …
69:30… ctor<Segment>;
70
71// Get the subgraphs of a graph that can be handled by TensorRT.
72//
73// @param tf_graph Graph of the network.
74// @graph_properties is …