Home
last modified time | relevance | path

Searched refs:graphdef (Results 1 – 25 of 123) sorted by relevance

12345

/external/tensorflow/tensorflow/compiler/tf2xla/
Dresource_util_test.cc50 const GraphDef& graphdef, FunctionLibraryDefinition* flib_def, in AnalyzeAndVerify() argument
55 ConvertGraphDefToGraph(GraphConstructorOptions(), graphdef, graph.get())); in AnalyzeAndVerify()
110 GraphDef graphdef; in TEST() local
111 TF_EXPECT_OK(builder.ToGraphDef(&graphdef)); in TEST()
116 AnalyzeAndVerify(graphdef, &flib_def, expected); in TEST()
150 GraphDef graphdef; in TEST() local
151 TF_EXPECT_OK(builder.ToGraphDef(&graphdef)); in TEST()
156 AnalyzeAndVerify(graphdef, &flib_def, expected); in TEST()
193 GraphDef graphdef; in TEST() local
194 TF_EXPECT_OK(builder.ToGraphDef(&graphdef)); in TEST()
[all …]
/external/tensorflow/tensorflow/compiler/jit/tests/
Dauto_clustering_test_helper.cc85 Status AssertGraphDefIsUnclustered(const GraphDef& graphdef) { in AssertGraphDefIsUnclustered() argument
89 for (const NodeDef& node : graphdef.node()) { in AssertGraphDefIsUnclustered()
112 GraphDef graphdef, absl::string_view golden_summary_file_path) { in RunAutoClusteringTestImpl() argument
128 TF_RETURN_IF_ERROR(AssertGraphDefIsUnclustered(graphdef)); in RunAutoClusteringTestImpl()
140 runner.Run(auto_clustering_pass, std::move(graphdef), &next)); in RunAutoClusteringTestImpl()
141 graphdef = std::move(next); in RunAutoClusteringTestImpl()
144 TF_ASSIGN_OR_RETURN(string clustering_summary, SummarizeClustering(graphdef)); in RunAutoClusteringTestImpl()
168 GraphDef graphdef; in RunAutoClusteringTestWithPbtxt() local
170 ReadTextProto(Env::Default(), string(pbtxt_file_path), &graphdef)); in RunAutoClusteringTestWithPbtxt()
171 return RunAutoClusteringTestImpl(std::move(graphdef), in RunAutoClusteringTestWithPbtxt()
[all …]
Dauto_clustering_test_helper.h58 Status RunAutoClusteringTestImpl(GraphDef graphdef,
/external/tensorflow/tensorflow/compiler/mlir/tfjs/translate/
Djson_translate.cc71 auto graphdef = absl::make_unique<tensorflow::GraphDef>(); in MlirToJSONTranslateFunction() local
72 graph->ToGraphDef(graphdef.get()); in MlirToJSONTranslateFunction()
75 auto nodes = graphdef->mutable_node(); in MlirToJSONTranslateFunction()
85 *graphdef, &json_output, json_options); in MlirToJSONTranslateFunction()
/external/tensorflow/tensorflow/compiler/jit/
Dencapsulate_subgraphs_pass_test.cc51 GraphDef graphdef; in AddGraphDefToFunctionLibrary() local
52 TF_RETURN_IF_ERROR(graphdef_builder.ToGraphDef(&graphdef)); in AddGraphDefToFunctionLibrary()
57 TF_RETURN_IF_ERROR(ConvertGraphDefToGraph(opts, graphdef, graph.get())); in AddGraphDefToFunctionLibrary()
480 Status Encapsulate(GraphDef* graphdef, FunctionDefLibrary* library, in Encapsulate() argument
489 s = ConvertGraphDefToGraph(options, *graphdef, graph.get()); in Encapsulate()
539 graphdef->Swap(&graphdef_out); in Encapsulate()
553 Status Encapsulate(GraphDef* graphdef, FunctionDefLibrary* library) { in Encapsulate() argument
555 return Encapsulate(graphdef, library, encapsulated_functions); in Encapsulate()
584 GraphDef graphdef; in TEST() local
598 TF_EXPECT_OK(b1.ToGraphDef(&graphdef)); in TEST()
[all …]
/external/tensorflow/tensorflow/compiler/mlir/tensorflow/tests/mlir2graphdef/
Dfunc_attr.mlir1 … RUN: tf-mlir-translate -mlir-to-graphdef %s | tf-mlir-translate -graphdef-to-mlir | tf-mlir-trans…
Dpreserve-entry-func-names.mlir1 // RUN: tf-mlir-translate -mlir-to-graphdef %s -o - | FileCheck %s
6 …// This node would be renamed to bar1 [note: if imported from TF graphdef this would not be possib…
Dnoop.mlir1 // RUN: tf-mlir-translate -mlir-to-graphdef %s -o - | FileCheck %s
Dmissing-main.mlir1 // RUN: not tf-mlir-translate -mlir-to-graphdef %s -o - 2>&1 | FileCheck %s
Dstringescape.mlir1 // RUN: tf-mlir-translate -mlir-to-graphdef %s -o - | FileCheck %s
/external/tensorflow/tensorflow/compiler/mlir/tensorflow/tests/graphdef2mlir/
Dadd.pbtxt1 # RUN: tf-mlir-translate -graphdef-to-mlir -tf-enable-shape-inference-on-import=false %s -tf-input-…
2 # RUN: tf-mlir-translate -graphdef-to-mlir -tf-enable-shape-inference-on-import=false %s -tf-input-…
3 # RUN: tf-mlir-translate -graphdef-to-mlir -tf-enable-shape-inference-on-import=false %s -tf-input-…
4 # RUN: tf-mlir-translate -graphdef-to-mlir -tf-enable-shape-inference-on-import=false %s -tf-input-…
5 # RUN: tf-mlir-translate -graphdef-to-mlir -tf-enable-shape-inference-on-import=false %s -tf-input-…
Dlegacy-fed-input-without-inputs.pbtxt1 # RUN: tf-mlir-translate -graphdef-to-mlir -tf-enable-shape-inference-on-import=false %s -tf-input-…
2 # RUN: tf-mlir-translate -graphdef-to-mlir -tf-enable-shape-inference-on-import=false %s -tf-input-…
Dgraph-as-function-control-ret.pbtxt1 # RUN: tf-mlir-translate -graphdef-to-mlir -tf-enable-shape-inference-on-import=false %s -tf-graph-…
2 # RUN: not tf-mlir-translate -graphdef-to-mlir -tf-enable-shape-inference-on-import=false %s -tf-gr…
3 # RUN: not tf-mlir-translate -graphdef-to-mlir -tf-enable-shape-inference-on-import=false %s -tf-gr…
Darg-data-type.pbtxt1 # RUN: tf-mlir-translate -graphdef-to-mlir -tf-enable-shape-inference-on-import=false %s -tf-input-…
2 # RUN: tf-mlir-translate -graphdef-to-mlir -tf-enable-shape-inference-on-import=false %s -tf-input-…
Dtarget.pbtxt1 # RUN: tf-mlir-translate -graphdef-to-mlir -tf-enable-shape-inference-on-import=false %s -tf-contro…
2 # RUN: tf-mlir-translate -graphdef-to-mlir -tf-enable-shape-inference-on-import=false %s -tf-prune-…
3 # RUN: tf-mlir-translate -graphdef-to-mlir -tf-enable-shape-inference-on-import=false %s -tf-prune-…
Dgraph-malformed.pbtxt1 # RUN: not tf-mlir-translate -graphdef-to-mlir -tf-enable-shape-inference-on-import=false %s -o - 2…
Doutput-shapes-attr.pbtxt1 # RUN: tf-mlir-translate -graphdef-to-mlir -tf-enable-shape-inference-on-import=false %s -o - | Fil…
Dinvalid-output-index.pbtxt1 # RUN: not tf-mlir-translate -graphdef-to-mlir -tf-enable-shape-inference-on-import=false %s -tf-in…
Dstring-attr.pbtxt1 # RUN: tf-mlir-translate -graphdef-to-splatted-mlir %s -o - | FileCheck %s
/external/tensorflow/tensorflow/compiler/mlir/tensorflow/translate/
Dtf_mlir_translate.cc54 GraphDef graphdef; in GraphdefToMlirImport() local
56 tensorflow::LoadProtoFromBuffer({input.data(), input.size()}, &graphdef)); in GraphdefToMlirImport()
90 graphdef, &pruned_graph_def, terminal_nodes)); in GraphdefToMlirImport()
94 pruned_graph_def.mutable_library()->Swap(graphdef.mutable_library()); in GraphdefToMlirImport()
95 pruned_graph_def.mutable_versions()->Swap(graphdef.mutable_versions()); in GraphdefToMlirImport()
98 specs.prune_unused_nodes ? pruned_graph_def : graphdef, debug_info, specs, in GraphdefToMlirImport()
/external/tensorflow/tensorflow/python/
Dpywrap_mlir.py27 def import_graphdef(graphdef, pass_pipeline, show_debug_info): argument
29 str(graphdef).encode('utf-8'), pass_pipeline.encode('utf-8'),
Dmlir_wrapper.cc26 [](const std::string &graphdef, const std::string &pass_pipeline, in PYBIND11_MODULE()
31 graphdef, pass_pipeline, show_debug_info, status.get()); in PYBIND11_MODULE()
/external/tensorflow/tensorflow/compiler/mlir/python/
Dmlir.cc77 GraphDef graphdef; in ImportGraphDef() local
78 auto s = tensorflow::LoadProtoFromBuffer(proto, &graphdef); in ImportGraphDef()
86 auto module = ConvertGraphdefToMlir(graphdef, debug_info, specs, &context); in ImportGraphDef()
/external/tensorflow/tensorflow/lite/testing/
Dtf_driver.cc139 tensorflow::GraphDef graphdef; in LoadModel() local
140 if (!graphdef.ParseFromIstream(&model)) { in LoadModel()
147 auto status = session_->Create(graphdef); in LoadModel()
/external/tensorflow/tensorflow/python/tpu/
Dtensor_tracer.proto19 tensorflow.GraphDef graphdef = 2; field
25 // this field and graphdef).

12345