Home
last modified time | relevance | path

Searched refs:GraphDefBuilder (Results 1 – 20 of 20) sorted by relevance

/external/tensorflow/tensorflow/core/graph/
Dgraph_def_builder.cc25 GraphDefBuilder::Options::Options(Graph* graph, Status* status) in Options()
27 GraphDefBuilder::Options::~Options() {} in ~Options()
29 GraphDefBuilder::Options GraphDefBuilder::Options::WithName( in WithName()
33 GraphDefBuilder::Options GraphDefBuilder::Options::WithDevice( in WithDevice()
37 GraphDefBuilder::Options GraphDefBuilder::Options::WithControlInput( in WithControlInput()
41 GraphDefBuilder::Options GraphDefBuilder::Options::WithControlInputs( in WithControlInputs()
45 GraphDefBuilder::Options GraphDefBuilder::Options::WithNameImpl( in WithNameImpl()
50 GraphDefBuilder::Options GraphDefBuilder::Options::WithDeviceImpl( in WithDeviceImpl()
55 GraphDefBuilder::Options GraphDefBuilder::Options::WithControlInputImpl( in WithControlInputImpl()
60 GraphDefBuilder::Options GraphDefBuilder::Options::WithControlInputsImpl( in WithControlInputsImpl()
[all …]
Dgraph_def_builder.h66 class GraphDefBuilder {
145 explicit GraphDefBuilder(
152 explicit GraphDefBuilder(
193 Node* SourceOp(const string& op_name, const GraphDefBuilder::Options& opts);
197 const GraphDefBuilder::Options& opts);
201 const GraphDefBuilder::Options& opts);
Dcollective_order_test.cc79 Node* CollectiveReduceNode(GraphDefBuilder* builder, Node* input, in CollectiveReduceNode()
111 GraphDefBuilder builder(GraphDefBuilder::kFailImmediately); in InitGraph()
167 GraphDefBuilder builder(GraphDefBuilder::kFailImmediately); in InitGraph2()
203 GraphDefBuilder builder(GraphDefBuilder::kFailImmediately); in InitGraphForPruning()
Dalgorithm_test.cc78 GraphDefBuilder b(GraphDefBuilder::kFailImmediately); in TEST()
130 GraphDefBuilder b(GraphDefBuilder::kFailImmediately); in TEST()
158 GraphDefBuilder b(GraphDefBuilder::kFailImmediately); in TEST()
Dgraph_def_builder_test.cc34 GraphDefBuilder builder(GraphDefBuilder::kFailImmediately); in TEST()
Dgraph_def_builder_util.h31 Status GraphDefBuilderToGraph(const GraphDefBuilder& builder, Graph* graph);
Dgraph_def_builder_util.cc21 Status GraphDefBuilderToGraph(const GraphDefBuilder& builder, Graph* graph) { in GraphDefBuilderToGraph()
Dsubgraph_test.cc355 GraphDefBuilder b(GraphDefBuilder::kFailImmediately); in BM_SubgraphHelper()
Dgraph_partition.cc425 const GraphDefBuilder::Options& bopts) { in AddControlSwitch()
435 const GraphDefBuilder::Options& bopts) { in AddControlNext()
442 Node* EmptyConst(const GraphDefBuilder::Options& options) { in EmptyConst()
457 const GraphDefBuilder::Options& bopts) { in AddControlConst()
498 GraphDefBuilder::Options bopts(g, &status); in AddControlLoop()
611 GraphDefBuilder::Options bopts(g, &status); in AddControlFlow()
/external/tensorflow/tensorflow/core/common_runtime/
Dplacer_test.cc219 Status BuildGraph(const GraphDefBuilder& builder, Graph* out_graph) { in BuildGraph()
320 GraphDefBuilder b(GraphDefBuilder::kFailImmediately); in TEST_F()
339 GraphDefBuilder b(GraphDefBuilder::kFailImmediately); in TEST_F()
358 GraphDefBuilder b(GraphDefBuilder::kFailImmediately); in TEST_F()
375 GraphDefBuilder b(GraphDefBuilder::kFailImmediately); in TEST_F()
435 GraphDefBuilder b(GraphDefBuilder::kFailImmediately); in TEST_F()
450 GraphDefBuilder b(GraphDefBuilder::kFailImmediately); in TEST_F()
465 GraphDefBuilder b(GraphDefBuilder::kFailImmediately); in TEST_F()
480 GraphDefBuilder b(GraphDefBuilder::kFailImmediately); in TEST_F()
495 GraphDefBuilder b(GraphDefBuilder::kFailImmediately); in TEST_F()
[all …]
/external/tensorflow/tensorflow/compiler/jit/
Dencapsulate_subgraphs_pass_test.cc47 Status AddGraphDefToFunctionLibrary(const GraphDefBuilder& graphdef_builder, in AddGraphDefToFunctionLibrary()
357 Node* Sequencer(const GraphDefBuilder::Options& opts, in Sequencer()
366 Node* Input(const GraphDefBuilder::Options& opts) { in Input()
370 Node* InputShaped(const GraphDefBuilder::Options& opts) { in InputShaped()
375 const GraphDefBuilder::Options& opts) { in KnownShapeBase()
390 const GraphDefBuilder::Options& opts) { in KnownShape()
394 Node* KeyPlaceholderShape(const GraphDefBuilder::Options& opts) { in KeyPlaceholderShape()
399 const GraphDefBuilder::Options& opts) { in KeyPlaceholder()
413 const GraphDefBuilder::Options& opts) { in RecvAtHost()
432 const GraphDefBuilder::Options& opts) { in SendFromHost()
[all …]
Dxla_fusion_optimizer_test.cc50 GraphDefBuilder builder(GraphDefBuilder::kFailImmediately); in TEST_F()
79 GraphDefBuilder builder(GraphDefBuilder::kFailImmediately); in TEST_F()
108 GraphDefBuilder builder(GraphDefBuilder::kFailImmediately); in TEST_F()
139 GraphDefBuilder builder(GraphDefBuilder::kFailImmediately); in TEST_F()
163 GraphDefBuilder builder(GraphDefBuilder::kFailImmediately); in TEST_F()
Dpartially_decluster_pass_test.cc130 GraphDefBuilder builder(GraphDefBuilder::kFailImmediately); in TEST()
166 GraphDefBuilder builder(GraphDefBuilder::kFailImmediately); in TEST()
195 GraphDefBuilder builder(GraphDefBuilder::kFailImmediately); in TEST()
225 GraphDefBuilder builder(GraphDefBuilder::kFailImmediately); in TEST()
256 GraphDefBuilder builder(GraphDefBuilder::kFailImmediately); in TEST()
449 GraphDefBuilder builder(GraphDefBuilder::kFailImmediately); in TEST()
Dmark_for_compilation_pass_test.cc92 GraphDefBuilder builder(GraphDefBuilder::kFailImmediately); in TEST()
118 GraphDefBuilder builder(GraphDefBuilder::kFailImmediately); in TEST()
139 GraphDefBuilder builder(GraphDefBuilder::kFailImmediately); in TEST()
161 GraphDefBuilder builder(GraphDefBuilder::kFailImmediately); in TEST()
181 GraphDefBuilder builder(GraphDefBuilder::kFailImmediately); in TEST()
218 GraphDefBuilder builder(GraphDefBuilder::kFailImmediately, &flib_def); in TEST()
248 GraphDefBuilder builder(GraphDefBuilder::kFailImmediately); in TEST()
316 GraphDefBuilder builder(GraphDefBuilder::kFailImmediately); in TEST()
384 GraphDefBuilder builder(GraphDefBuilder::kFailImmediately); in TEST()
418 GraphDefBuilder builder(GraphDefBuilder::kFailImmediately); in TEST()
[all …]
Dencapsulate_subgraphs_pass.cc1221 GraphDefBuilder::Options options(graph_out, /*status=*/nullptr); in AddHostComputeKeyPlaceholder()
1843 GraphDefBuilder::Options options(graph_out, /*status=*/nullptr); in AddDummyShapedNode()
/external/tensorflow/tensorflow/core/util/
Dequal_graph_def_test.cc35 Node* Input(const GraphDefBuilder::Options& opts) { in Input()
39 Node* Alternate(const GraphDefBuilder::Options& opts) { in Alternate()
44 const GraphDefBuilder::Options& opts) { in Combine()
51 : e_(GraphDefBuilder::kFailImmediately), in EqualGraphDefTest()
52 a_(GraphDefBuilder::kFailImmediately) {} in EqualGraphDefTest()
73 GraphDefBuilder e_;
74 GraphDefBuilder a_;
/external/tensorflow/tensorflow/core/framework/
Ddataset.cc195 std::unique_ptr<const GraphDefBuilder::Options> opts( in AddDataset()
196 new GraphDefBuilder::Options(b_->opts())); in AddDataset()
202 opts.reset(new GraphDefBuilder::Options( in AddDataset()
206 opts.reset(new GraphDefBuilder::Options( in AddDataset()
211 new GraphDefBuilder::Options(opts->WithAttr(attr.first, attr.second))); in AddDataset()
366 GraphDefBuilder b; in Save()
Ddataset.h51 class GraphDefBuilder; variable
88 explicit GraphDefBuilderWrapper(GraphDefBuilder* b) : b_(b) {} in GraphDefBuilderWrapper()
262 GraphDefBuilder* b_;
663 DatasetGraphDefBuilder(GraphDefBuilder* b) : GraphDefBuilderWrapper(b) {} in DatasetGraphDefBuilder()
/external/tensorflow/tensorflow/core/kernels/data/
Ddataset_ops.cc33 GraphDefBuilder b; in Compute()
Dgraph_rewrite_dataset.cc33 GraphDefBuilder b; in Optimize()