Home
last modified time | relevance | path

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

12345678910>>...21

/external/tensorflow/tensorflow/core/lib/io/
Dinputbuffer_test.cc46 TF_CHECK_OK(env->NewRandomAccessFile(fname, &file)); in TEST()
57 TF_CHECK_OK( in TEST()
62 TF_CHECK_OK(env->NewRandomAccessFile(fname, &file)); in TEST()
65 TF_CHECK_OK(in.ReadLine(&line)); in TEST()
67 TF_CHECK_OK(in.ReadLine(&line)); in TEST()
69 TF_CHECK_OK(in.ReadLine(&line)); in TEST()
85 TF_CHECK_OK(env->NewRandomAccessFile(fname, &file)); in TEST()
88 TF_CHECK_OK(in.ReadLine(&line)); in TEST()
90 TF_CHECK_OK(in.ReadLine(&line)); in TEST()
92 TF_CHECK_OK(in.ReadLine(&line)); in TEST()
[all …]
Drecord_reader_writer_test.cc50 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/common_runtime/
Dkernel_benchmark_testlib.cc115 TF_CHECK_OK(NewExecutor(executor_type, params, *init, &init_exec)); in Benchmark()
119 TF_CHECK_OK(init_exec->Run(args)); in Benchmark()
122 TF_CHECK_OK(NewExecutor(executor_type, params, *g, &exec_)); in Benchmark()
149 TF_CHECK_OK(GetNodeAttr(node->attrs(), "send_device", &send_device)); in GetRendezvousKey()
151 TF_CHECK_OK(GetNodeAttr(node->attrs(), "recv_device", &recv_device)); in GetRendezvousKey()
153 TF_CHECK_OK(GetNodeAttr(node->attrs(), "tensor_name", &tensor_name)); in GetRendezvousKey()
155 TF_CHECK_OK(GetNodeAttr(node->attrs(), "send_device_incarnation", in GetRendezvousKey()
182 TF_CHECK_OK(Rendezvous::ParseKey(p.first, &parsed)); in RunWithRendezvousArgs()
183 TF_CHECK_OK(rendez_->Send(parsed, Rendezvous::Args(), p.second, false)); in RunWithRendezvousArgs()
185 TF_CHECK_OK(exec_->Run(args)); in RunWithRendezvousArgs()
[all …]
Ddynamic_device_mgr_test.cc76 TF_CHECK_OK(dm->AddDevices(std::move(added_devices))); in TEST()
90 TF_CHECK_OK(dm->AddDevices(std::move(devices))); in TEST()
94 TF_CHECK_OK(dm->RemoveDevices(removed_devices)); in TEST()
111 TF_CHECK_OK(dm->AddDevices(std::move(added_devices))); in TEST()
113 TF_CHECK_OK(dm->RemoveDevices(removed_devices)); in TEST()
122 TF_CHECK_OK(dm->AddDevices(std::move(added_devices))); in TEST()
124 TF_CHECK_OK(dm->RemoveDevices(removed_devices)); in TEST()
139 TF_CHECK_OK(dm->AddDevices(std::move(devices))); in TEST()
143 TF_CHECK_OK(dm->RemoveDevicesByName(removed_devices)); in TEST()
154 TF_CHECK_OK(dm->AddDevices(std::move(devices))); in TEST()
[all …]
Dprocess_function_library_runtime_test.cc106 TF_CHECK_OK(DeviceFactory::AddDevices(options, "/job:a/replica:0/task:0", in ProcessFunctionLibraryRuntimeTest()
113 TF_CHECK_OK(device_mgr_->AddDevices(std::move(created_devices))); in ProcessFunctionLibraryRuntimeTest()
114 TF_CHECK_OK(device_mgr_->LookupDevice( in ProcessFunctionLibraryRuntimeTest()
116 TF_CHECK_OK(device_mgr_->LookupDevice( in ProcessFunctionLibraryRuntimeTest()
224 TF_CHECK_OK(pflr->IsCrossProcess(handle, &is_cross_process)); in RunWithRuntime()
388 TF_CHECK_OK( in TEST_F()
401 TF_CHECK_OK(Run("FindDevice", opts, {}, instantiate_opts, {}, {&y})); in TEST_F()
419 TF_CHECK_OK( in TEST_F()
422 TF_CHECK_OK( in TEST_F()
435 TF_CHECK_OK(Run("FindDevice", opts, {}, instantiate_opts, {}, {&y})); in TEST_F()
[all …]
Dmkl_layout_pass.cc1047 TF_CHECK_OK(GetNodeAttr(n->def(), attr_name, &value)); in GetTensorListLength()
1050 TF_CHECK_OK(GetNodeAttr(n->def(), attr_name, &N)); in GetTensorListLength()
1124 TF_CHECK_OK(GetNodeAttr(m->def(), "T", &T_m)); in GetConv2DOrBiasAdd()
1136 TF_CHECK_OK(m->input_node(0, &n)); in GetConv2DOrBiasAdd()
1169 TF_CHECK_OK(GetNodeAttr(m->def(), "T", &T_m)); in GetPadOrConv2D()
1205 TF_CHECK_OK(GetNodeAttr(conv_node->def(), "padding", &padding)); in GetPadOrConv2D()
1254 TF_CHECK_OK(GetNodeAttr(conv_node->def(), "padding", &padding)); in GetPadOrFusedConv2D()
1292 TF_CHECK_OK(GetNodeAttr(m->def(), "T", &T_m)); in GetConv2DBackpropFilterOrBiasAddGrad()
1305 TF_CHECK_OK(m->input_node(0, &g)); in GetConv2DBackpropFilterOrBiasAddGrad()
1322 TF_CHECK_OK(m->input_node(2, &g)); in GetConv2DBackpropFilterOrBiasAddGrad()
[all …]
/external/tensorflow/tensorflow/core/kernels/data/
Dhash_utils_test.cc47 TF_CHECK_OK(HashNode(graph_def, *node, &hash)); in GetHash()
74 TF_CHECK_OK(HashNode(graph, node, &hash)); in GetHash()
80 TF_CHECK_OK(HashTensor(tensor, &hash)); in GetHash()
162 TF_CHECK_OK(NodeDefBuilder("O", "Add") in TEST_F()
166 TF_CHECK_OK(NodeDefBuilder("A", "Abs") in TEST_F()
169 TF_CHECK_OK(NodeDefBuilder("B", "Add") in TEST_F()
173 TF_CHECK_OK(NodeDefBuilder("C", "Ceil") in TEST_F()
176 TF_CHECK_OK(NodeDefBuilder("D", "Cos") in TEST_F()
179 TF_CHECK_OK(NodeDefBuilder("E", "Floor") in TEST_F()
195 TF_CHECK_OK(NodeDefBuilder("graph_1/node_1", "Const") in TEST_F()
[all …]
/external/tensorflow/tensorflow/core/graph/
Dtestlib.cc38 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 …]
/external/tensorflow/tensorflow/compiler/jit/
Dextract_outside_compilation_pass_test.cc53 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 …]
Drearrange_function_argument_pass_test.cc53 TF_CHECK_OK(s.ToGraph(g.get())); in TEST()
55 TF_CHECK_OK(GraphToFunctionDef(*g, "f1", xla_fdef)); in TEST()
68 TF_CHECK_OK(s.ToGraph(g.get())); in TEST()
70 TF_CHECK_OK(GraphToFunctionDef(*g, "f2", xla_fdef)); in TEST()
81 TF_CHECK_OK(s.ToGraph(g.get())); in TEST()
83 TF_CHECK_OK(GraphToFunctionDef(*g, "f3", xla_fdef)); in TEST()
110 TF_CHECK_OK(s.ToGraph(g.get())); in TEST()
113 TF_CHECK_OK(RearrangeFunctionArguments( in TEST()
140 TF_CHECK_OK(if_node->input_node(1, &input_node)); in TEST()
142 TF_CHECK_OK(if_node->input_node(2, &input_node)); in TEST()
[all …]
/external/tensorflow/tensorflow/core/grappler/clusters/
Dsingle_machine_test.cc54 TF_CHECK_OK(cluster_->EnablePeakMemoryStats()); in SetUp()
55 TF_CHECK_OK(cluster_->Provision()); in SetUp()
60 TF_CHECK_OK(cluster_->Shutdown()); in TearDown()
79 TF_CHECK_OK(cluster_->Initialize(item)); in TEST_F()
83 TF_CHECK_OK(cluster_->Run(item.graph, item.feed, item.fetch, &metadata)); in TEST_F()
117 TF_CHECK_OK(cluster_->Initialize(item)); in TEST_F()
119 TF_CHECK_OK(cluster_->Run(item.graph, item.feed, item.fetch, &metadata)); in TEST_F()
129 TF_CHECK_OK(cluster_->Initialize(item)); in TEST_F()
131 TF_CHECK_OK(cluster_->Run(item.graph, item.feed, item.fetch, &metadata1)); in TEST_F()
133 TF_CHECK_OK(cluster_->Run(item.graph, item.feed, item.fetch, &metadata2)); in TEST_F()
[all …]
/external/tensorflow/tensorflow/core/grappler/inputs/
Dutils_test.cc30 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/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()
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/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/core/summary/
Dsummary_file_writer_test.cc52 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/c/kernels/
Dbitcast_op_test.cc105 TF_CHECK_OK(OpRegistry::Global()->LookUp("Bitcast", &reg)); in TEST()
108 TF_CHECK_OK(NodeDefBuilder("dummy", &op_def) in TEST()
115 TF_CHECK_OK(c.input("input", &input_shapes)); in TEST()
117 TF_CHECK_OK(reg->shape_inference_fn(&c)); in TEST()
123 TF_CHECK_OK(OpRegistry::Global()->LookUp("Bitcast", &reg)); in TEST()
126 TF_CHECK_OK(NodeDefBuilder("dummy", &op_def) in TEST()
134 TF_CHECK_OK(c.input("input", &input_shapes)); in TEST()
136 TF_CHECK_OK(reg->shape_inference_fn(&c)); in TEST()
142 TF_CHECK_OK(OpRegistry::Global()->LookUp("Bitcast", &reg)); in TEST()
145 TF_CHECK_OK(NodeDefBuilder("dummy", &op_def) in TEST()
[all …]
/external/tensorflow/tensorflow/python/client/
Ddebug_events_writer_wrapper.cc46 TF_CHECK_OK( in PYBIND11_MODULE()
56 TF_CHECK_OK( in PYBIND11_MODULE()
66 TF_CHECK_OK( in PYBIND11_MODULE()
76 TF_CHECK_OK( in PYBIND11_MODULE()
86 TF_CHECK_OK( in PYBIND11_MODULE()
96 TF_CHECK_OK( in PYBIND11_MODULE()
105 TF_CHECK_OK(DebugEventsWriter::LookUpDebugEventsWriter(dump_root, &writer)); in PYBIND11_MODULE()
110 TF_CHECK_OK(DebugEventsWriter::LookUpDebugEventsWriter(dump_root, &writer)); in PYBIND11_MODULE()
115 TF_CHECK_OK(DebugEventsWriter::LookUpDebugEventsWriter(dump_root, &writer)); in PYBIND11_MODULE()
120 TF_CHECK_OK(DebugEventsWriter::LookUpDebugEventsWriter(dump_root, &writer)); in PYBIND11_MODULE()
/external/tensorflow/tensorflow/core/lib/io/snappy/
Dsnappy_test.cc103 TF_CHECK_OK( in TestMultipleWritesWriteFile()
110 TF_CHECK_OK( in TestMultipleWritesWriteFile()
112 TF_CHECK_OK(corrupt_file_writer->Flush()); in TestMultipleWritesWriteFile()
113 TF_CHECK_OK(corrupt_file_writer->Close()); in TestMultipleWritesWriteFile()
210 TF_CHECK_OK(env->NewWritableFile(fname, &file_writer)); in TestTellWriteFile()
213 TF_CHECK_OK(out.Write(StringPiece(data))); in TestTellWriteFile()
214 TF_CHECK_OK(out.Flush()); in TestTellWriteFile()
215 TF_CHECK_OK(file_writer->Flush()); in TestTellWriteFile()
216 TF_CHECK_OK(file_writer->Close()); in TestTellWriteFile()
233 TF_CHECK_OK(env->NewRandomAccessFile(fname, &file_reader)); in TestTell()
[all …]
/external/tensorflow/tensorflow/core/grappler/optimizers/
Dgraph_optimizer_stage_test.cc149 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/util/
Dtensor_slice_reader_test.cc73 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/platform/
Denv_test.cc65 void SetUp() override { TF_CHECK_OK(env_->CreateDir(BaseDir())); } in SetUp()
69 TF_CHECK_OK( in TearDown()
222 TF_CHECK_OK(env_->RecursivelyCreateDir(create_path)); in TEST_F()
223 TF_CHECK_OK(env_->RecursivelyCreateDir(create_path)); // repeat creation. in TEST_F()
228 TF_CHECK_OK(env_->RecursivelyCreateDir("")); in TEST_F()
234 TF_CHECK_OK(env_->CreateDir(io::JoinPath(BaseDir(), "a"))); in TEST_F()
235 TF_CHECK_OK(env_->CreateDir(subdir_path)); in TEST_F()
240 TF_CHECK_OK(env_->RecursivelyCreateDir(create_path)); in TEST_F()
241 TF_CHECK_OK(env_->RecursivelyCreateDir(create_path)); // repeat creation. in TEST_F()
364 TF_CHECK_OK(env->RecursivelyCreateDir(create_path)); in TEST_F()
[all …]
/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 …]
Dresource_mgr_test.cc60 TF_CHECK_OK(rm.Lookup(container, name, &r)); in Find()
70 TF_CHECK_OK(rm->LookupOrCreate<T>(container, name, &r, [&label](T** ret) { in LookupOrCreate()
95 TF_CHECK_OK(rm.Create("foo", "bar", new Resource("cat"))); in TEST()
96 TF_CHECK_OK(rm.Create("foo", "baz", new Resource("dog"))); in TEST()
97 TF_CHECK_OK(rm.Create("foo", "bar", new Other("tiger"))); in TEST()
114 TF_CHECK_OK(rm.Delete<Resource>("foo", "bar")); in TEST()
117 TF_CHECK_OK(rm.Create("foo", "bar", new Resource("kitty"))); in TEST()
121 TF_CHECK_OK(rm.Cleanup("foo")); in TEST()
125 TF_CHECK_OK(rm.Cleanup("foo")); in TEST()
129 TF_CHECK_OK(rm.Cleanup("bar")); in TEST()
[all …]
/external/tensorflow/tensorflow/compiler/xla/service/
Ddynamic_dimension_inference_test.cc116 TF_CHECK_OK(module_->dynamic_parameter_binding().Bind( in TEST_F()
135 TF_CHECK_OK(module_->dynamic_parameter_binding().Bind( in TEST_F()
161 TF_CHECK_OK(module_->dynamic_parameter_binding().Bind( in TEST_F()
192 TF_CHECK_OK(module_->dynamic_parameter_binding().Bind( in TEST_F()
223 TF_CHECK_OK(module_->dynamic_parameter_binding().Bind( in TEST_F()
255 TF_CHECK_OK(module_->dynamic_parameter_binding().Bind( in TEST_F()
279 TF_CHECK_OK(module_->dynamic_parameter_binding().Bind( in TEST_F()
332 TF_CHECK_OK(module_->dynamic_parameter_binding().Bind( in TEST_F()
337 TF_CHECK_OK(module_->dynamic_parameter_binding().Bind( in TEST_F()
340 TF_CHECK_OK(module_->dynamic_parameter_binding().Bind( 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()
208 TF_CHECK_OK(session->Run({{"x", x}}, {"y:0"}, {}, &outputs)); in BM_Helper()
211 TF_CHECK_OK(session->Close()); in BM_Helper()

12345678910>>...21