/external/tensorflow/tensorflow/core/graph/ |
graph_def_builder_test.cc | 43 GraphDef graph_def; local 44 TF_EXPECT_OK(builder.ToGraphDef(&graph_def)); 45 ASSERT_EQ(graph_def.versions().producer(), TF_GRAPH_DEF_VERSION); 46 ASSERT_EQ(graph_def.versions().min_consumer(),
|
graph_def_builder_util.cc | 22 GraphDef graph_def; local 23 TF_RETURN_IF_ERROR(builder.ToGraphDef(&graph_def)); 25 return ConvertGraphDefToGraph(opts, graph_def, graph);
|
validate_test.cc | 44 GraphDef graph_def; local 46 CHECK(parser.MergeFromString(graph_def_str, &graph_def)) << graph_def_str; 47 TF_ASSERT_OK(graph::ValidateGraphDef(graph_def, *OpRegistry::Global())); 59 GraphDef graph_def; local 61 CHECK(parser.MergeFromString(graph_def_str, &graph_def)) << graph_def_str; 62 Status s = graph::ValidateGraphDef(graph_def, *OpRegistry::Global()); 67 TF_ASSERT_OK(AddDefaultAttrsToGraphDef(&graph_def, *OpRegistry::Global(), 0)); 70 TF_ASSERT_OK(graph::ValidateGraphDef(graph_def, *OpRegistry::Global())); 82 GraphDef graph_def; local 84 CHECK(parser.MergeFromString(graph_def_str, &graph_def)) << graph_def_str 104 GraphDef graph_def; local 116 GraphDef graph_def; local 159 GraphDef graph_def; local 174 GraphDef graph_def; local [all...] |
benchmark_testlib.h | 74 GraphDef graph_def; local 86 *graph_def.add_node() = std::move(node); 97 *graph_def.add_node() = std::move(node); 109 *graph_def.add_node() = std::move(node); 111 return graph_def;
|
/external/tensorflow/tensorflow/core/kernels/data/ |
dataset_ops.cc | 41 GraphDef graph_def; variable 42 OP_REQUIRES_OK(ctx, b.ToGraphDef(&graph_def)); 45 result->scalar<string>()() = graph_def.SerializeAsString();
|
/external/tensorflow/tensorflow/core/util/ |
stat_summarizer_test.cc | 58 GraphDef graph_def; local 59 ASSERT_TRUE(protobuf::TextFormat::ParseFromString(graph_def_str, &graph_def)); 63 TF_ASSERT_OK(session->Create(graph_def)); 73 StatSummarizer stats(graph_def);
|
dump_graph.cc | 112 string DumpGraphDefToFile(const string& name, GraphDef const& graph_def, 114 return WriteTextProtoToUniqueFile(Env::Default(), name, "GraphDef", graph_def, 121 GraphDef graph_def; local 122 graph.ToGraphDef(&graph_def); 124 *graph_def.mutable_library() = flib_def->ToProto(); 126 return DumpGraphDefToFile(name, graph_def, dirname);
|
/external/tensorflow/tensorflow/tools/graph_transforms/ |
add_default_attributes_test.cc | 39 GraphDef graph_def; local 41 NodeDef* lrn_node1 = graph_def.add_node(); 45 NodeDef* lrn_node2 = graph_def.add_node(); 54 TF_ASSERT_OK(AddDefaultAttributes(graph_def, {}, &result));
|
remove_device_test.cc | 39 GraphDef graph_def; local 41 NodeDef* mul_node1 = graph_def.add_node(); 48 NodeDef* add_node2 = graph_def.add_node(); 55 NodeDef* add_node3 = graph_def.add_node(); 61 NodeDef* const_node1 = graph_def.add_node(); 65 NodeDef* const_node2 = graph_def.add_node(); 69 NodeDef* const_node3 = graph_def.add_node(); 73 NodeDef* add_node4 = graph_def.add_node(); 83 TF_ASSERT_OK(RemoveDevice(graph_def, context, &result));
|
rename_op_test.cc | 39 GraphDef graph_def; local 41 NodeDef* mul_node1 = graph_def.add_node(); 47 NodeDef* add_node2 = graph_def.add_node(); 53 NodeDef* add_node3 = graph_def.add_node(); 59 NodeDef* const_node1 = graph_def.add_node(); 63 NodeDef* const_node2 = graph_def.add_node(); 67 NodeDef* const_node3 = graph_def.add_node(); 71 NodeDef* add_node4 = graph_def.add_node(); 85 TF_ASSERT_OK(RenameOp(graph_def, context, &result));
|
set_device_test.cc | 38 GraphDef graph_def; local 40 NodeDef* mul_node1 = graph_def.add_node(); 47 NodeDef* add_node2 = graph_def.add_node(); 54 NodeDef* add_node3 = graph_def.add_node(); 60 NodeDef* const_node1 = graph_def.add_node(); 64 NodeDef* const_node2 = graph_def.add_node(); 68 NodeDef* const_node3 = graph_def.add_node(); 72 NodeDef* add_node4 = graph_def.add_node(); 78 return graph_def; 83 GraphDef graph_def = CreateDeviceGraph() local 104 GraphDef graph_def = CreateDeviceGraph(); local [all...] |
/external/tensorflow/tensorflow/compiler/tf2xla/ |
xla_jit_compiled_cpu_function_test.cc | 41 GraphDef graph_def; local 42 NodeDef* x = graph_def.add_node(); 46 NodeDef* y = graph_def.add_node(); 50 NodeDef* sum = graph_def.add_node(); 56 return graph_def; 74 GraphDef graph_def = SumGraph(); local 79 XlaJitCompiledCpuFunction::Compile(graph_def, config, 135 GraphDef graph_def = SumGraph(); local 137 (*graph_def.mutable_node(2)->mutable_attr())["junk"] = 141 EXPECT_FALSE(XlaJitCompiledCpuFunction::Compile(graph_def, config [all...] |
/external/tensorflow/tensorflow/contrib/util/ |
convert_graphdef_memmapped_format_test.cc | 29 bool GraphHasImmutableConstNodes(const GraphDef& graph_def) { 30 for (const auto& node : graph_def.node()) { 58 GraphDef graph_def; local 59 TF_ASSERT_OK(root.ToGraphDef(&graph_def)); 61 graph_def.SerializeToString(&graph_def_serialized); 109 GraphDef graph_def; local 110 TF_ASSERT_OK(root.ToGraphDef(&graph_def)); 112 graph_def.SerializeToString(&graph_def_serialized);
|
/external/tensorflow/tensorflow/core/common_runtime/data/ |
standalone_test.cc | 153 GraphDef graph_def; local 154 protobuf::TextFormat::ParseFromString(kGraphProto, &graph_def); 165 auto s = Dataset::FromGraph({}, graph_def, test_case.fetch_node, &dataset);
|
/external/tensorflow/tensorflow/core/kernels/hexagon/ |
hexagon_rewriter_transform_test.cc | 60 GraphDef graph_def; local 61 TF_ASSERT_OK(root.ToGraphDef(&graph_def)); 72 RewriteQuantizedStrippedModelForHexagon(graph_def, context, &result));
|
/external/tensorflow/tensorflow/core/lib/strings/ |
proto_serialization_test.cc | 31 GraphDef graph_def; local 33 NodeDef* node = graph_def.add_node(); 39 return graph_def; 45 GraphDef graph_def = MakeGraphDef(num_nodes); local 50 SerializeToStringDeterministic(graph_def, &serialized)); 58 GraphDef graph_def = MakeGraphDef(num_nodes); local 60 const size_t size = graph_def.ByteSizeLong(); 64 SerializeToBufferDeterministic(graph_def, buf.data(), size)); 72 GraphDef graph_def = MakeGraphDef(num_nodes); local 75 testing::DoNotOptimize(DeterministicProtoHash64(graph_def)); [all...] |
/external/tensorflow/tensorflow/lite/tools/accuracy/ |
eval_pipeline_builder.cc | 88 GraphDef graph_def; local 89 TF_RETURN_IF_ERROR(scope.ToGraphDef(&graph_def)); 94 absl::make_unique<EvalPipeline>(graph_def, params, accuracy_eval_);
|
eval_pipeline_test.cc | 60 GraphDef graph_def; local 61 TF_CHECK_OK(scope.ToGraphDef(&graph_def)); 67 EvalPipeline eval_pipeline(graph_def, params, &accuracy_eval); 85 GraphDef graph_def; local 86 TF_CHECK_OK(scope.ToGraphDef(&graph_def)); 92 EvalPipeline eval_pipeline(graph_def, params, &accuracy_eval); 108 GraphDef graph_def; local 109 TF_CHECK_OK(scope.ToGraphDef(&graph_def)); 115 EvalPipeline eval_pipeline(graph_def, params, &accuracy_eval);
|
file_reader_stage_test.cc | 69 GraphDef graph_def; local 70 TF_CHECK_OK(scope.ToGraphDef(&graph_def)); 72 TF_CHECK_OK(session->Create(graph_def)); 89 GraphDef graph_def; local 90 TF_CHECK_OK(scope.ToGraphDef(&graph_def)); 92 TF_CHECK_OK(session->Create(graph_def));
|
run_tflite_model_op_test.cc | 69 GraphDef graph_def; local 70 TF_CHECK_OK(scope.ToGraphDef(&graph_def)); 73 TF_CHECK_OK(session->Create(graph_def)); 129 GraphDef graph_def; local 130 TF_CHECK_OK(scope.ToGraphDef(&graph_def)); 132 TF_CHECK_OK(session->Create(graph_def)); 176 GraphDef graph_def; local 177 TF_CHECK_OK(scope.ToGraphDef(&graph_def)); 179 TF_CHECK_OK(session->Create(graph_def));
|
/external/tensorflow/tensorflow/compiler/aot/ |
tfcompile_main.cc | 87 GraphDef graph_def; local 88 TF_RETURN_IF_ERROR(ReadProtoFile(flags.graph, &graph_def)); 90 TF_RETURN_IF_ERROR(CompileGraph(graph_def, config, flags, &compile_result));
|
/external/tensorflow/tensorflow/core/framework/ |
graph_to_functiondef_test.cc | 65 GraphDef graph_def; local 66 TF_EXPECT_OK(root.ToGraphDef(&graph_def)); 70 TF_EXPECT_OK(ConvertGraphDefToGraph(options, graph_def, graph.get())); 102 GraphDef graph_def; local 103 TF_EXPECT_OK(root.ToGraphDef(&graph_def)); 107 TF_EXPECT_OK(ConvertGraphDefToGraph(options, graph_def, graph.get()));
|
/external/tensorflow/tensorflow/core/kernels/fuzzing/ |
fuzz_session.h | 96 GraphDef graph_def; local 97 TF_CHECK_OK(root.ToGraphDef(&graph_def)); 99 Status status = session_->Create(graph_def);
|
/external/tensorflow/tensorflow/lite/tools/accuracy/ilsvrc/ |
inception_preprocessing_test.cc | 65 GraphDef graph_def; local 66 TF_CHECK_OK(scope.ToGraphDef(&graph_def)); 68 TF_CHECK_OK(session->Create(graph_def)); 94 GraphDef graph_def; local 95 TF_CHECK_OK(scope.ToGraphDef(&graph_def)); 97 TF_CHECK_OK(session->Create(graph_def)); 123 GraphDef graph_def; local 124 TF_CHECK_OK(scope.ToGraphDef(&graph_def)); 126 TF_CHECK_OK(session->Create(graph_def));
|
/external/tensorflow/tensorflow/tools/benchmark/ |
benchmark_model_test.cc | 31 string* output_name, GraphDef* graph_def) { 47 TF_ASSERT_OK(root.ToGraphDef(graph_def)); 57 GraphDef graph_def; local 58 CreateTestGraph(root, &input, &output_name, &graph_def); 60 graph_def.SerializeToString(&graph_def_serialized); 85 GraphDef graph_def; local 86 CreateTestGraph(root, &input, &output_name, &graph_def); 87 TF_ASSERT_OK(WriteTextProto(Env::Default(), filename_txt, graph_def));
|