Home
last modified time | relevance | path

Searched refs:TF_ASSERT_OK_AND_ASSIGN (Results 1 – 25 of 133) sorted by relevance

123456

/external/tensorflow/tensorflow/compiler/xla/service/
Dwhile_loop_trip_count_annotator_test.cc53 TF_ASSERT_OK_AND_ASSIGN(auto m, ParseAndReturnVerifiedModule(kModuleStr)); in TEST_F()
55 TF_ASSERT_OK_AND_ASSIGN(bool changed, RunHloPass(&pass, m.get())); in TEST_F()
58 TF_ASSERT_OK_AND_ASSIGN(auto config, in TEST_F()
89 TF_ASSERT_OK_AND_ASSIGN(auto m, ParseAndReturnVerifiedModule(kModuleStr)); in TEST_F()
91 TF_ASSERT_OK_AND_ASSIGN(bool changed, RunHloPass(&pass, m.get())); in TEST_F()
94 TF_ASSERT_OK_AND_ASSIGN(auto config, in TEST_F()
125 TF_ASSERT_OK_AND_ASSIGN(auto m, ParseAndReturnVerifiedModule(kModuleStr)); in TEST_F()
127 TF_ASSERT_OK_AND_ASSIGN(bool changed, RunHloPass(&pass, m.get())); in TEST_F()
130 TF_ASSERT_OK_AND_ASSIGN(auto config, in TEST_F()
161 TF_ASSERT_OK_AND_ASSIGN(auto m, ParseAndReturnVerifiedModule(kModuleStr)); in TEST_F()
[all …]
Dwhile_loop_constant_sinking_test.cc56 TF_ASSERT_OK_AND_ASSIGN(auto module, in TEST_F()
59 TF_ASSERT_OK_AND_ASSIGN(bool changed, in TEST_F()
96 TF_ASSERT_OK_AND_ASSIGN(auto module, in TEST_F()
99 TF_ASSERT_OK_AND_ASSIGN(bool changed, in TEST_F()
137 TF_ASSERT_OK_AND_ASSIGN(auto module, in TEST_F()
140 TF_ASSERT_OK_AND_ASSIGN(bool changed, in TEST_F()
185 TF_ASSERT_OK_AND_ASSIGN(auto module, in TEST_F()
188 TF_ASSERT_OK_AND_ASSIGN(bool changed, in TEST_F()
227 TF_ASSERT_OK_AND_ASSIGN(auto module, in TEST_F()
230 TF_ASSERT_OK_AND_ASSIGN(bool changed, in TEST_F()
[all …]
Dhlo_verifier_test.cc170 TF_ASSERT_OK_AND_ASSIGN(auto module, in TEST_F()
200 TF_ASSERT_OK_AND_ASSIGN(auto module, in TEST_F()
235 TF_ASSERT_OK_AND_ASSIGN(auto module, in TEST_F()
266 TF_ASSERT_OK_AND_ASSIGN(auto module, in TEST_F()
285 TF_ASSERT_OK_AND_ASSIGN(auto module, in TEST_F()
305 TF_ASSERT_OK_AND_ASSIGN(auto module, in TEST_F()
325 TF_ASSERT_OK_AND_ASSIGN(auto module, in TEST_F()
343 TF_ASSERT_OK_AND_ASSIGN(auto module, in TEST_F()
409 TF_ASSERT_OK_AND_ASSIGN(auto module, in TEST_F()
421 TF_ASSERT_OK_AND_ASSIGN(auto module, in TEST_F()
[all …]
Dslice_sinker_test.cc63 TF_ASSERT_OK_AND_ASSIGN(auto module, in TEST_F()
66 TF_ASSERT_OK_AND_ASSIGN(bool result, RunHloPass(&slice_sinker, module.get())); in TEST_F()
103 TF_ASSERT_OK_AND_ASSIGN(auto module, in TEST_F()
106 TF_ASSERT_OK_AND_ASSIGN(bool result, RunHloPass(&slice_sinker, module.get())); in TEST_F()
149 TF_ASSERT_OK_AND_ASSIGN(auto module, in TEST_F()
152 TF_ASSERT_OK_AND_ASSIGN(bool result, RunHloPass(&slice_sinker, module.get())); in TEST_F()
198 TF_ASSERT_OK_AND_ASSIGN(auto module, in TEST_F()
201 TF_ASSERT_OK_AND_ASSIGN(bool result, RunHloPass(&slice_sinker, module.get())); in TEST_F()
238 TF_ASSERT_OK_AND_ASSIGN(auto module, in TEST_F()
241 TF_ASSERT_OK_AND_ASSIGN(bool result, RunHloPass(&slice_sinker, module.get())); in TEST_F()
[all …]
Dhlo_evaluator_test.cc89 TF_ASSERT_OK_AND_ASSIGN(Literal result, Evaluate()); in TestUnaryOp()
109 TF_ASSERT_OK_AND_ASSIGN(Literal result, Evaluate()); in TestBinaryOp()
127 TF_ASSERT_OK_AND_ASSIGN(Literal result, Evaluate()); in TestTernaryOp()
184 TF_ASSERT_OK_AND_ASSIGN(Literal result, Evaluate()); in ReduceWindowMaxIotaTest()
235 TF_ASSERT_OK_AND_ASSIGN(Literal result, Evaluate()); in TEST_P()
261 TF_ASSERT_OK_AND_ASSIGN(Literal result, Evaluate()); in TEST_P()
283 TF_ASSERT_OK_AND_ASSIGN(Literal result, Evaluate()); in TEST_P()
308 TF_ASSERT_OK_AND_ASSIGN(Literal result, Evaluate({})); in TEST_P()
483 TF_ASSERT_OK_AND_ASSIGN(Literal result, Evaluate(args)); in TEST_F()
494 TF_ASSERT_OK_AND_ASSIGN(auto literal, in TEST_F()
[all …]
Dhlo_creation_utils_test.cc69 TF_ASSERT_OK_AND_ASSIGN(HloInstruction * first_1_dims_collapsed, in TEST_F()
74 TF_ASSERT_OK_AND_ASSIGN( in TEST_F()
88 TF_ASSERT_OK_AND_ASSIGN(HloInstruction * first_2_dims_collapsed, in TEST_F()
93 TF_ASSERT_OK_AND_ASSIGN( in TEST_F()
111 TF_ASSERT_OK_AND_ASSIGN(HloInstruction * with_1_degenerate_dim_prepended, in TEST_F()
116 TF_ASSERT_OK_AND_ASSIGN( in TEST_F()
130 TF_ASSERT_OK_AND_ASSIGN(HloInstruction * with_2_degenerate_dims_prepended, in TEST_F()
135 TF_ASSERT_OK_AND_ASSIGN( in TEST_F()
149 TF_ASSERT_OK_AND_ASSIGN(HloInstruction * with_2_degenerate_dims_prepended, in TEST_F()
154 TF_ASSERT_OK_AND_ASSIGN( in TEST_F()
[all …]
Dhlo_domain_test.cc139 TF_ASSERT_OK_AND_ASSIGN(auto module, in TEST_F()
144 TF_ASSERT_OK_AND_ASSIGN(bool isolator_changed, isolator.Run(module.get())); in TEST_F()
156 TF_ASSERT_OK_AND_ASSIGN(bool remover_changed, remover.Run(module.get())); in TEST_F()
182 TF_ASSERT_OK_AND_ASSIGN(auto module, in TEST_F()
187 TF_ASSERT_OK_AND_ASSIGN(bool isolator_changed, isolator.Run(module.get())); in TEST_F()
210 TF_ASSERT_OK_AND_ASSIGN(auto module, in TEST_F()
215 TF_ASSERT_OK_AND_ASSIGN(bool isolator_changed, isolator.Run(module.get())); in TEST_F()
226 TF_ASSERT_OK_AND_ASSIGN(bool remover_changed, remover.Run(module.get())); in TEST_F()
248 TF_ASSERT_OK_AND_ASSIGN(auto module, in TEST_F()
253 TF_ASSERT_OK_AND_ASSIGN(bool isolator_changed, isolator.Run(module.get())); in TEST_F()
[all …]
Dhlo_replication_analysis_test.cc72 TF_ASSERT_OK_AND_ASSIGN(auto module, in TEST_F()
77 TF_ASSERT_OK_AND_ASSIGN(std::unique_ptr<HloReplicationAnalysis> analysis, in TEST_F()
147 TF_ASSERT_OK_AND_ASSIGN(auto module, in TEST_F()
149 TF_ASSERT_OK_AND_ASSIGN( in TEST_F()
206 TF_ASSERT_OK_AND_ASSIGN(auto module, in TEST_F()
211 TF_ASSERT_OK_AND_ASSIGN(std::unique_ptr<HloReplicationAnalysis> analysis, in TEST_F()
259 TF_ASSERT_OK_AND_ASSIGN(auto module, in TEST_F()
264 TF_ASSERT_OK_AND_ASSIGN(std::unique_ptr<HloReplicationAnalysis> analysis, in TEST_F()
309 TF_ASSERT_OK_AND_ASSIGN(auto module, in TEST_F()
314 TF_ASSERT_OK_AND_ASSIGN(std::unique_ptr<HloReplicationAnalysis> analysis, in TEST_F()
[all …]
Dhlo_constant_folding_test.cc57 TF_ASSERT_OK_AND_ASSIGN(bool result, const_folder.Run(module.get())); in TEST_F()
79 TF_ASSERT_OK_AND_ASSIGN(bool result, const_folder.Run(module.get())); in TEST_F()
101 TF_ASSERT_OK_AND_ASSIGN(bool result, const_folder.Run(module.get())); in TEST_F()
141 TF_ASSERT_OK_AND_ASSIGN(bool result, const_folder.Run(module.get())); in TEST_F()
156 TF_ASSERT_OK_AND_ASSIGN(auto literal, in TEST_F()
168 TF_ASSERT_OK_AND_ASSIGN(bool result, const_folder.Run(module.get())); in TEST_F()
179 TF_ASSERT_OK_AND_ASSIGN(auto literal, in TEST_F()
193 TF_ASSERT_OK_AND_ASSIGN(bool result, const_folder.Run(module.get())); in TEST_F()
226 TF_ASSERT_OK_AND_ASSIGN(auto m, in TEST_F()
229 TF_ASSERT_OK_AND_ASSIGN(bool result, const_folder.Run(m.get())); in TEST_F()
[all …]
Dar_crs_combiner_test.cc41 TF_ASSERT_OK_AND_ASSIGN(std::unique_ptr<HloModule> module, in TEST_F()
61 TF_ASSERT_OK_AND_ASSIGN(std::unique_ptr<HloModule> module, in TEST_F()
80 TF_ASSERT_OK_AND_ASSIGN(std::unique_ptr<HloModule> module, in TEST_F()
101 TF_ASSERT_OK_AND_ASSIGN(std::unique_ptr<HloModule> module, in TEST_F()
121 TF_ASSERT_OK_AND_ASSIGN(std::unique_ptr<HloModule> module, in TEST_F()
141 TF_ASSERT_OK_AND_ASSIGN(std::unique_ptr<HloModule> module, in TEST_F()
163 TF_ASSERT_OK_AND_ASSIGN(std::unique_ptr<HloModule> module, in TEST_F()
185 TF_ASSERT_OK_AND_ASSIGN(std::unique_ptr<HloModule> module, in TEST_F()
208 TF_ASSERT_OK_AND_ASSIGN(std::unique_ptr<HloModule> module, in TEST_F()
244 TF_ASSERT_OK_AND_ASSIGN(std::unique_ptr<HloModule> module, in TEST_F()
[all …]
Dhlo_schedule_test.cc54 TF_ASSERT_OK_AND_ASSIGN(std::unique_ptr<HloModule> module, in TEST_F()
56 TF_ASSERT_OK_AND_ASSIGN( in TEST_F()
88 TF_ASSERT_OK_AND_ASSIGN(std::unique_ptr<HloModule> module, in TEST_F()
90 TF_ASSERT_OK_AND_ASSIGN( in TEST_F()
137 TF_ASSERT_OK_AND_ASSIGN(std::unique_ptr<HloModule> module, in TEST_F()
139 TF_ASSERT_OK_AND_ASSIGN( in TEST_F()
181 TF_ASSERT_OK_AND_ASSIGN(std::unique_ptr<HloModule> module, in TEST_F()
183 TF_ASSERT_OK_AND_ASSIGN( in TEST_F()
242 TF_ASSERT_OK_AND_ASSIGN(std::unique_ptr<HloModule> module, in TEST_F()
244 TF_ASSERT_OK_AND_ASSIGN( in TEST_F()
[all …]
Dhlo_pass_pipeline_test.cc109 TF_ASSERT_OK_AND_ASSIGN(std::unique_ptr<VerifiedHloModule> module, in TEST_F()
116 TF_ASSERT_OK_AND_ASSIGN(bool changed, pipeline.Run(module.get())); in TEST_F()
132 TF_ASSERT_OK_AND_ASSIGN(std::unique_ptr<VerifiedHloModule> module, in TEST_F()
137 TF_ASSERT_OK_AND_ASSIGN(bool changed, pipeline.Run(module.get())); in TEST_F()
162 TF_ASSERT_OK_AND_ASSIGN(HloModuleGroup module_group, in TEST_F()
176 TF_ASSERT_OK_AND_ASSIGN(bool changed, in TEST_F()
194 TF_ASSERT_OK_AND_ASSIGN(std::unique_ptr<VerifiedHloModule> module, in TEST_F()
202 TF_ASSERT_OK_AND_ASSIGN(bool changed, pipeline.Run(module.get())); in TEST_F()
246 TF_ASSERT_OK_AND_ASSIGN(std::unique_ptr<VerifiedHloModule> module, in TEST_F()
Dhlo_module_group_test.cc46 TF_ASSERT_OK_AND_ASSIGN(std::unique_ptr<HloModule> module, in TEST_F()
55 TF_ASSERT_OK_AND_ASSIGN(HloModuleGroup group_copy, in TEST_F()
85 TF_ASSERT_OK_AND_ASSIGN(std::unique_ptr<HloModule> module_0, in TEST_F()
87 TF_ASSERT_OK_AND_ASSIGN(std::unique_ptr<HloModule> module_1, in TEST_F()
100 TF_ASSERT_OK_AND_ASSIGN(HloModuleGroup group_copy, in TEST_F()
124 TF_ASSERT_OK_AND_ASSIGN(std::unique_ptr<HloModule> module_0, in TEST_F()
126 TF_ASSERT_OK_AND_ASSIGN(std::unique_ptr<HloModule> module_1, in TEST_F()
180 TF_ASSERT_OK_AND_ASSIGN(std::unique_ptr<HloModule> module, in TEST_F()
188 TF_ASSERT_OK_AND_ASSIGN(auto metadata, in TEST_F()
Dbatch_dot_simplification_test.cc40 TF_ASSERT_OK_AND_ASSIGN(std::unique_ptr<HloModule> m, in TEST_F()
64 TF_ASSERT_OK_AND_ASSIGN(std::unique_ptr<HloModule> m, in TEST_F()
88 TF_ASSERT_OK_AND_ASSIGN(std::unique_ptr<HloModule> m, in TEST_F()
112 TF_ASSERT_OK_AND_ASSIGN(std::unique_ptr<HloModule> m, in TEST_F()
136 TF_ASSERT_OK_AND_ASSIGN(std::unique_ptr<HloModule> m, in TEST_F()
160 TF_ASSERT_OK_AND_ASSIGN(std::unique_ptr<HloModule> m, in TEST_F()
187 TF_ASSERT_OK_AND_ASSIGN(std::unique_ptr<HloModule> m, in TEST_F()
208 TF_ASSERT_OK_AND_ASSIGN(std::unique_ptr<HloModule> m, in TEST_F()
/external/tensorflow/tensorflow/compiler/xla/service/cpu/
Dparallel_task_assignment_test.cc60 TF_ASSERT_OK_AND_ASSIGN(std::unique_ptr<HloModule> m, in TEST_F()
62 TF_ASSERT_OK_AND_ASSIGN(bool changed, RunParallelTaskAssigner(m.get())); in TEST_F()
88 TF_ASSERT_OK_AND_ASSIGN(std::unique_ptr<HloModule> m, in TEST_F()
90 TF_ASSERT_OK_AND_ASSIGN(bool changed, RunParallelTaskAssigner(m.get())); in TEST_F()
105 TF_ASSERT_OK_AND_ASSIGN(std::unique_ptr<HloModule> m, in TEST_F()
107 TF_ASSERT_OK_AND_ASSIGN(bool changed, RunParallelTaskAssigner(m.get())); in TEST_F()
122 TF_ASSERT_OK_AND_ASSIGN(std::unique_ptr<HloModule> m, in TEST_F()
124 TF_ASSERT_OK_AND_ASSIGN(bool changed, RunParallelTaskAssigner(m.get())); in TEST_F()
167 TF_ASSERT_OK_AND_ASSIGN(std::unique_ptr<HloModule> m, in TEST_F()
169 TF_ASSERT_OK_AND_ASSIGN(bool changed, RunParallelTaskAssigner(m.get())); in TEST_F()
/external/tensorflow/tensorflow/compiler/xla/tests/
Ddeconstruct_tuple_test.cc68 TF_ASSERT_OK_AND_ASSIGN(literal, client_->Transfer(*handles[0])); in TEST_F()
70 TF_ASSERT_OK_AND_ASSIGN(literal, client_->Transfer(*handles[1])); in TEST_F()
90 TF_ASSERT_OK_AND_ASSIGN(literal, client_->Transfer(*handles1[0])); in TEST_F()
92 TF_ASSERT_OK_AND_ASSIGN(literal, client_->Transfer(*handles1[1])); in TEST_F()
98 TF_ASSERT_OK_AND_ASSIGN(literal, client_->Transfer(*handles2[0])); in TEST_F()
100 TF_ASSERT_OK_AND_ASSIGN(literal, client_->Transfer(*handles2[1])); in TEST_F()
120 TF_ASSERT_OK_AND_ASSIGN(literal, client_->Transfer(*handles[0])); in XLA_TEST_F()
122 TF_ASSERT_OK_AND_ASSIGN(literal, client_->Transfer(*handles[1])); in XLA_TEST_F()
124 TF_ASSERT_OK_AND_ASSIGN(literal, client_->Transfer(*handles[2])); in XLA_TEST_F()
126 TF_ASSERT_OK_AND_ASSIGN(literal, client_->Transfer(*handles[3])); in XLA_TEST_F()
[all …]
Dclient_test.cc48 TF_ASSERT_OK_AND_ASSIGN(auto computation, b.Build()); in XLA_TEST_F()
55 TF_ASSERT_OK_AND_ASSIGN( in XLA_TEST_F()
62 TF_ASSERT_OK_AND_ASSIGN( in XLA_TEST_F()
78 TF_ASSERT_OK_AND_ASSIGN(auto computation, b.Build()); in XLA_TEST_F()
91 TF_ASSERT_OK_AND_ASSIGN( in XLA_TEST_F()
119 TF_ASSERT_OK_AND_ASSIGN(std::unique_ptr<GlobalData> const_arg, in XLA_TEST_F()
126 TF_ASSERT_OK_AND_ASSIGN(add_with_one_arg, b.Build()); in XLA_TEST_F()
132 TF_ASSERT_OK_AND_ASSIGN(std::vector<xla::DeviceHandle> devices, in XLA_TEST_F()
141 TF_ASSERT_OK_AND_ASSIGN(auto results, in XLA_TEST_F()
145 TF_ASSERT_OK_AND_ASSIGN( in XLA_TEST_F()
Doutfeed_in_nested_computation_test.cc41 TF_ASSERT_OK_AND_ASSIGN(XlaComputation loop_cond, [&] { in XLA_TEST_F()
51 TF_ASSERT_OK_AND_ASSIGN(XlaComputation loop_body, [&] { in XLA_TEST_F()
71 TF_ASSERT_OK_AND_ASSIGN(XlaComputation computation, b.Build()); in XLA_TEST_F()
89 TF_ASSERT_OK_AND_ASSIGN(Literal r, in XLA_TEST_F()
102 TF_ASSERT_OK_AND_ASSIGN(Literal r, in XLA_TEST_F()
110 TF_ASSERT_OK_AND_ASSIGN(Literal r, in XLA_TEST_F()
127 TF_ASSERT_OK_AND_ASSIGN(XlaComputation true_computation, [&] { in XLA_TEST_F()
135 TF_ASSERT_OK_AND_ASSIGN(XlaComputation false_computation, [&] { in XLA_TEST_F()
146 TF_ASSERT_OK_AND_ASSIGN(XlaComputation computation, b.Build()); in XLA_TEST_F()
159 TF_ASSERT_OK_AND_ASSIGN(Literal r, in XLA_TEST_F()
Dxla_hlo_profile_test.cc140 TF_ASSERT_OK_AND_ASSIGN( in ExecuteAndFetchProfile()
144 TF_ASSERT_OK_AND_ASSIGN( in ExecuteAndFetchProfile()
151 TF_ASSERT_OK_AND_ASSIGN( in ExecuteAndFetchProfile()
160 TF_ASSERT_OK_AND_ASSIGN( in ExecuteAndFetchProfile()
178 TF_ASSERT_OK_AND_ASSIGN( in ExecuteAndFetchProfile()
195 TF_ASSERT_OK_AND_ASSIGN(se::Platform * platform, in XLA_TEST_F()
197 TF_ASSERT_OK_AND_ASSIGN(LocalClient * client, in XLA_TEST_F()
205 TF_ASSERT_OK_AND_ASSIGN(auto computation, builder.Build()); in XLA_TEST_F()
240 TF_ASSERT_OK_AND_ASSIGN(ParsedProfileOutputLine total_profile, in XLA_TEST_F()
242 TF_ASSERT_OK_AND_ASSIGN(ParsedProfileOutputLine dot_profile, in XLA_TEST_F()
[all …]
Dtoken_hlo_test.cc37 TF_ASSERT_OK_AND_ASSIGN(Literal result, Execute(std::move(module), {})); in XLA_TEST_F()
49 TF_ASSERT_OK_AND_ASSIGN(Literal result, Execute(std::move(module), {})); in XLA_TEST_F()
66 TF_ASSERT_OK_AND_ASSIGN(Literal result, Execute(std::move(module), {})); in XLA_TEST_F()
146 TF_ASSERT_OK_AND_ASSIGN( in XLA_TEST_F()
184 TF_ASSERT_OK_AND_ASSIGN( in XLA_TEST_F()
188 TF_ASSERT_OK_AND_ASSIGN(Literal result, Execute(std::move(module), {&arg})); in XLA_TEST_F()
194 TF_ASSERT_OK_AND_ASSIGN( in XLA_TEST_F()
198 TF_ASSERT_OK_AND_ASSIGN(Literal result, Execute(std::move(module), {&arg})); in XLA_TEST_F()
222 TF_ASSERT_OK_AND_ASSIGN( in XLA_TEST_F()
243 TF_ASSERT_OK_AND_ASSIGN( in XLA_TEST_F()
[all …]
Dprng_test.cc206 TF_ASSERT_OK_AND_ASSIGN(std::unique_ptr<GlobalData> param0_data, in XLA_TEST_F()
213 TF_ASSERT_OK_AND_ASSIGN(auto computation, builder.Build()); in XLA_TEST_F()
217 TF_ASSERT_OK_AND_ASSIGN( in XLA_TEST_F()
252 TF_ASSERT_OK_AND_ASSIGN(auto computation, build_computation()); in XLA_TEST_F()
253 TF_ASSERT_OK_AND_ASSIGN( in XLA_TEST_F()
260 TF_ASSERT_OK_AND_ASSIGN(auto computation, build_computation()); in XLA_TEST_F()
261 TF_ASSERT_OK_AND_ASSIGN( in XLA_TEST_F()
264 TF_ASSERT_OK_AND_ASSIGN( in XLA_TEST_F()
273 TF_ASSERT_OK_AND_ASSIGN(auto computation, build_computation()); in XLA_TEST_F()
274 TF_ASSERT_OK_AND_ASSIGN( in XLA_TEST_F()
[all …]
Dtransfer_manager_test.cc79 TF_ASSERT_OK_AND_ASSIGN( in XLA_TEST_F()
95 TF_ASSERT_OK_AND_ASSIGN( in XLA_TEST_F()
117 TF_ASSERT_OK_AND_ASSIGN( in XLA_TEST_F()
135 TF_ASSERT_OK_AND_ASSIGN( in XLA_TEST_F()
153 TF_ASSERT_OK_AND_ASSIGN( in XLA_TEST_F()
171 TF_ASSERT_OK_AND_ASSIGN( in XLA_TEST_F()
187 TF_ASSERT_OK_AND_ASSIGN( in XLA_TEST_F()
207 TF_ASSERT_OK_AND_ASSIGN( in XLA_TEST_F()
227 TF_ASSERT_OK_AND_ASSIGN( in XLA_TEST_F()
241 TF_ASSERT_OK_AND_ASSIGN( in XLA_TEST_F()
[all …]
/external/tensorflow/tensorflow/compiler/xla/client/
Dxla_builder_test.cc73 TF_ASSERT_OK_AND_ASSIGN(auto module, BuildHloModule(&b)); in TEST_F()
84 TF_ASSERT_OK_AND_ASSIGN(auto module, BuildHloModule(&b)); in TEST_F()
98 TF_ASSERT_OK_AND_ASSIGN(auto module, BuildHloModule(&b)); in TEST_F()
129 TF_ASSERT_OK_AND_ASSIGN(auto module, BuildHloModule(&b)); in TEST_F()
143 TF_ASSERT_OK_AND_ASSIGN(auto module, BuildHloModule(&b)); in TEST_F()
159 TF_ASSERT_OK_AND_ASSIGN(auto module, BuildHloModule(&b)); in TEST_F()
184 TF_ASSERT_OK_AND_ASSIGN(auto module, BuildHloModule(&b)); in TEST_F()
197 TF_ASSERT_OK_AND_ASSIGN(auto add_shape, b.GetShape(add)); in TEST_F()
200 TF_ASSERT_OK_AND_ASSIGN(auto module, BuildHloModule(&b)); in TEST_F()
209 TF_ASSERT_OK_AND_ASSIGN(auto module, BuildHloModule(&b)); in TEST_F()
[all …]
/external/tensorflow/tensorflow/compiler/xla/python/
Dshared_device_buffer_test.cc31 TF_ASSERT_OK_AND_ASSIGN( in TEST()
48 TF_ASSERT_OK_AND_ASSIGN( in TEST()
52 TF_ASSERT_OK_AND_ASSIGN( in TEST()
56 TF_ASSERT_OK_AND_ASSIGN( in TEST()
78 TF_ASSERT_OK_AND_ASSIGN( in TEST()
82 TF_ASSERT_OK_AND_ASSIGN( in TEST()
86 TF_ASSERT_OK_AND_ASSIGN( in TEST()
92 TF_ASSERT_OK_AND_ASSIGN( in TEST()
96 TF_ASSERT_OK_AND_ASSIGN( in TEST()
139 TF_ASSERT_OK_AND_ASSIGN( in TEST()
/external/tensorflow/tensorflow/compiler/tf2xla/
Dxla_expression_test.cc85 TF_ASSERT_OK_AND_ASSIGN(xla::XlaComputation computation, in TEST_F()
87 TF_ASSERT_OK_AND_ASSIGN(xla::Literal value, in TEST_F()
97 TF_ASSERT_OK_AND_ASSIGN(TensorShape resource_shape, in TEST_F()
101 TF_ASSERT_OK_AND_ASSIGN(TensorShape op_shape, in TEST_F()
105 TF_ASSERT_OK_AND_ASSIGN(TensorShape constant_shape, in TEST_F()
116 TF_ASSERT_OK_AND_ASSIGN( in TEST_F()
122 TF_ASSERT_OK_AND_ASSIGN(absl::optional<Tensor> op_nonconstant, in TEST_F()
127 TF_ASSERT_OK_AND_ASSIGN( in TEST_F()

123456