Home
last modified time | relevance | path

Searched refs:GraphDefBuilder (Results 1 – 22 of 22) 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.cc81 GraphDefBuilder b(GraphDefBuilder::kFailImmediately); in TEST()
133 GraphDefBuilder b(GraphDefBuilder::kFailImmediately); in TEST()
161 GraphDefBuilder b(GraphDefBuilder::kFailImmediately); in TEST()
Dgraph_def_builder_test.cc34 GraphDefBuilder builder(GraphDefBuilder::kFailImmediately); in TEST()
Dsubgraph_test.cc355 GraphDefBuilder b(GraphDefBuilder::kFailImmediately); in BM_SubgraphHelper()
Dgraph_partition.cc452 const GraphDefBuilder::Options& bopts) { in AddControlSwitch()
462 const GraphDefBuilder::Options& bopts) { in AddControlNext()
469 Node* EmptyConst(const GraphDefBuilder::Options& options) { in EmptyConst()
484 const GraphDefBuilder::Options& bopts) { in AddControlConst()
525 GraphDefBuilder::Options bopts(g, &status); in AddControlLoop()
638 GraphDefBuilder::Options bopts(g, &status); in AddControlFlow()
/external/tensorflow/tensorflow/core/common_runtime/
Dplacer_test.cc243 Status BuildGraph(const GraphDefBuilder& builder, Graph* out_graph) { in BuildGraph()
408 GraphDefBuilder b(GraphDefBuilder::kFailImmediately); in TEST_F()
427 GraphDefBuilder b(GraphDefBuilder::kFailImmediately); in TEST_F()
447 GraphDefBuilder b(GraphDefBuilder::kFailImmediately); in TEST_F()
471 GraphDefBuilder b(GraphDefBuilder::kFailImmediately); in TEST_F()
488 GraphDefBuilder b(GraphDefBuilder::kFailImmediately); in TEST_F()
550 GraphDefBuilder b(GraphDefBuilder::kFailImmediately); in TEST_F()
565 GraphDefBuilder b(GraphDefBuilder::kFailImmediately); in TEST_F()
580 GraphDefBuilder b(GraphDefBuilder::kFailImmediately); in TEST_F()
595 GraphDefBuilder b(GraphDefBuilder::kFailImmediately); in TEST_F()
[all …]
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()
/external/tensorflow/tensorflow/compiler/jit/
Dencapsulate_subgraphs_pass_test.cc48 Status AddGraphDefToFunctionLibrary(const GraphDefBuilder& graphdef_builder, in AddGraphDefToFunctionLibrary()
345 Node* Sequencer(const GraphDefBuilder::Options& opts, in Sequencer()
354 Node* Input(const GraphDefBuilder::Options& opts) { in Input()
358 Node* InputShaped(const GraphDefBuilder::Options& opts) { in InputShaped()
363 const GraphDefBuilder::Options& opts) { in KnownShapeBase()
378 const GraphDefBuilder::Options& opts) { in KnownShape()
382 Node* KeyPlaceholderShape(const GraphDefBuilder::Options& opts) { in KeyPlaceholderShape()
387 const GraphDefBuilder::Options& opts) { in KeyPlaceholder()
402 const GraphDefBuilder::Options& opts) { in RecvAtHost()
422 const GraphDefBuilder::Options& opts) { in SendFromHost()
[all …]
Dcompilability_check_util_test.cc117 GraphDefBuilder builder(GraphDefBuilder::kFailImmediately); in TEST_F()
160 GraphDefBuilder builder(GraphDefBuilder::kFailImmediately, flib_def_.get()); in TEST_F()
211 GraphDefBuilder builder(GraphDefBuilder::kFailImmediately, flib_def_.get()); in TEST_F()
423 GraphDefBuilder b(GraphDefBuilder::kFailImmediately); in TEST_F()
451 GraphDefBuilder b(GraphDefBuilder::kFailImmediately); in TEST_F()
479 GraphDefBuilder b(GraphDefBuilder::kFailImmediately); in TEST_F()
Dcluster_scoping_pass_test.cc66 Node* BuildStageNode(GraphDefBuilder& builder, string name, in BuildStageNode()
94 GraphDefBuilder builder(GraphDefBuilder::kFailImmediately); in TEST()
138 GraphDefBuilder builder(GraphDefBuilder::kFailImmediately); in TEST()
Dpartially_decluster_pass_test.cc127 GraphDefBuilder builder(GraphDefBuilder::kFailImmediately); in TEST()
163 GraphDefBuilder builder(GraphDefBuilder::kFailImmediately); in TEST()
192 GraphDefBuilder builder(GraphDefBuilder::kFailImmediately); in TEST()
222 GraphDefBuilder builder(GraphDefBuilder::kFailImmediately); in TEST()
253 GraphDefBuilder builder(GraphDefBuilder::kFailImmediately); in TEST()
415 GraphDefBuilder builder(GraphDefBuilder::kFailImmediately); in TEST()
Dmark_for_compilation_pass_test.cc96 GraphDefBuilder builder(GraphDefBuilder::kFailImmediately); in TEST()
121 GraphDefBuilder builder(GraphDefBuilder::kFailImmediately); in TEST()
141 GraphDefBuilder builder(GraphDefBuilder::kFailImmediately); in TEST()
162 GraphDefBuilder builder(GraphDefBuilder::kFailImmediately); in TEST()
181 GraphDefBuilder builder(GraphDefBuilder::kFailImmediately); in TEST()
256 GraphDefBuilder builder(GraphDefBuilder::kFailImmediately, &flib_def); in TEST()
293 GraphDefBuilder builder(GraphDefBuilder::kFailImmediately, &flib_def); in TEST()
382 GraphDefBuilder builder(GraphDefBuilder::kFailImmediately); in TEST()
449 GraphDefBuilder builder(GraphDefBuilder::kFailImmediately); in TEST()
482 GraphDefBuilder builder(GraphDefBuilder::kFailImmediately); in TEST()
[all …]
Dforce_xla_constants_on_host_pass_test.cc58 GraphDefBuilder b(GraphDefBuilder::kFailImmediately); in TEST()
Dencapsulate_subgraphs_pass.cc1010 GraphDefBuilder::Options options(graph_out, /*status=*/nullptr); in AddDummyShapedNode()
/external/tensorflow/tensorflow/compiler/tf2xla/
Dresource_util_test.cc87 GraphDefBuilder builder(GraphDefBuilder::kFailImmediately, &flib_def); in TEST()
122 GraphDefBuilder builder(GraphDefBuilder::kFailImmediately, &flib_def); in TEST()
162 GraphDefBuilder builder(GraphDefBuilder::kFailImmediately, &flib_def); in TEST()
205 GraphDefBuilder builder(GraphDefBuilder::kFailImmediately, &flib_def); in TEST()
254 GraphDefBuilder builder(GraphDefBuilder::kFailImmediately, &flib_def); in TEST()
317 GraphDefBuilder builder(GraphDefBuilder::kFailImmediately, &flib_def); in TEST()
393 GraphDefBuilder builder(GraphDefBuilder::kFailImmediately, &flib_def); in TEST()
450 GraphDefBuilder builder(GraphDefBuilder::kFailImmediately, &flib_def); in TEST()
500 GraphDefBuilder builder(GraphDefBuilder::kFailImmediately, &flib_def); in TEST()
/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.cc218 std::unique_ptr<const GraphDefBuilder::Options> opts( in AddDataset()
219 new GraphDefBuilder::Options(b_->opts())); in AddDataset()
225 opts.reset(new GraphDefBuilder::Options( in AddDataset()
229 opts.reset(new GraphDefBuilder::Options( in AddDataset()
234 new GraphDefBuilder::Options(opts->WithAttr(attr.first, attr.second))); in AddDataset()
Ddataset.h55 class GraphDefBuilder; variable
138 explicit GraphDefBuilderWrapper(GraphDefBuilder* b) : b_(b) {} in GraphDefBuilderWrapper()
261 GraphDefBuilder* builder() { return b_; } in builder()
290 GraphDefBuilder* b_;
914 explicit DatasetGraphDefBuilder(GraphDefBuilder* b) in DatasetGraphDefBuilder()
/external/tensorflow/tensorflow/core/kernels/data/
Dserialization_utils.cc98 GraphDefBuilder b; in AsGraphDef()