Home
last modified time | relevance | path

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

12345

/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.cc57 TF_ASSERT_OK_AND_ASSIGN(std::unique_ptr<HloModule> module, in TEST_F()
60 TF_ASSERT_OK_AND_ASSIGN(bool changed, in TEST_F()
97 TF_ASSERT_OK_AND_ASSIGN(std::unique_ptr<HloModule> module, in TEST_F()
100 TF_ASSERT_OK_AND_ASSIGN(bool changed, in TEST_F()
138 TF_ASSERT_OK_AND_ASSIGN(std::unique_ptr<HloModule> module, in TEST_F()
141 TF_ASSERT_OK_AND_ASSIGN(bool changed, in TEST_F()
186 TF_ASSERT_OK_AND_ASSIGN(std::unique_ptr<HloModule> module, in TEST_F()
189 TF_ASSERT_OK_AND_ASSIGN(bool changed, in TEST_F()
228 TF_ASSERT_OK_AND_ASSIGN(std::unique_ptr<HloModule> module, in TEST_F()
231 TF_ASSERT_OK_AND_ASSIGN(bool changed, in TEST_F()
[all …]
Dhlo_creation_utils_test.cc54 TF_ASSERT_OK_AND_ASSIGN(HloInstruction * first_1_dims_collapsed, in TEST_F()
59 TF_ASSERT_OK_AND_ASSIGN( in TEST_F()
73 TF_ASSERT_OK_AND_ASSIGN(HloInstruction * first_2_dims_collapsed, in TEST_F()
78 TF_ASSERT_OK_AND_ASSIGN( in TEST_F()
96 TF_ASSERT_OK_AND_ASSIGN(HloInstruction * with_1_degenerate_dim_prepended, in TEST_F()
101 TF_ASSERT_OK_AND_ASSIGN( in TEST_F()
115 TF_ASSERT_OK_AND_ASSIGN(HloInstruction * with_2_degenerate_dims_prepended, in TEST_F()
120 TF_ASSERT_OK_AND_ASSIGN( in TEST_F()
134 TF_ASSERT_OK_AND_ASSIGN(HloInstruction * with_2_degenerate_dims_prepended, in TEST_F()
139 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_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()
166 TF_ASSERT_OK_AND_ASSIGN(Literal result, Evaluate()); in TEST_P()
192 TF_ASSERT_OK_AND_ASSIGN(Literal result, Evaluate()); in TEST_P()
214 TF_ASSERT_OK_AND_ASSIGN(Literal result, Evaluate()); in TEST_P()
239 TF_ASSERT_OK_AND_ASSIGN(Literal result, Evaluate({})); in TEST_P()
407 TF_ASSERT_OK_AND_ASSIGN(Literal result, Evaluate(args)); in TEST_F()
418 TF_ASSERT_OK_AND_ASSIGN(auto literal, in TEST_F()
431 TF_ASSERT_OK_AND_ASSIGN(Literal result, Evaluate({})); 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 …]
Dhlo_verifier_test.cc169 TF_ASSERT_OK_AND_ASSIGN(auto module, ParseHloString(hlo_string)); in TEST_F()
198 TF_ASSERT_OK_AND_ASSIGN(auto module, ParseHloString(hlo_string)); in TEST_F()
217 TF_ASSERT_OK_AND_ASSIGN(auto module, ParseHloString(hlo_string)); in TEST_F()
235 TF_ASSERT_OK_AND_ASSIGN(auto module, ParseHloString(hlo_string)); in TEST_F()
254 TF_ASSERT_OK_AND_ASSIGN(auto module, ParseHloString(hlo_string)); in TEST_F()
273 TF_ASSERT_OK_AND_ASSIGN(auto module, ParseHloString(hlo_string)); in TEST_F()
290 TF_ASSERT_OK_AND_ASSIGN(auto module, ParseHloString(hlo_string)); in TEST_F()
355 TF_ASSERT_OK_AND_ASSIGN(auto module, ParseHloString(kConvHloString)); in TEST_F()
366 TF_ASSERT_OK_AND_ASSIGN(auto module, ParseHloString(kConvHloString)); in TEST_F()
386 TF_ASSERT_OK_AND_ASSIGN(auto module, ParseHloString(kAddWithLayoutChangeHlo)); 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.cc55 TF_ASSERT_OK_AND_ASSIGN(std::unique_ptr<HloModule> module, in TEST_F()
57 TF_ASSERT_OK_AND_ASSIGN( in TEST_F()
89 TF_ASSERT_OK_AND_ASSIGN(std::unique_ptr<HloModule> module, in TEST_F()
91 TF_ASSERT_OK_AND_ASSIGN( in TEST_F()
138 TF_ASSERT_OK_AND_ASSIGN(std::unique_ptr<HloModule> module, in TEST_F()
140 TF_ASSERT_OK_AND_ASSIGN( in TEST_F()
182 TF_ASSERT_OK_AND_ASSIGN(std::unique_ptr<HloModule> module, in TEST_F()
184 TF_ASSERT_OK_AND_ASSIGN( in TEST_F()
243 TF_ASSERT_OK_AND_ASSIGN(std::unique_ptr<HloModule> module, in TEST_F()
245 TF_ASSERT_OK_AND_ASSIGN( in TEST_F()
[all …]
Dhlo_module_group_test.cc47 TF_ASSERT_OK_AND_ASSIGN(std::unique_ptr<HloModule> module, in TEST_F()
56 TF_ASSERT_OK_AND_ASSIGN(HloModuleGroup group_copy, in TEST_F()
86 TF_ASSERT_OK_AND_ASSIGN(std::unique_ptr<HloModule> module_0, in TEST_F()
88 TF_ASSERT_OK_AND_ASSIGN(std::unique_ptr<HloModule> module_1, in TEST_F()
101 TF_ASSERT_OK_AND_ASSIGN(HloModuleGroup group_copy, in TEST_F()
125 TF_ASSERT_OK_AND_ASSIGN(std::unique_ptr<HloModule> module_0, in TEST_F()
127 TF_ASSERT_OK_AND_ASSIGN(std::unique_ptr<HloModule> module_1, in TEST_F()
179 TF_ASSERT_OK_AND_ASSIGN( in TEST_F()
187 TF_ASSERT_OK_AND_ASSIGN(auto metadata, in TEST_F()
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()
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()
Dwhile_util_test.cc68 TF_ASSERT_OK_AND_ASSIGN( in TEST()
75 TF_ASSERT_OK_AND_ASSIGN( in TEST()
99 TF_ASSERT_OK_AND_ASSIGN( in TEST()
106 TF_ASSERT_OK_AND_ASSIGN( in TEST()
154 TF_ASSERT_OK_AND_ASSIGN(std::unique_ptr<HloModule> module, in TEST()
195 TF_ASSERT_OK_AND_ASSIGN(std::unique_ptr<HloModule> module, in TEST()
202 TF_ASSERT_OK_AND_ASSIGN( in TEST()
Dhlo_element_type_converter_test.cc44 TF_ASSERT_OK_AND_ASSIGN(bool converted, type_converter.Run(module.get())); in TEST_F()
60 TF_ASSERT_OK_AND_ASSIGN(bool converted, type_converter.Run(module.get())); in TEST_F()
79 TF_ASSERT_OK_AND_ASSIGN(bool converted, type_converter.Run(module.get())); in TEST_F()
107 TF_ASSERT_OK_AND_ASSIGN(bool converted, type_converter.Run(module.get())); in TEST_F()
131 TF_ASSERT_OK_AND_ASSIGN(bool converted, type_converter.Run(module.get())); in TEST_F()
158 TF_ASSERT_OK_AND_ASSIGN(bool converted, type_converter.Run(module.get())); in TEST_F()
189 TF_ASSERT_OK_AND_ASSIGN(bool converted, RunHloPass(&converter, module.get())); in TEST_F()
Ddynamic_index_splitter_test.cc48 TF_ASSERT_OK_AND_ASSIGN(auto module, ParseHloString(kDynamicSlice, config)); in TEST_F()
49 TF_ASSERT_OK_AND_ASSIGN(bool changed, in TEST_F()
85 TF_ASSERT_OK_AND_ASSIGN(auto module, in TEST_F()
87 TF_ASSERT_OK_AND_ASSIGN(bool changed, in TEST_F()
124 TF_ASSERT_OK_AND_ASSIGN(auto module, ParseHloString(kDynamicSlice, config)); in TEST_F()
125 TF_ASSERT_OK_AND_ASSIGN(bool changed, 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()
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 …]
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()
Dtoken_hlo_test.cc38 TF_ASSERT_OK_AND_ASSIGN(Literal result, Execute(std::move(module), {})); in XLA_TEST_F()
50 TF_ASSERT_OK_AND_ASSIGN(Literal result, Execute(std::move(module), {})); in XLA_TEST_F()
67 TF_ASSERT_OK_AND_ASSIGN(Literal result, Execute(std::move(module), {})); in XLA_TEST_F()
147 TF_ASSERT_OK_AND_ASSIGN( in XLA_TEST_F()
185 TF_ASSERT_OK_AND_ASSIGN( in XLA_TEST_F()
189 TF_ASSERT_OK_AND_ASSIGN(Literal result, Execute(std::move(module), {&arg})); in XLA_TEST_F()
195 TF_ASSERT_OK_AND_ASSIGN( in XLA_TEST_F()
199 TF_ASSERT_OK_AND_ASSIGN(Literal result, Execute(std::move(module), {&arg})); in XLA_TEST_F()
223 TF_ASSERT_OK_AND_ASSIGN( in XLA_TEST_F()
244 TF_ASSERT_OK_AND_ASSIGN( in XLA_TEST_F()
[all …]
Dprng_test.cc198 TF_ASSERT_OK_AND_ASSIGN(std::unique_ptr<GlobalData> param0_data, in XLA_TEST_F()
205 TF_ASSERT_OK_AND_ASSIGN(auto computation, builder.Build()); in XLA_TEST_F()
209 TF_ASSERT_OK_AND_ASSIGN( in XLA_TEST_F()
243 TF_ASSERT_OK_AND_ASSIGN(auto computation, build_computation()); in XLA_TEST_F()
244 TF_ASSERT_OK_AND_ASSIGN( in XLA_TEST_F()
251 TF_ASSERT_OK_AND_ASSIGN(auto computation, build_computation()); in XLA_TEST_F()
252 TF_ASSERT_OK_AND_ASSIGN( in XLA_TEST_F()
255 TF_ASSERT_OK_AND_ASSIGN( in XLA_TEST_F()
264 TF_ASSERT_OK_AND_ASSIGN(auto computation, build_computation()); in XLA_TEST_F()
265 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()
113 TF_ASSERT_OK_AND_ASSIGN( in XLA_TEST_F()
129 TF_ASSERT_OK_AND_ASSIGN( in XLA_TEST_F()
145 TF_ASSERT_OK_AND_ASSIGN( in XLA_TEST_F()
165 TF_ASSERT_OK_AND_ASSIGN( in XLA_TEST_F()
185 TF_ASSERT_OK_AND_ASSIGN( in XLA_TEST_F()
199 TF_ASSERT_OK_AND_ASSIGN( in XLA_TEST_F()
218 TF_ASSERT_OK_AND_ASSIGN( in XLA_TEST_F()
233 TF_ASSERT_OK_AND_ASSIGN( in XLA_TEST_F()
[all …]
/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()
/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/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()

12345