Home
last modified time | relevance | path

Searched refs:TF_CHECK_OK (Results 1 – 25 of 273) sorted by relevance

1234567891011

/external/tensorflow/tensorflow/core/lib/io/
Dinputbuffer_test.cc44 TF_CHECK_OK(env->NewRandomAccessFile(fname, &file)); in TEST()
54 TF_CHECK_OK( in TEST()
59 TF_CHECK_OK(env->NewRandomAccessFile(fname, &file)); in TEST()
62 TF_CHECK_OK(in.ReadLine(&line)); in TEST()
64 TF_CHECK_OK(in.ReadLine(&line)); in TEST()
66 TF_CHECK_OK(in.ReadLine(&line)); in TEST()
81 TF_CHECK_OK(env->NewRandomAccessFile(fname, &file)); in TEST()
84 TF_CHECK_OK(in.ReadLine(&line)); in TEST()
86 TF_CHECK_OK(in.ReadLine(&line)); in TEST()
88 TF_CHECK_OK(in.ReadLine(&line)); in TEST()
[all …]
Drecord_reader_writer_test.cc43 TF_CHECK_OK(env->NewWritableFile(fname, &file)); in TEST()
50 TF_CHECK_OK(writer.Flush()); in TEST()
56 TF_CHECK_OK(env->NewRandomAccessFile(fname, &read_file)); in TEST()
62 TF_CHECK_OK(reader.ReadRecord(&offset, &record)); in TEST()
64 TF_CHECK_OK(reader.ReadRecord(&offset, &record)); in TEST()
79 TF_CHECK_OK(env->NewWritableFile(fname, &file)); in TEST()
87 TF_CHECK_OK(writer.Flush()); in TEST()
93 TF_CHECK_OK(env->NewRandomAccessFile(fname, &read_file)); in TEST()
100 TF_CHECK_OK(reader.ReadRecord(&offset, &record)); in TEST()
102 TF_CHECK_OK(reader.ReadRecord(&offset, &record)); in TEST()
/external/tensorflow/tensorflow/core/common_runtime/
Dkernel_benchmark_testlib.cc89 TF_CHECK_OK( in Benchmark()
94 TF_CHECK_OK(init_exec->Run(args)); in Benchmark()
98 TF_CHECK_OK(NewLocalExecutor(params, std::unique_ptr<Graph>(g), &exec_)); in Benchmark()
114 TF_CHECK_OK(GetNodeAttr(node->attrs(), "send_device", &send_device)); in GetRendezvousKey()
116 TF_CHECK_OK(GetNodeAttr(node->attrs(), "recv_device", &recv_device)); in GetRendezvousKey()
118 TF_CHECK_OK(GetNodeAttr(node->attrs(), "tensor_name", &tensor_name)); in GetRendezvousKey()
120 TF_CHECK_OK(GetNodeAttr(node->attrs(), "send_device_incarnation", in GetRendezvousKey()
156 TF_CHECK_OK(Rendezvous::ParseKey(p.first, &parsed)); in RunWithArgs()
157 TF_CHECK_OK(rendez_->Send(parsed, Rendezvous::Args(), p.second, false)); in RunWithArgs()
159 TF_CHECK_OK(exec_->Run(args)); in RunWithArgs()
[all …]
Dprocess_function_library_runtime_test.cc38 TF_CHECK_OK(DeviceFactory::AddDevices(options, "/job:a/replica:0/task:0", in Init()
170 TF_CHECK_OK( in TEST_F()
185 TF_CHECK_OK(Run("FindDevice", opts, {}, instantiate_opts, {}, {&y})); in TEST_F()
202 TF_CHECK_OK( in TEST_F()
205 TF_CHECK_OK( in TEST_F()
220 TF_CHECK_OK(Run("FindDevice", opts, {}, instantiate_opts, {}, {&y})); in TEST_F()
224 TF_CHECK_OK(Run("FindDevice", opts, {}, instantiate_opts, {}, {&y})); in TEST_F()
240 TF_CHECK_OK(Run("FindDevice", opts, {}, instantiate_opts_0, {}, {&y})); in TEST_F()
246 TF_CHECK_OK(Run("FindDevice", opts, {}, instantiate_opts_1, {}, {&y})); in TEST_F()
/external/tensorflow/tensorflow/core/graph/
Dtestlib.cc61 TF_CHECK_OK(NodeBuilder(g->NewName("n"), "_Send") in Send()
78 TF_CHECK_OK(NodeBuilder(g->NewName("n"), "_Recv") in Recv()
91 TF_CHECK_OK(NodeBuilder(g->NewName("n"), "Const") in Constant()
100 TF_CHECK_OK(NodeBuilder(name, "Const") in Constant()
113 TF_CHECK_OK(NodeBuilder(name, "HostConst") in HostConstant()
122 TF_CHECK_OK(NodeBuilder(g->NewName("n"), "Variable") in Var()
132 TF_CHECK_OK(NodeBuilder(name, "Variable") in Var()
141 TF_CHECK_OK(NodeBuilder(g->NewName("n"), "Assign") in Assign()
152 TF_CHECK_OK(NodeBuilder(g->NewName("n"), reduce, g->op_registry()) in Reduce()
162 TF_CHECK_OK(NodeBuilder(g->NewName("n"), "Quantize") in QuantizeToUINT8()
[all …]
Dmkl_layout_pass.cc528 TF_CHECK_OK(GetNodeAttr(n->def(), attr_name, &value)); in GetTensorListLength()
531 TF_CHECK_OK(GetNodeAttr(n->def(), attr_name, &N)); in GetTensorListLength()
682 TF_CHECK_OK(n->input_node(0, &bias_add_grad_inp)); in IsBiasAddGradInConv2DWithBiasContext()
697 TF_CHECK_OK(e->dst()->input_node(2, &third_input)); in IsBiasAddGradInConv2DWithBiasContext()
709 TF_CHECK_OK(e->dst()->input_node(2, &third_input)); in IsBiasAddGradInConv2DWithBiasContext()
746 TF_CHECK_OK(backprop_input_node->input_node(1, &second_inp_of_input)); in IsBiasAddGradInConv2DWithBiasContext()
747 TF_CHECK_OK(backprop_filter_node->input_node(0, &first_inp_of_filter)); in IsBiasAddGradInConv2DWithBiasContext()
774 TF_CHECK_OK(backprop_filter_node->input_node(0, &first_inp_of_filter)); in IsBiasAddGradInConv2DWithBiasContext()
1037 TF_CHECK_OK(NodeBuilder((*g)->NewName("DMT"), "Const") in GetDummyMklTensorNode()
1057 TF_CHECK_OK( in GetDummyMklTensorNode()
[all …]
/external/tensorflow/tensorflow/core/grappler/clusters/
Dsingle_machine_test.cc48 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 …]
Dvirtual_cluster_test.cc47 TF_CHECK_OK(cluster_->Provision()); in SetUp()
51 TF_CHECK_OK(cluster_->Shutdown()); in TearDown()
69 TF_CHECK_OK(cluster_->Initialize(item)); in TEST_F()
72 TF_CHECK_OK(cluster_->Run(item.graph, item.feed, item.fetch, &metadata)); in TEST_F()
112 TF_CHECK_OK(root.ToGraphDef(&item.graph)); in TEST_F()
115 TF_CHECK_OK(cluster_->Initialize(item)); in TEST_F()
/external/tensorflow/tensorflow/core/distributed_runtime/rpc/
Dgrpc_session_test.cc94 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()
162 TF_CHECK_OK(test::TestCluster::MakeTestCluster(Devices(1, 0), 2, &cluster)); in TEST()
174 TF_CHECK_OK(session->Create(graph_copy)); in TEST()
175 TF_CHECK_OK(session->Run({}, {}, {node_names[2]}, nullptr)); in TEST()
176 TF_CHECK_OK(session->Close()); in TEST()
[all …]
/external/tensorflow/tensorflow/core/ops/
Darray_grad_test.cc51 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/contrib/tensorboard/db/
Dsummary_file_writer_test.cc50 TF_CHECK_OK(CreateSummaryFileWriter(1, 1, testing::TmpDir(), test_name, in SummaryTestHelper()
54 TF_CHECK_OK(writer_fn(writer)); in SummaryTestHelper()
55 TF_CHECK_OK(writer->Flush()); in SummaryTestHelper()
58 TF_CHECK_OK(env_.GetChildren(testing::TmpDir(), &files)); in SummaryTestHelper()
67 TF_CHECK_OK(env_.NewRandomAccessFile(io::JoinPath(testing::TmpDir(), f), in SummaryTestHelper()
72 TF_CHECK_OK( in SummaryTestHelper()
75 TF_CHECK_OK(reader.ReadRecord(&offset, &record)); in SummaryTestHelper()
91 TF_CHECK_OK(SummaryTestHelper("tensor_test", in TEST_F()
109 TF_CHECK_OK(SummaryTestHelper( in TEST_F()
127 TF_CHECK_OK(SummaryTestHelper("hist_test", in TEST_F()
[all …]
Dloader.cc77 TF_CHECK_OK(Sqlite::Open( in main()
83 TF_CHECK_OK(SetupTensorboardSqliteDb(db)); in main()
87 TF_CHECK_OK(CreateSummaryDbWriter(db, experiment_name, run_name, user_name, in main()
93 TF_CHECK_OK(env->NewRandomAccessFile(events, &file)); in main()
104 TF_CHECK_OK(s); in main()
110 TF_CHECK_OK(db_writer->WriteEvent(std::move(event))); in main()
/external/tensorflow/tensorflow/core/grappler/costs/
Dgraph_properties_test.cc43 TF_CHECK_OK(cluster_->Provision()); in SetUp()
47 TF_CHECK_OK(cluster_->Shutdown()); in TearDown()
80 TF_CHECK_OK(s); in TEST_F()
123 TF_CHECK_OK(cluster_->Initialize(item)); in TEST_F()
125 TF_CHECK_OK(s); in TEST_F()
165 TF_CHECK_OK(NodeDefBuilder("Var", "Variable") in TEST_F()
173 TF_CHECK_OK(NodeDefBuilder("InitialVal", "Const") in TEST_F()
177 TF_CHECK_OK(NodeDefBuilder("InitVar", "Assign") in TEST_F()
185 TF_CHECK_OK(static_properties.InferStatically(false)); in TEST_F()
197 TF_CHECK_OK(cluster_->Initialize(item)); in TEST_F()
[all …]
/external/tensorflow/tensorflow/core/util/
Dtensor_slice_reader_test.cc72 TF_CHECK_OK(writer.Add("test", shape, slice, data)); in SimpleFloatHelper()
73 TF_CHECK_OK(writer.Finish()); in SimpleFloatHelper()
96 TF_CHECK_OK(writer.Add("test", shape, slice, data)); in SimpleFloatHelper()
102 TF_CHECK_OK(writer.Add("test", shape, slice, data)); in SimpleFloatHelper()
104 TF_CHECK_OK(writer.Finish()); in SimpleFloatHelper()
197 TF_CHECK_OK(writer.Add("test", shape, slice, data)); in SimpleIntXHelper()
198 TF_CHECK_OK(writer.Finish()); in SimpleIntXHelper()
221 TF_CHECK_OK(writer.Add("test", shape, slice, data)); in SimpleIntXHelper()
227 TF_CHECK_OK(writer.Add("test", shape, slice, data)); in SimpleIntXHelper()
229 TF_CHECK_OK(writer.Finish()); in SimpleIntXHelper()
[all …]
Dtensor_slice_set_test.cc54 TF_CHECK_OK(tss.Register(slice_1, "", src_1)); in TEST()
63 TF_CHECK_OK(tss.Register(slice_2, "", src_2)); in TEST()
72 TF_CHECK_OK(tss.Register(slice_3, "", src_3)); in TEST()
153 TF_CHECK_OK(tss.Register(slice_1, "slice_1", nullptr)); in TEST()
161 TF_CHECK_OK(tss.Register(slice_2, "slice_2", nullptr)); in TEST()
169 TF_CHECK_OK(tss.Register(slice_3, "slice_3", nullptr)); in TEST()
245 TF_CHECK_OK(slice_set.Register(part, part.DebugString(), nullptr)); in BM_RegisterOneByOne()
/external/tensorflow/tensorflow/core/framework/
Dop_gen_lib_test.cc254 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 …]
Dcommon_shape_fns_test.cc53 TF_CHECK_OK(OpDefBuilder("Assert") in TEST()
60 TF_CHECK_OK(NodeDefBuilder("test", "Assert") in TEST()
73 TF_CHECK_OK(OpDefBuilder("L2Loss") in TEST()
80 TF_CHECK_OK( in TEST()
101 TF_CHECK_OK(OpDefBuilder("MatMul") in TEST()
111 TF_CHECK_OK(NodeDefBuilder("test", "MatMul") in TEST()
183 TF_CHECK_OK(NodeDefBuilder("test", "MatMul") in TEST()
201 TF_CHECK_OK(NodeDefBuilder("test", "MatMul") in TEST()
220 TF_CHECK_OK(OpDefBuilder("BiasAdd") in TEST()
228 TF_CHECK_OK(NodeDefBuilder("test", "BiasAdd") in TEST()
[all …]
Dresource_mgr_test.cc54 TF_CHECK_OK(rm.Lookup(container, name, &r)); in Find()
64 TF_CHECK_OK(rm->LookupOrCreate<T>(container, name, &r, [&label](T** ret) { in LookupOrCreate()
89 TF_CHECK_OK(rm.Create("foo", "bar", new Resource("cat"))); in TEST()
90 TF_CHECK_OK(rm.Create("foo", "baz", new Resource("dog"))); in TEST()
91 TF_CHECK_OK(rm.Create("foo", "bar", new Other("tiger"))); in TEST()
108 TF_CHECK_OK(rm.Delete<Resource>("foo", "bar")); in TEST()
111 TF_CHECK_OK(rm.Create("foo", "bar", new Resource("kitty"))); in TEST()
115 TF_CHECK_OK(rm.Cleanup("foo")); in TEST()
119 TF_CHECK_OK(rm.Cleanup("foo")); in TEST()
123 TF_CHECK_OK(rm.Cleanup("bar")); in TEST()
[all …]
/external/tensorflow/tensorflow/core/platform/
Denv_test.cc37 TF_CHECK_OK(WriteStringToFile(env, filename, input)); in CreateTestFile()
58 void SetUp() override { TF_CHECK_OK(env_->CreateDir(BaseDir())); } in SetUp()
62 TF_CHECK_OK( in TearDown()
204 TF_CHECK_OK(env_->RecursivelyCreateDir(create_path)); in TEST_F()
205 TF_CHECK_OK(env_->RecursivelyCreateDir(create_path)); // repeat creation. in TEST_F()
210 TF_CHECK_OK(env_->RecursivelyCreateDir("")); in TEST_F()
216 TF_CHECK_OK(env_->CreateDir(io::JoinPath(BaseDir(), "a"))); in TEST_F()
217 TF_CHECK_OK(env_->CreateDir(subdir_path)); in TEST_F()
222 TF_CHECK_OK(env_->RecursivelyCreateDir(create_path)); in TEST_F()
223 TF_CHECK_OK(env_->RecursivelyCreateDir(create_path)); // repeat creation. in TEST_F()
[all …]
/external/tensorflow/tensorflow/core/distributed_runtime/
Drpcbench_test.cc87 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/core/profiler/internal/
Dtfprof_show_test.cc48 TF_CHECK_OK( in TFProfShowTest()
56 TF_CHECK_OK( in TFProfShowTest()
63 TF_CHECK_OK(ReadBinaryProto(Env::Default(), op_log_path, op_log_pb.get())); in TFProfShowTest()
90 TF_CHECK_OK(ReadFileToString(Env::Default(), in TestToFromProto()
110 TF_CHECK_OK(ReadFileToString(Env::Default(), dump_file, &dump_str)); in TEST_F()
137 TF_CHECK_OK(ReadFileToString(Env::Default(), dump_file, &dump_str)); in TEST_F()
185 TF_CHECK_OK(ReadFileToString(Env::Default(), dump_file, &dump_str)); in TEST_F()
/external/tensorflow/tensorflow/cc/ops/
Dconst_op_test.cc62 TF_CHECK_OK(root.status()); in TEST()
71 TF_CHECK_OK(root.status()); in TEST()
75 TF_CHECK_OK(root.status()); in TEST()
79 TF_CHECK_OK(root.status()); in TEST()
83 TF_CHECK_OK(root.status()); in TEST()
93 TF_CHECK_OK(root.status()); in TEST()
98 TF_CHECK_OK(root.status()); in TEST()
112 TF_CHECK_OK(root.status()); in TEST()
/external/tensorflow/tensorflow/core/grappler/optimizers/
Dlayout_optimizer_test.cc169 TF_CHECK_OK(s.ToGraphDef(&item.graph)); in TEST_F()
192 TF_CHECK_OK(s.ToGraphDef(&item.graph)); in TEST_F()
214 TF_CHECK_OK(s.ToGraphDef(&item.graph)); in TEST_F()
227 TF_CHECK_OK(s.ToGraphDef(&item.graph)); in TEST_F()
240 TF_CHECK_OK(s.ToGraphDef(&item.graph)); in TEST_F()
253 TF_CHECK_OK(s.ToGraphDef(&item.graph)); in TEST_F()
266 TF_CHECK_OK(s.ToGraphDef(&item.graph)); in TEST_F()
281 TF_CHECK_OK(s.ToGraphDef(&item.graph)); in TEST_F()
308 TF_CHECK_OK(s.ToGraphDef(&item.graph)); in TEST_F()
345 TF_CHECK_OK(s.ToGraphDef(&item.graph)); in TEST_F()
[all …]
/external/tensorflow/tensorflow/cc/training/
Dqueue_runner_test.cc105 TF_CHECK_OK(session->Create(graph_def)); in BuildSessionAndInitVariable()
107 TF_CHECK_OK(session->Run({}, {}, {kAssignOpName}, nullptr)); in BuildSessionAndInitVariable()
120 TF_CHECK_OK(qr->Start(session.get())); in TEST()
206 TF_CHECK_OK(session->Create(graph_def)); in TEST()
212 TF_CHECK_OK(qr->Start(session.get())); in TEST()
243 TF_CHECK_OK(session->Create(graph_def)); in TEST()
249 TF_CHECK_OK(qr->Start(session.get())); in TEST()
287 TF_CHECK_OK(session->Create(graph_def)); in TEST()
304 TF_CHECK_OK(session->Create(graph_def)); in TEST()
316 TF_CHECK_OK(qr0->Start(session.get())); in TEST()
[all …]
/external/tensorflow/tensorflow/core/kernels/
Dvariable_ops_test.cc37 TF_CHECK_OK( in ManyManyVariablesHelper()
51 TF_CHECK_OK(sess->Create(gd)); in ManyManyVariablesHelper()
52 TF_CHECK_OK(sess->Run({}, {}, targets, nullptr)); in ManyManyVariablesHelper()
55 TF_CHECK_OK(sess->Run({}, {}, targets, nullptr)); in ManyManyVariablesHelper()

1234567891011