| /external/tensorflow/tensorflow/core/lib/io/ |
| D | inputbuffer_test.cc | 45 TF_CHECK_OK(env->NewRandomAccessFile(fname, &file)); in TEST()
55 TF_CHECK_OK( in TEST()
60 TF_CHECK_OK(env->NewRandomAccessFile(fname, &file)); in TEST()
63 TF_CHECK_OK(in.ReadLine(&line)); in TEST()
65 TF_CHECK_OK(in.ReadLine(&line)); in TEST()
67 TF_CHECK_OK(in.ReadLine(&line)); in TEST()
82 TF_CHECK_OK(env->NewRandomAccessFile(fname, &file)); in TEST()
85 TF_CHECK_OK(in.ReadLine(&line)); in TEST()
87 TF_CHECK_OK(in.ReadLine(&line)); in TEST()
89 TF_CHECK_OK(in.ReadLine(&line)); in TEST()
[all …]
|
| D | record_reader_writer_test.cc | 50 TF_CHECK_OK(env->GetFileSize(fname, &fsize)); in GetFileSize()
66 TF_CHECK_OK(env->NewWritableFile(fname, &file)); in VerifyFlush()
70 TF_CHECK_OK(env->NewRandomAccessFile(fname, &read_file)); in VerifyFlush()
80 TF_CHECK_OK(writer.Flush()); in VerifyFlush()
81 TF_CHECK_OK(file->Flush()); in VerifyFlush()
92 TF_CHECK_OK(reader.ReadRecord(&offset, &record)); in VerifyFlush()
127 TF_CHECK_OK(env->NewWritableFile(fname, &file)); in TEST()
134 TF_CHECK_OK(writer.Flush()); in TEST()
140 TF_CHECK_OK(env->NewRandomAccessFile(fname, &read_file)); in TEST()
146 TF_CHECK_OK(reader.ReadRecord(&offset, &record)); in TEST()
[all …]
|
| /external/tensorflow/tensorflow/core/graph/ |
| D | testlib.cc | 38 TF_CHECK_OK(NodeBuilder(g->NewName("n"), "_Send") in Send()
55 TF_CHECK_OK(NodeBuilder(g->NewName("n"), "_Recv") in Recv()
68 TF_CHECK_OK(NodeBuilder(g->NewName("n"), "Const") in Constant()
77 TF_CHECK_OK(NodeBuilder(name, "Const") in Constant()
90 TF_CHECK_OK(NodeBuilder(name, "HostConst") in HostConstant()
99 TF_CHECK_OK(NodeBuilder(g->NewName("n"), "Variable") in Var()
109 TF_CHECK_OK(NodeBuilder(name, "Variable") in Var()
118 TF_CHECK_OK(NodeBuilder(g->NewName("n"), "Assign") in Assign()
128 TF_CHECK_OK(NodeBuilder(g->NewName("n"), "Cumsum") in Cumsum()
140 TF_CHECK_OK(NodeBuilder(g->NewName("n"), reduce, g->op_registry()) in Reduce()
[all …]
|
| D | mkl_layout_pass.cc | 788 TF_CHECK_OK(GetNodeAttr(n->def(), attr_name, &value)); in GetTensorListLength()
791 TF_CHECK_OK(GetNodeAttr(n->def(), attr_name, &N)); in GetTensorListLength()
865 TF_CHECK_OK(GetNodeAttr(m->def(), "T", &T_m)); in GetConv2DOrBiasAdd()
872 TF_CHECK_OK(m->input_node(0, &n)); in GetConv2DOrBiasAdd()
905 TF_CHECK_OK(GetNodeAttr(m->def(), "T", &T_m)); in GetPadOrConv2D()
936 TF_CHECK_OK(GetNodeAttr(conv_node->def(), "padding", &padding)); in GetPadOrConv2D()
985 TF_CHECK_OK(GetNodeAttr(conv_node->def(), "padding", &padding)); in GetPadOrFusedConv2D()
1022 TF_CHECK_OK(GetNodeAttr(m->def(), "T", &T_m)); in GetConv2DBackpropFilterOrBiasAddGrad()
1030 TF_CHECK_OK(m->input_node(0, &g)); in GetConv2DBackpropFilterOrBiasAddGrad()
1045 TF_CHECK_OK(m->input_node(2, &g)); in GetConv2DBackpropFilterOrBiasAddGrad()
[all …]
|
| /external/tensorflow/tensorflow/core/common_runtime/ |
| D | kernel_benchmark_testlib.cc | 91 TF_CHECK_OK(NewExecutor(executor_type, params, std::unique_ptr<Graph>(init), in Benchmark()
96 TF_CHECK_OK(init_exec->Run(args)); in Benchmark()
99 TF_CHECK_OK( in Benchmark()
119 TF_CHECK_OK(GetNodeAttr(node->attrs(), "send_device", &send_device)); in GetRendezvousKey()
121 TF_CHECK_OK(GetNodeAttr(node->attrs(), "recv_device", &recv_device)); in GetRendezvousKey()
123 TF_CHECK_OK(GetNodeAttr(node->attrs(), "tensor_name", &tensor_name)); in GetRendezvousKey()
125 TF_CHECK_OK(GetNodeAttr(node->attrs(), "send_device_incarnation", in GetRendezvousKey()
161 TF_CHECK_OK(Rendezvous::ParseKey(p.first, &parsed)); in RunWithArgs()
162 TF_CHECK_OK(rendez_->Send(parsed, Rendezvous::Args(), p.second, false)); in RunWithArgs()
164 TF_CHECK_OK(exec_->Run(args)); in RunWithArgs()
[all …]
|
| /external/tensorflow/tensorflow/compiler/jit/ |
| D | extract_outside_compilation_pass_test.cc | 53 TF_CHECK_OK(s.ToGraph(g.get())); in TEST()
64 TF_CHECK_OK( in TEST()
78 TF_CHECK_OK( in TEST()
91 TF_CHECK_OK( in TEST()
118 TF_CHECK_OK(GetNodeAttr(AttrSlice(&call_node_def.attr()), in TEST()
129 TF_CHECK_OK(s.ToGraph(g.get())); in TEST()
135 TF_CHECK_OK( in TEST()
155 TF_CHECK_OK(s.ToGraph(g.get())); in TEST()
161 TF_CHECK_OK( in TEST()
179 TF_CHECK_OK(s.ToGraph(g.get())); in TEST()
[all …]
|
| /external/tensorflow/tensorflow/core/grappler/clusters/ |
| D | single_machine_test.cc | 48 TF_CHECK_OK(cluster_->EnablePeakMemoryStats(true)); in SetUp()
49 TF_CHECK_OK(cluster_->Provision()); in SetUp()
54 TF_CHECK_OK(cluster_->Shutdown()); in TearDown()
73 TF_CHECK_OK(cluster_->Initialize(item)); in TEST_F()
77 TF_CHECK_OK(cluster_->Run(item.graph, item.feed, item.fetch, &metadata)); in TEST_F()
106 TF_CHECK_OK(cluster_->Initialize(item)); in TEST_F()
108 TF_CHECK_OK(cluster_->Run(item.graph, item.feed, item.fetch, &metadata)); in TEST_F()
118 TF_CHECK_OK(cluster_->Initialize(item)); in TEST_F()
120 TF_CHECK_OK(cluster_->Run(item.graph, item.feed, item.fetch, &metadata1)); in TEST_F()
122 TF_CHECK_OK(cluster_->Run(item.graph, item.feed, item.fetch, &metadata2)); in TEST_F()
[all …]
|
| /external/tensorflow/tensorflow/core/distributed_runtime/rpc/ |
| D | grpc_session_test.cc | 94 TF_CHECK_OK(test::TestCluster::MakeTestCluster(Devices(1, 0), 2, &cluster)); in TEST()
101 TF_CHECK_OK(session->Create(graph)); in TEST()
106 TF_CHECK_OK(session->Run(inputs, {}, targets, nullptr)); in TEST()
114 TF_CHECK_OK(session->Run(inputs, names, targets, &outputs)); in TEST()
119 TF_CHECK_OK(session->Close()); in TEST()
130 TF_CHECK_OK(test::TestCluster::MakeTestCluster(Devices(1, 0), 2, &cluster)); in TEST()
137 TF_CHECK_OK(session->Create(graph)); in TEST()
143 TF_CHECK_OK(session->MakeCallable(opts, &handle)); in TEST()
144 TF_CHECK_OK(session->RunCallable(handle, {}, nullptr, nullptr)); in TEST()
145 TF_CHECK_OK(session->ReleaseCallable(handle)); in TEST()
[all …]
|
| /external/tensorflow/tensorflow/core/grappler/costs/ |
| D | graph_properties_test.cc | 47 TF_CHECK_OK(cluster_->Provision()); in SetUp()
75 TF_CHECK_OK(cluster_->Shutdown()); in TearDown()
130 TF_CHECK_OK(s); in TEST_F()
174 TF_CHECK_OK(s); in TEST_F()
200 TF_CHECK_OK(cluster_->Initialize(item)); in TEST_F()
202 TF_CHECK_OK(s); in TEST_F()
242 TF_CHECK_OK(NodeDefBuilder("Var", "Variable") in TEST_F()
250 TF_CHECK_OK(NodeDefBuilder("InitialVal", "Const") in TEST_F()
254 TF_CHECK_OK(NodeDefBuilder("InitVar", "Assign") in TEST_F()
262 TF_CHECK_OK(static_properties.InferStatically(false)); in TEST_F()
[all …]
|
| /external/tensorflow/tensorflow/tools/optimization/ |
| D | gpu_optimization_pass_runner_main.cc | 36 TF_CHECK_OK(runner.AddDevice( in main()
39 TF_CHECK_OK(runner.AddDevice( in main()
42 TF_CHECK_OK(runner.AddDevice( in main()
45 TF_CHECK_OK(runner.AddDevice( in main()
48 TF_CHECK_OK(runner.AddDevice( in main()
51 TF_CHECK_OK(runner.AddDevice( in main()
56 TF_CHECK_OK(runner.SetJitLevel(tensorflow::OptimizerOptions::GlobalJitLevel:: in main()
59 TF_CHECK_OK(runner.RunMain(argc, argv)); in main()
|
| /external/tensorflow/tensorflow/core/grappler/inputs/ |
| D | utils_test.cc | 30 TF_CHECK_OK(env_->CreateDir(BaseDir())); in SetUp()
33 TF_CHECK_OK(WriteStringToFile(env_, actual_file_, "Some test data")); in SetUp()
46 TF_CHECK_OK(WriteTextProto(env_, text_graph_def_file_, graph_def_)); in SetUp()
47 TF_CHECK_OK(WriteBinaryProto(env_, binary_graph_def_file_, graph_def_)); in SetUp()
49 TF_CHECK_OK( in SetUp()
51 TF_CHECK_OK( in SetUp()
57 TF_CHECK_OK( in TearDown()
88 TF_CHECK_OK(ReadGraphDefFromFile(text_graph_def_file_, &result)); in TEST_F()
94 TF_CHECK_OK(ReadGraphDefFromFile(binary_graph_def_file_, &result)); in TEST_F()
100 TF_CHECK_OK(ReadMetaGraphDefFromFile(text_meta_graph_def_file_, &result)); in TEST_F()
[all …]
|
| /external/tensorflow/tensorflow/core/ops/ |
| D | array_grad_test.cc | 51 TF_CHECK_OK(sess->Create(gdef)); in PackGrad()
53 TF_CHECK_OK(sess->Run({{"x0:0", x0}, in PackGrad()
59 TF_CHECK_OK(sess->Close()); in PackGrad()
92 TF_CHECK_OK(sess->Create(gdef)); in UnpackGrad()
94 TF_CHECK_OK(sess->Run({{"x:0", x}, in UnpackGrad()
100 TF_CHECK_OK(sess->Close()); in UnpackGrad()
131 TF_CHECK_OK(sess->Create(gdef)); in ConcatGrad()
133 TF_CHECK_OK(sess->Run( in ConcatGrad()
137 TF_CHECK_OK(sess->Close()); in ConcatGrad()
155 TF_CHECK_OK(sess->Create(gdef)); in ConcatGradV2()
[all …]
|
| /external/tensorflow/tensorflow/lite/tools/accuracy/ilsvrc/ |
| D | inception_preprocessing_test.cc | 63 TF_CHECK_OK(scope.status()); in TEST()
66 TF_CHECK_OK(scope.ToGraphDef(&graph_def)); in TEST()
68 TF_CHECK_OK(session->Create(graph_def)); in TEST()
74 TF_CHECK_OK(run_status); in TEST()
92 TF_CHECK_OK(scope.status()); in TEST()
95 TF_CHECK_OK(scope.ToGraphDef(&graph_def)); in TEST()
97 TF_CHECK_OK(session->Create(graph_def)); in TEST()
103 TF_CHECK_OK(run_status); in TEST()
121 TF_CHECK_OK(scope.status()); in TEST()
124 TF_CHECK_OK(scope.ToGraphDef(&graph_def)); in TEST()
[all …]
|
| /external/tensorflow/tensorflow/core/summary/ |
| D | summary_file_writer_test.cc | 52 TF_CHECK_OK(CreateSummaryFileWriter(1, 1, testing::TmpDir(), test_name, in SummaryTestHelper()
56 TF_CHECK_OK(writer_fn(writer)); in SummaryTestHelper()
57 TF_CHECK_OK(writer->Flush()); in SummaryTestHelper()
60 TF_CHECK_OK(env_.GetChildren(testing::TmpDir(), &files)); in SummaryTestHelper()
69 TF_CHECK_OK(env_.NewRandomAccessFile(io::JoinPath(testing::TmpDir(), f), in SummaryTestHelper()
74 TF_CHECK_OK( in SummaryTestHelper()
77 TF_CHECK_OK(reader.ReadRecord(&offset, &record)); in SummaryTestHelper()
93 TF_CHECK_OK(SummaryTestHelper("tensor_test", in TEST_F()
108 TF_CHECK_OK(SummaryTestHelper( in TEST_F()
128 TF_CHECK_OK(SummaryTestHelper( in TEST_F()
[all …]
|
| /external/tensorflow/tensorflow/core/platform/ |
| D | env_test.cc | 39 TF_CHECK_OK(WriteStringToFile(env, filename, input)); in CreateTestFile()
60 void SetUp() override { TF_CHECK_OK(env_->CreateDir(BaseDir())); } in SetUp()
64 TF_CHECK_OK( in TearDown()
206 TF_CHECK_OK(env_->RecursivelyCreateDir(create_path)); in TEST_F()
207 TF_CHECK_OK(env_->RecursivelyCreateDir(create_path)); // repeat creation. in TEST_F()
212 TF_CHECK_OK(env_->RecursivelyCreateDir("")); in TEST_F()
218 TF_CHECK_OK(env_->CreateDir(io::JoinPath(BaseDir(), "a"))); in TEST_F()
219 TF_CHECK_OK(env_->CreateDir(subdir_path)); in TEST_F()
224 TF_CHECK_OK(env_->RecursivelyCreateDir(create_path)); in TEST_F()
225 TF_CHECK_OK(env_->RecursivelyCreateDir(create_path)); // repeat creation. in TEST_F()
[all …]
|
| /external/tensorflow/tensorflow/lite/tools/accuracy/ |
| D | run_tflite_model_op_test.cc | 39 TF_CHECK_OK(scope.status()); in TEST()
67 TF_CHECK_OK(scope.status()); in TEST()
70 TF_CHECK_OK(scope.ToGraphDef(&graph_def)); in TEST()
73 TF_CHECK_OK(session->Create(graph_def)); in TEST()
76 TF_CHECK_OK( in TEST()
99 TF_CHECK_OK(scope.status()); in TEST()
127 TF_CHECK_OK(scope.status()); in TEST()
130 TF_CHECK_OK(scope.ToGraphDef(&graph_def)); in TEST()
132 TF_CHECK_OK(session->Create(graph_def)); in TEST()
146 TF_CHECK_OK(scope.status()); in TEST()
[all …]
|
| D | file_reader_stage_test.cc | 68 TF_CHECK_OK(scope.status()); in TEST()
70 TF_CHECK_OK(scope.ToGraphDef(&graph_def)); in TEST()
72 TF_CHECK_OK(session->Create(graph_def)); in TEST()
78 TF_CHECK_OK(run_status); in TEST()
88 TF_CHECK_OK(scope.status()); in TEST()
90 TF_CHECK_OK(scope.ToGraphDef(&graph_def)); in TEST()
92 TF_CHECK_OK(session->Create(graph_def)); in TEST()
|
| /external/tensorflow/tensorflow/core/util/ |
| D | tensor_slice_reader_test.cc | 73 TF_CHECK_OK(writer.Add("test", shape, slice, data)); in SimpleFloatHelper()
74 TF_CHECK_OK(writer.Finish()); in SimpleFloatHelper()
97 TF_CHECK_OK(writer.Add("test", shape, slice, data)); in SimpleFloatHelper()
103 TF_CHECK_OK(writer.Add("test", shape, slice, data)); in SimpleFloatHelper()
105 TF_CHECK_OK(writer.Finish()); in SimpleFloatHelper()
198 TF_CHECK_OK(writer.Add("test", shape, slice, data)); in SimpleIntXHelper()
199 TF_CHECK_OK(writer.Finish()); in SimpleIntXHelper()
222 TF_CHECK_OK(writer.Add("test", shape, slice, data)); in SimpleIntXHelper()
228 TF_CHECK_OK(writer.Add("test", shape, slice, data)); in SimpleIntXHelper()
230 TF_CHECK_OK(writer.Finish()); in SimpleIntXHelper()
[all …]
|
| /external/tensorflow/tensorflow/core/grappler/optimizers/ |
| D | graph_optimizer_stage_test.cc | 149 TF_CHECK_OK(s.ToGraphDef(&item.graph)); in TEST_F()
152 TF_CHECK_OK(properties.InferStatically(/*assume_valid_feeds*/ false)); in TEST_F()
165 TF_CHECK_OK(stage.GetInputNode("Add", &add_node)); in TEST_F()
171 TF_CHECK_OK(stage.GetTensorProperties("Add", &add_properties)); in TEST_F()
175 TF_CHECK_OK(stage.GetTensorProperties("a:0", &a_properties)); in TEST_F()
179 TF_CHECK_OK(stage.GetTensorProperties("b:0", &b_properties)); in TEST_F()
191 TF_CHECK_OK(s.ToGraphDef(&item.graph)); in TEST_F()
194 TF_CHECK_OK(properties.InferStatically(/*assume_valid_feeds*/ false)); in TEST_F()
207 TF_CHECK_OK(stage.GetInputNode("Add", &add_node)); in TEST_F()
219 TF_CHECK_OK(stage.GetInputNode("Add_1", &add_node_copy_by_name)); in TEST_F()
[all …]
|
| /external/tensorflow/tensorflow/core/framework/ |
| D | op_gen_lib_test.cc | 254 TF_CHECK_OK(api_map.LoadApiDef(kTestApiDef)); in TEST()
286 TF_CHECK_OK(api_map.LoadApiDef(kTestApiDef)); in TEST()
292 TF_CHECK_OK(api_map.LoadApiDef(api_def1)); in TEST()
297 TF_CHECK_OK(api_map.LoadApiDef(api_def2)); in TEST()
314 TF_CHECK_OK(api_map.LoadApiDef(kTestApiDef)); in TEST()
319 TF_CHECK_OK(api_map.LoadApiDef(api_def1)); in TEST()
344 TF_CHECK_OK(api_map.LoadApiDef(kTestApiDef)); in TEST()
345 TF_CHECK_OK(api_map.LoadApiDef(api_def1)); in TEST()
385 TF_CHECK_OK(api_map.LoadApiDef(kTestApiDef)); in TEST()
386 TF_CHECK_OK(api_map.LoadApiDef(api_def1)); in TEST()
[all …]
|
| D | resource_mgr_test.cc | 56 TF_CHECK_OK(rm.Lookup(container, name, &r)); in Find()
66 TF_CHECK_OK(rm->LookupOrCreate<T>(container, name, &r, [&label](T** ret) { in LookupOrCreate()
91 TF_CHECK_OK(rm.Create("foo", "bar", new Resource("cat"))); in TEST()
92 TF_CHECK_OK(rm.Create("foo", "baz", new Resource("dog"))); in TEST()
93 TF_CHECK_OK(rm.Create("foo", "bar", new Other("tiger"))); in TEST()
110 TF_CHECK_OK(rm.Delete<Resource>("foo", "bar")); in TEST()
113 TF_CHECK_OK(rm.Create("foo", "bar", new Resource("kitty"))); in TEST()
117 TF_CHECK_OK(rm.Cleanup("foo")); in TEST()
121 TF_CHECK_OK(rm.Cleanup("foo")); in TEST()
125 TF_CHECK_OK(rm.Cleanup("bar")); in TEST()
[all …]
|
| D | common_shape_fns_test.cc | 54 TF_CHECK_OK(OpDefBuilder("Assert") in TEST()
61 TF_CHECK_OK(NodeDefBuilder("test", "Assert") in TEST()
74 TF_CHECK_OK(OpDefBuilder("L2Loss") in TEST()
81 TF_CHECK_OK( in TEST()
102 TF_CHECK_OK(OpDefBuilder("MatMul") in TEST()
112 TF_CHECK_OK(NodeDefBuilder("test", "MatMul") in TEST()
181 TF_CHECK_OK(NodeDefBuilder("test", "MatMul") in TEST()
199 TF_CHECK_OK(NodeDefBuilder("test", "MatMul") in TEST()
218 TF_CHECK_OK(OpDefBuilder("BiasAdd") in TEST()
226 TF_CHECK_OK(NodeDefBuilder("test", "BiasAdd") in TEST()
[all …]
|
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | dynamic_dimension_inference_test.cc | 93 TF_CHECK_OK(module_->dynamic_parameter_binding().Bind( in TEST_F()
112 TF_CHECK_OK(module_->dynamic_parameter_binding().Bind( in TEST_F()
138 TF_CHECK_OK(module_->dynamic_parameter_binding().Bind( in TEST_F()
169 TF_CHECK_OK(module_->dynamic_parameter_binding().Bind( in TEST_F()
200 TF_CHECK_OK(module_->dynamic_parameter_binding().Bind( in TEST_F()
232 TF_CHECK_OK(module_->dynamic_parameter_binding().Bind( in TEST_F()
268 TF_CHECK_OK(module_->dynamic_parameter_binding().Bind( in TEST_F()
273 TF_CHECK_OK(module_->dynamic_parameter_binding().Bind( in TEST_F()
276 TF_CHECK_OK(module_->dynamic_parameter_binding().Bind( in TEST_F()
320 TF_CHECK_OK(module_->dynamic_parameter_binding().Bind( in TEST_F()
[all …]
|
| /external/tensorflow/tensorflow/core/distributed_runtime/ |
| D | rpcbench_test.cc | 87 TF_CHECK_OK(NewServer(server, &svr)); in MakeGRPCCluster()
88 TF_CHECK_OK(svr->Start()); in MakeGRPCCluster()
89 TF_CHECK_OK(svr->Join()); in MakeGRPCCluster()
98 TF_CHECK_OK(GrpcSession::Create(options_copy, &session)); in MakeGRPCCluster()
99 TF_CHECK_OK(session->ListDevices(devices)); in MakeGRPCCluster()
153 TF_CHECK_OK(s.ToGraphDef(&def)); in CreateGraphDef()
180 TF_CHECK_OK(session->Create(def)); in BM_Helper()
195 TF_CHECK_OK(session->Run({{"x", x}}, {"y:0"}, {}, &outputs)); in BM_Helper()
209 TF_CHECK_OK(session->Run({{"x", x}}, {"y:0"}, {}, &outputs)); in BM_Helper()
213 TF_CHECK_OK(session->Close()); in BM_Helper()
|
| /external/tensorflow/tensorflow/lite/testing/ |
| D | generated_examples_zip_test.cc | 139 TF_CHECK_OK( in TearDown()
149 TF_CHECK_OK(MakeTemporaryDirectory(&dir)); in UnArchive()
153 TF_CHECK_OK(env->FileExists(unzip_binary)); in UnArchive()
154 TF_CHECK_OK(env->FileExists(zip)); in UnArchive()
158 TF_CHECK_OK(env->FileExists(tar_binary)); in UnArchive()
159 TF_CHECK_OK(env->FileExists(tar)); in UnArchive()
187 TF_CHECK_OK(env->CreateDir(*temporary)); in MakeTemporaryDirectory()
236 TF_CHECK_OK(tensorflow::Status(tensorflow::error::UNKNOWN, in UnarchiveAndFindTestNames()
240 TF_CHECK_OK(archive_environment()->UnArchive(zip_file, tar_file, in UnarchiveAndFindTestNames()
244 TF_CHECK_OK(ReadManifest(zip_file, decompress_tmp_dir, &stuff)); in UnarchiveAndFindTestNames()
[all …]
|