Home
last modified time | relevance | path

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

12345

/external/tensorflow/tensorflow/compiler/xla/service/
Dwhile_loop_concat_code_motion_test.cc91 op::Tuple(op::GetTupleElement(loop), op::Slice(op::GetTupleElement(loop)), in TEST_F()
92 op::Slice(op::GetTupleElement(loop)))); in TEST_F()
200 op::Tuple(op::GetTupleElement(loop), op::Slice(op::GetTupleElement(loop)), in TEST_F()
201 op::Slice(op::GetTupleElement(loop)), in TEST_F()
202 op::Slice(op::GetTupleElement(loop)), in TEST_F()
203 op::Slice(op::GetTupleElement(loop)))); in TEST_F()
268 op::Tuple(op::GetTupleElement(loop), op::Slice(op::GetTupleElement(loop)), in TEST_F()
269 op::Slice(op::GetTupleElement(loop)), in TEST_F()
270 op::Slice(op::GetTupleElement(loop)), in TEST_F()
271 op::Slice(op::GetTupleElement(loop)))); in TEST_F()
[all …]
Dconditional_code_motion_test.cc79 EXPECT_THAT(root, AllOf(op::Tuple(op::Convert(), op::GetTupleElement()))); in TEST_F()
293 op::Convert(op::Reshape(op::GetTupleElement(op::Conditional()))), in TEST_F()
294 op::Convert(op::Reshape(op::GetTupleElement(op::Conditional()))))))); in TEST_F()
374 op::GetTupleElement(op::Conditional(), *on_false_sub_idx), in TEST_F()
376 op::GetTupleElement(op::Conditional(), *on_false_add_idx)))); in TEST_F()
441 op::GetTupleElement(op::Conditional(), *on_false_gte_idx), in TEST_F()
442 op::GetTupleElement(op::Conditional(), *on_false_const_idx))); in TEST_F()
494 op::Tuple(op::GetTupleElement(op::Parameter(0), 0), in TEST_F()
495 op::GetTupleElement(op::Parameter(0), 0))); in TEST_F()
514 op::GetTupleElement(op::Parameter(0), 0)); in TEST_F()
[all …]
Dwhile_util_test.cc88 op::Tuple(op::GetTupleElement(::testing::Eq(new_while_instr), 0), in TEST_F()
89 op::GetTupleElement(::testing::Eq(new_while_instr), 1))); in TEST_F()
92 op::Tuple(op::GetTupleElement(op::Parameter(0), 0), in TEST_F()
93 op::GetTupleElement(op::Parameter(0), 1)); in TEST_F()
96 op::Tuple(op::GetTupleElement(param_reconstructed, 0), in TEST_F()
97 op::GetTupleElement(param_reconstructed, 1))); in TEST_F()
122 op::Tuple(op::GetTupleElement(::testing::Eq(new_while_instr), 0), in TEST_F()
123 op::GetTupleElement(::testing::Eq(new_while_instr), 1))); in TEST_F()
126 op::Tuple(op::GetTupleElement(op::Parameter(0), 0), in TEST_F()
127 op::GetTupleElement(op::Parameter(0), 1)); in TEST_F()
[all …]
Dtuple_util_test.cc52 EXPECT_THAT(prefix, op::Tuple(op::GetTupleElement(op::Parameter(0), 0), in TEST_F()
53 op::GetTupleElement(op::Parameter(0), 1))); in TEST_F()
76 EXPECT_THAT(with_suffix, op::Tuple(op::GetTupleElement(op::Parameter(0), 0), in TEST_F()
77 op::GetTupleElement(op::Parameter(0), 1), in TEST_F()
78 op::GetTupleElement(op::Parameter(0), 2), in TEST_F()
123 op::GetTupleElement(op::Parameter(0), 1))); in TEST_F()
146 op::Tuple(op::GetTupleElement(op::Parameter(0), 0), in TEST_F()
148 op::GetTupleElement( in TEST_F()
149 op::GetTupleElement(op::Parameter(0), 1), 1)))); in TEST_F()
170 EXPECT_THAT(new_gte, op::GetTupleElement(existing_gte, 0)); in TEST_F()
Dwhile_loop_constant_sinking_test.cc153 op::GetTupleElement(op::Parameter(0)), in TEST_F()
154 op::GetTupleElement(op::Parameter(0)))); in TEST_F()
193 op::Tuple(op::GetTupleElement(op::Constant(), 0), in TEST_F()
194 op::GetTupleElement(op::Parameter(0)))); in TEST_F()
242 op::GetTupleElement(op::Parameter(0)), in TEST_F()
243 op::GetTupleElement(op::Parameter(0)))); in TEST_F()
283 op::Tuple(op::GetTupleElement(), op::GetTupleElement(), in TEST_F()
284 op::GetTupleElement())); in TEST_F()
369 op::Lt(_, op::GetTupleElement(op::Constant()))); in TEST_F()
Dtopk_rewriter_test.cc140 op::Tuple(op::GetTupleElement(op::CustomCall(op::Parameter(0)), 0), in TEST_F()
141 op::GetTupleElement(op::CustomCall(op::Parameter(0)), 1))); in TEST_F()
171 op::Tuple(op::GetTupleElement(op::CustomCall(op::Parameter(0)), 0), in TEST_F()
172 op::GetTupleElement(op::CustomCall(op::Parameter(0)), 1))); in TEST_F()
202 op::Tuple(op::GetTupleElement(op::CustomCall(op::Parameter(0)), 0), in TEST_F()
203 op::GetTupleElement(op::CustomCall(op::Parameter(0)), 1))); in TEST_F()
234 op::Tuple(op::Transpose(op::GetTupleElement( in TEST_F()
236 op::Transpose(op::GetTupleElement( in TEST_F()
263 op::GetTupleElement(op::CustomCall(op::Parameter(0)), 0)); in TEST_F()
Dstable_sort_expander_test.cc108 EXPECT_THAT(root, GmockMatch(m::GetTupleElement( in TEST_F()
154 EXPECT_THAT(root, GmockMatch(m::GetTupleElement( in TEST_F()
186 m::GetTupleElement( in TEST_F()
188 m::GetTupleElement( in TEST_F()
252 EXPECT_THAT(root, GmockMatch(m::GetTupleElement( in TEST_F()
289 EXPECT_THAT(root, GmockMatch(m::GetTupleElement( in TEST_F()
320 EXPECT_THAT(root, GmockMatch(m::GetTupleElement( in TEST_F()
351 EXPECT_THAT(root, GmockMatch(m::Negate(m::GetTupleElement( in TEST_F()
Dall_gather_combiner_test.cc83 op::Tuple(op::GetTupleElement(combined_all_gather, 0), in TEST_F()
84 op::GetTupleElement(combined_all_gather, 1))); in TEST_F()
121 op::Tuple(op::GetTupleElement(combined_all_gather0, 0), in TEST_F()
122 op::GetTupleElement(combined_all_gather0, 1), in TEST_F()
123 op::GetTupleElement(combined_all_gather1, 0), in TEST_F()
124 op::GetTupleElement(combined_all_gather1, 1), in TEST_F()
125 op::GetTupleElement(combined_all_gather0, 2))); in TEST_F()
/external/tensorflow/tensorflow/compiler/xla/client/lib/
Dsorting_test.cc35 xla::GetTupleElement(xla::TopK(x, 3), 0); in XLA_TEST_F()
43 xla::GetTupleElement(xla::TopK(x_rev, 3), 1); in XLA_TEST_F()
52 xla::GetTupleElement(xla::TopK(x_rev, 3), 1); in XLA_TEST_F()
65 xla::GetTupleElement(xla::TopK(x, kSize), 0); in XLA_TEST_F()
75 xla::GetTupleElement(xla::TopK(a, 5), 1); in XLA_TEST_F()
83 xla::GetTupleElement(xla::TopKWithPartitions(x, 3, /*num_partitions=*/2), 0); in XLA_TEST_F()
91 xla::GetTupleElement(xla::TopKWithPartitions(x_rev, 3, /*num_partitions=*/2), in XLA_TEST_F()
100 xla::GetTupleElement(xla::TopKWithPartitions(x, 3, /*num_partitions=*/3), 0); in XLA_TEST_F()
108 xla::GetTupleElement(xla::TopKWithPartitions(x_rev, 3, /*num_partitions=*/3), in XLA_TEST_F()
117 xla::GetTupleElement(xla::TopKWithPartitions(x, 3, /*num_partitions=*/5), 0); in XLA_TEST_F()
[all …]
Dlu_decomposition.cc52 return LuDecompositionResult{GetTupleElement(result, 0), in LuDecomposition()
53 GetTupleElement(result, 1), in LuDecomposition()
54 GetTupleElement(result, 2)}; in LuDecomposition()
/external/tensorflow/tensorflow/compiler/xla/tests/
Dtuple_test.cc119 XLA_TEST_F(TupleTest, GetTupleElement) { in XLA_TEST_F() argument
129 GetTupleElement(tuple_data, 1); in XLA_TEST_F()
141 GetTupleElement(tuple_data, 1); in XLA_TEST_F()
148 GetTupleElement(value, 1); in XLA_TEST_F()
168 auto vector_element = GetTupleElement(tuple_data, 0); in XLA_TEST_F()
169 auto matrix_element = GetTupleElement(tuple_data, 1); in XLA_TEST_F()
198 {GetTupleElement(tuple_data, 1), GetTupleElement(tuple_data, 0)}); in XLA_TEST_F()
232 auto new_tuple01 = Tuple(&builder, {GetTupleElement(tuple_data, 0), in XLA_TEST_F()
233 GetTupleElement(tuple_data, 1)}); in XLA_TEST_F()
234 auto new_tuple10 = Tuple(&builder, {GetTupleElement(tuple_data, 1), in XLA_TEST_F()
[all …]
Dwhile_test.cc365 auto iteration = GetTupleElement(prev, 0); in XLA_TEST_F()
376 auto iteration = GetTupleElement(prev, 0); in XLA_TEST_F()
377 auto w1 = GetTupleElement(prev, 1); in XLA_TEST_F()
378 auto w2 = GetTupleElement(prev, 2); in XLA_TEST_F()
379 auto w3 = GetTupleElement(prev, 3); in XLA_TEST_F()
418 auto iteration = GetTupleElement(prev, 0); in XLA_TEST_F()
429 auto iteration = GetTupleElement(prev, 0); in XLA_TEST_F()
430 auto w1 = GetTupleElement(prev, 1); in XLA_TEST_F()
431 auto w2 = GetTupleElement(prev, 2); in XLA_TEST_F()
432 auto w3 = GetTupleElement(prev, 3); in XLA_TEST_F()
[all …]
Dconditional_test.cc85 auto x = GetTupleElement(tuple, 0); in CreateTupleCeilComputation()
86 auto y = GetTupleElement(tuple, 1); in CreateTupleCeilComputation()
107 auto x = GetTupleElement(tuple, 0); in CreateTupleFloorComputation()
108 auto y = GetTupleElement(tuple, 1); in CreateTupleFloorComputation()
129 auto x = GetTupleElement(tuple, 0); in CreateTupleAddComputation()
130 auto y = GetTupleElement(tuple, 1); in CreateTupleAddComputation()
149 auto x = GetTupleElement(tuple, 0); in CreateTupleSubComputation()
150 auto y = GetTupleElement(tuple, 1); in CreateTupleSubComputation()
426 GetTupleElement(p, 0)), in XLA_TEST_P()
427 GetTupleElement(p, 1)); in XLA_TEST_P()
[all …]
/external/tensorflow/tensorflow/compiler/xla/tools/
Dhlo_control_flow_flattening_test.cc72 EXPECT_THAT(root, op::Tuple(op::GetTupleElement(while_op, 0), in TEST_F()
73 op::GetTupleElement(while_op, 1))); in TEST_F()
75 op::While(op::Tuple(op::GetTupleElement(), op::GetTupleElement(), in TEST_F()
80 op::Compare(op::GetTupleElement(op::Parameter(0), 2), op::Constant())); in TEST_F()
84 op::Tuple(op::GetTupleElement(), op::GetTupleElement(), in TEST_F()
85 op::Add(op::GetTupleElement(op::Parameter(0), 2), in TEST_F()
133 EXPECT_THAT(root, op::Tuple(op::GetTupleElement(while_op, 0), in TEST_F()
134 op::GetTupleElement(while_op, 1))); in TEST_F()
136 op::While(op::Tuple(op::GetTupleElement(), op::GetTupleElement(), in TEST_F()
141 op::Compare(op::GetTupleElement(op::Parameter(0), 2), op::Constant())); in TEST_F()
[all …]
/external/tensorflow/tensorflow/compiler/xla/service/spmd/
Dspmd_partitioner_test.cc292 op::GetTupleElement(op::Parameter()), op::Broadcast())))); in TEST_F()
296 op::GetTupleElement(op::Parameter()), op::Broadcast())))); in TEST_F()
323 op::DynamicSlice(op::GetTupleElement(op::Parameter()), offset, in TEST_F()
326 op::DynamicSlice(op::GetTupleElement(op::Parameter()), offset, in TEST_F()
348 op::GetTupleElement( in TEST_F()
370 root, AllOf(op::Shape("f32[5,2]"), op::GetTupleElement(op::Conditional( in TEST_F()
380 op::Tuple(op::Pad(op::GetTupleElement(second_infeed), in TEST_F()
382 op::GetTupleElement(second_infeed)))); in TEST_F()
401 op::GetTupleElement(op::Conditional( in TEST_F()
412 op::Tuple(op::Tuple(op::Pad(op::GetTupleElement( in TEST_F()
[all …]
/external/tensorflow/tensorflow/compiler/xla/service/gpu/
Dcudnn_pad_for_convolutions_test.cc96 m::Slice(m::GetTupleElement(m::CustomCall( in TEST_F()
117 EXPECT_THAT(root, GmockMatch(m::GetTupleElement(m::Tuple( in TEST_F()
118 m::Slice(m::GetTupleElement(m::CustomCall( in TEST_F()
140 GmockMatch(m::GetTupleElement(m::Tuple( in TEST_F()
141 m::Slice(m::GetTupleElement(m::CustomCall( in TEST_F()
162 EXPECT_THAT(root, GmockMatch(m::GetTupleElement(m::Tuple( in TEST_F()
163 m::Slice(m::GetTupleElement(m::CustomCall( in TEST_F()
232 m::Slice(m::GetTupleElement(m::CustomCall( in TEST_F()
255 m::Slice(m::GetTupleElement(m::CustomCall( in TEST_F()
279 m::Slice(m::GetTupleElement(m::CustomCall( in TEST_F()
[all …]
Dcudnn_fused_conv_rewriter_test.cc455 GmockMatch(m::GetTupleElement( in TEST_F()
492 GmockMatch(m::GetTupleElement( in TEST_F()
533 m::GetTupleElement( in TEST_F()
582 GmockMatch(m::GetTupleElement( in TEST_F()
631 m::GetTupleElement( in TEST_F()
676 m::GetTupleElement( in TEST_F()
714 m::GetTupleElement( in TEST_F()
755 m::GetTupleElement( in TEST_F()
795 m::GetTupleElement(m::CustomCall(&conv1), 0), in TEST_F()
799 m::GetTupleElement(m::CustomCall(&conv2), 0)))))); in TEST_F()
[all …]
Dcudnn_simplify_padding_test.cc144 m::Slice(m::Reshape(m::GetTupleElement(m::CustomCall( in TEST_F()
146 m::GetTupleElement( in TEST_F()
177 GmockMatch(m::Pad(&pad, m::GetTupleElement(m::CustomCall(), 0), in TEST_F()
236 m::Pad(&pad, m::Reshape(m::GetTupleElement(m::CustomCall(), 0)), in TEST_F()
269 m::Pad(&pad, m::Reshape(m::GetTupleElement(m::CustomCall(), 0)), in TEST_F()
305 m::Reshape(m::Transpose(m::GetTupleElement(m::CustomCall(), 0))), in TEST_F()
332 GmockMatch(m::Pad(m::Slice(m::GetTupleElement(m::CustomCall( in TEST_F()
349 GmockMatch(m::Pad(&pad, m::GetTupleElement(m::CustomCall(), 0), in TEST_F()
376 GmockMatch(m::Pad(m::Slice(m::GetTupleElement(m::CustomCall( in TEST_F()
579 &slice, m::GetTupleElement(m::CustomCall(), 0)))); in TEST_F()
Dgpu_conv_rewriter_test.cc130 op::GetTupleElement( in TEST_F()
168 op::GetTupleElement(op::CustomCall(kCudnnConvForwardCallTarget), 0)); in TEST_F()
197 op::GetTupleElement( in TEST_F()
227 op::GetTupleElement( in TEST_F()
256 op::GetTupleElement( in TEST_F()
311 op::GetTupleElement( in TEST_F()
360 op::GetTupleElement( in TEST_F()
396 op::GetTupleElement(op::CustomCall(kCudnnConvForwardCallTarget), 0)); in TEST_F()
450 op::GetTupleElement( in TEST_F()
502 op::GetTupleElement(op::CustomCall(kCudnnConvForwardCallTarget), 0)); in TEST_F()
[all …]
Dall_reduce_blueconnect_test.cc163 op::AllReduce(op::GetTupleElement(reduce_scatter, 0), in TEST_F()
164 op::GetTupleElement(reduce_scatter, 1)), in TEST_F()
167 op::AllGather(op::GetTupleElement(all_reduce, 0), in TEST_F()
168 op::GetTupleElement(all_reduce, 1)), in TEST_F()
171 op::Bitcast(op::GetTupleElement(all_gather, 0))); in TEST_F()
173 op::Bitcast(op::GetTupleElement(all_gather, 1))); in TEST_F()
Dmulti_output_fusion_test.cc431 EXPECT_THAT(root, op::Tuple(op::GetTupleElement(), op::GetTupleElement())); in TEST_F()
459 EXPECT_THAT(root, op::Tuple(op::GetTupleElement(), op::GetTupleElement())); in TEST_F()
505 EXPECT_THAT(root, op::Tuple(op::GetTupleElement(), op::GetTupleElement(), in TEST_F()
506 op::GetTupleElement())); in TEST_F()
581 EXPECT_THAT(root, op::Tuple(op::GetTupleElement(), op::GetTupleElement(), in TEST_F()
582 op::GetTupleElement())); in TEST_F()
/external/tensorflow/tensorflow/compiler/tf2xla/kernels/
Dunique_op.cc99 auto counter = xla::GetTupleElement(param, 0); in RollingSelectR1()
110 auto counter = xla::GetTupleElement(param, 0); in RollingSelectR1()
112 auto data_stack = xla::GetTupleElement(param, 1); in RollingSelectR1()
116 auto mask_stack = xla::GetTupleElement(param, 2); in RollingSelectR1()
125 auto accum_stack = xla::GetTupleElement(param, 3); in RollingSelectR1()
142 return xla::GetTupleElement(xla::While(cond, body, init), 3); in RollingSelectR1()
190 perm = xla::GetTupleElement(sorted, sort_keys.size() - 1); in CompileWithAxis()
219 mask = xla::GetTupleElement(sort_by_iperm, 1); in CompileWithAxis()
221 auto perm_sort = xla::GetTupleElement(sort_by_iperm, 2); in CompileWithAxis()
230 auto mask_permute = xla::GetTupleElement(mask_sort, 1); in CompileWithAxis()
[all …]
Dbatch_norm_op.cc105 xla::ConvertElementType(xla::GetTupleElement(output, 0), input_type); in CompileImpl()
114 xla::XlaOp variance = xla::GetTupleElement(output, 2); in CompileImpl()
130 ctx->SetOutput(1, xla::GetTupleElement(output, 1)); in CompileImpl()
139 ctx->SetOutput(1, xla::GetTupleElement(output, 1)); in CompileImpl()
149 xla::Mul(xla::GetTupleElement(output, 1), beta)); in CompileImpl()
165 ctx->SetOutput(3, xla::GetTupleElement(output, 1)); in CompileImpl()
299 x_backprop = xla::GetTupleElement(output, 0); in Compile()
300 scale_backprop = xla::GetTupleElement(output, 1); in Compile()
301 offset_backprop = xla::GetTupleElement(output, 2); in Compile()
Dtensor_list_utils.cc156 *buffer = xla::GetTupleElement(list, 0); in GetTensorListBuffer()
168 *push_index = xla::GetTupleElement(list, tuple_size - 1); in GetTensorListPushIndex()
184 result_parts.push_back(xla::GetTupleElement(list, i)); in SetTensorListPushIndex()
215 auto buffer = xla::GetTupleElement(list, 0); in GetLeadingDimForTensorList()
340 sub_element = xla::GetTupleElement(element, i); in GetInitializedTensorListForElement()
366 xla::XlaOp push_index = xla::GetTupleElement(list, list_tuple_size - 1); in ExecuteTensorListPushBack()
376 xla::XlaOp element_part = xla::GetTupleElement(element, i); in ExecuteTensorListPushBack()
387 xla::XlaOp list_part = xla::GetTupleElement(list, i); in ExecuteTensorListPushBack()
403 xla::XlaOp list_part = xla::GetTupleElement(list, 0); in ExecuteTensorListPushBack()
431 xla::XlaOp push_index = xla::GetTupleElement(list, list_tuple_size - 1); in ExecuteTensorListPopBack()
[all …]
/external/tensorflow/tensorflow/compiler/xla/service/cpu/tests/
Dcpu_infeed_test.cc213 GetTupleElement(prev, 1); in TEST_F()
229 Reduce(GetTupleElement(infeed, 0), ConstantR0<float>(&builder, 0.0f), in TEST_F()
231 auto result = Add(GetTupleElement(prev, 0), addend); in TEST_F()
232 Tuple(&builder, {result, GetTupleElement(infeed, 1)}); in TEST_F()
241 &builder, {GetTupleElement(while1, 0), ConstantR0<bool>(&builder, true)}); in TEST_F()
246 GetTupleElement(while2, 0); in TEST_F()

12345