Home
last modified time | relevance | path

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

123

/external/tensorflow/tensorflow/compiler/xla/service/
Dwhile_util_test.cc83 op::Tuple(op::GetTupleElement(::testing::Eq(new_while_instr), 0), in TEST()
84 op::GetTupleElement(::testing::Eq(new_while_instr), 1))); in TEST()
87 op::Tuple(op::GetTupleElement(op::Parameter(0), 0), in TEST()
88 op::GetTupleElement(op::Parameter(0), 1)); in TEST()
91 op::Tuple(op::GetTupleElement(param_reconstructed, 0), in TEST()
92 op::GetTupleElement(param_reconstructed, 1))); in TEST()
117 op::Tuple(op::GetTupleElement(::testing::Eq(new_while_instr), 0), in TEST()
118 op::GetTupleElement(::testing::Eq(new_while_instr), 1))); in TEST()
121 op::Tuple(op::GetTupleElement(op::Parameter(0), 0), in TEST()
122 op::GetTupleElement(op::Parameter(0), 1)); in TEST()
[all …]
Dwhile_loop_constant_sinking_test.cc107 op::GetTupleElement(op::Parameter(0)), in TEST_F()
108 op::GetTupleElement(op::Parameter(0)))); in TEST_F()
147 op::Tuple(op::GetTupleElement(op::Constant(), 0), in TEST_F()
148 op::GetTupleElement(op::Parameter(0)))); in TEST_F()
196 op::GetTupleElement(op::Parameter(0)), in TEST_F()
197 op::GetTupleElement(op::Parameter(0)))); in TEST_F()
237 op::Tuple(op::GetTupleElement(), op::GetTupleElement(), in TEST_F()
238 op::GetTupleElement())); in TEST_F()
323 op::Lt(_, op::GetTupleElement(op::Constant()))); in TEST_F()
Dtuple_util_test.cc59 EXPECT_THAT(prefix, op::Tuple(op::GetTupleElement(op::Parameter(0), 0), in TEST()
60 op::GetTupleElement(op::Parameter(0), 1))); in TEST()
74 EXPECT_THAT(with_suffix, op::Tuple(op::GetTupleElement(op::Parameter(0), 0), in TEST()
75 op::GetTupleElement(op::Parameter(0), 1), in TEST()
76 op::GetTupleElement(op::Parameter(0), 2), in TEST()
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()
Dhlo_element_type_converter_test.cc114 op::Tuple(op::Convert(op::GetTupleElement(batch_norm, 0)), in TEST_F()
115 op::Convert(op::GetTupleElement(batch_norm, 1)), in TEST_F()
116 op::Convert(op::GetTupleElement(batch_norm, 2)))); in TEST_F()
Dwhile_loop_simplifier_test.cc425 op::Add(op::GetTupleElement(op::Parameter(0), /*tuple_index=*/0), in TEST_F()
427 op::GetTupleElement(op::Parameter(0), /*tuple_index=*/1))); in TEST_F()
431 op::GetTupleElement(op::Parameter(0), /*tuple_index=*/1))); in TEST_F()
724 op::Tuple(op::GetTupleElement(op::Parameter(), 0), in TEST_F()
725 op::GetTupleElement(op::Parameter(), 1), op::Add(), in TEST_F()
726 op::Add(op::GetTupleElement(op::Parameter(), 3), in TEST_F()
Dhlo_computation_test.cc289 m::Copy(m::GetTupleElement(m::Op().Is(tuple))), in TEST_F()
290 m::Copy(m::GetTupleElement(m::Op().Is(tuple)))))); in TEST_F()
342 m::Copy(m::GetTupleElement(m::Op().Is(tuple))) in TEST_F()
344 m::Copy(m::GetTupleElement(m::Op().Is(tuple))) in TEST_F()
359 GmockMatch(m::Tuple(m::GetTupleElement(m::Op().Is(tuple)), in TEST_F()
360 m::GetTupleElement(m::Op().Is(tuple))))); in TEST_F()
377 m::Copy(m::GetTupleElement(m::Op().Is(tuple))), in TEST_F()
378 m::GetTupleElement(m::Op().Is(tuple))))); in TEST_F()
414 m::GetTupleElement(m::Op().Is(tuple)), in TEST_F()
415 m::Copy(m::GetTupleElement(m::Op().Is(tuple)))))); in TEST_F()
Dsort_simplifier_test.cc99 m::GetTupleElement(m::Sort(m::Parameter(0), m::Parameter(2)), 0), in TEST_F()
100 m::GetTupleElement(m::Sort(m::Parameter(0), m::Parameter(2)), 1)))); in TEST_F()
/external/tensorflow/tensorflow/compiler/xla/tests/
Dwhile_test.cc369 auto iteration = GetTupleElement(prev, 0); in XLA_TEST_F()
380 auto iteration = GetTupleElement(prev, 0); in XLA_TEST_F()
381 auto w1 = GetTupleElement(prev, 1); in XLA_TEST_F()
382 auto w2 = GetTupleElement(prev, 2); in XLA_TEST_F()
383 auto w3 = GetTupleElement(prev, 3); in XLA_TEST_F()
423 auto iteration = GetTupleElement(prev, 0); in XLA_TEST_F()
434 auto iteration = GetTupleElement(prev, 0); in XLA_TEST_F()
435 auto w1 = GetTupleElement(prev, 1); in XLA_TEST_F()
436 auto w2 = GetTupleElement(prev, 2); in XLA_TEST_F()
437 auto w3 = GetTupleElement(prev, 3); in XLA_TEST_F()
[all …]
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()
256 auto new_tuple01 = Tuple(&builder, {GetTupleElement(tuple_data, 0), in XLA_TEST_F()
257 GetTupleElement(tuple_data, 1)}); in XLA_TEST_F()
258 auto new_tuple10 = Tuple(&builder, {GetTupleElement(tuple_data, 1), in XLA_TEST_F()
[all …]
Dconditional_test.cc83 auto x = GetTupleElement(tuple, 0); in CreateTupleCeilComputation()
84 auto y = GetTupleElement(tuple, 1); in CreateTupleCeilComputation()
105 auto x = GetTupleElement(tuple, 0); in CreateTupleFloorComputation()
106 auto y = GetTupleElement(tuple, 1); in CreateTupleFloorComputation()
127 auto x = GetTupleElement(tuple, 0); in CreateTupleAddComputation()
128 auto y = GetTupleElement(tuple, 1); in CreateTupleAddComputation()
147 auto x = GetTupleElement(tuple, 0); in CreateTupleSubComputation()
148 auto y = GetTupleElement(tuple, 1); in CreateTupleSubComputation()
425 GetTupleElement(p, 0)), in XLA_TEST_P()
426 GetTupleElement(p, 1)); in XLA_TEST_P()
[all …]
Doutfeed_in_nested_computation_test.cc45 XlaOp loop_counter = GetTupleElement(state_tuple, 0); in XLA_TEST_F()
54 XlaOp loop_counter = GetTupleElement(state_tuple, 0); in XLA_TEST_F()
55 XlaOp buffer_inside = GetTupleElement(state_tuple, 1); in XLA_TEST_F()
70 GetTupleElement(result_tuple, 0); in XLA_TEST_F()
Dlocal_client_execute_test.cc292 auto x_0 = GetTupleElement(x, 0); in XLA_TEST_F()
293 auto x_1 = GetTupleElement(x, 1); in XLA_TEST_F()
294 auto y_0 = GetTupleElement(y, 0); in XLA_TEST_F()
295 auto y_1 = GetTupleElement(y, 1); in XLA_TEST_F()
337 auto inner_tuple = GetTupleElement(param, 0); in XLA_TEST_F()
338 auto inner_array = GetTupleElement(inner_tuple, 0); in XLA_TEST_F()
339 auto inner_vector = GetTupleElement(inner_tuple, 1); in XLA_TEST_F()
340 auto outer_vector = GetTupleElement(param, 1); in XLA_TEST_F()
374 auto element_0 = GetTupleElement(param, 0); in XLA_TEST_F()
375 auto element_1 = GetTupleElement(param, 1); in XLA_TEST_F()
[all …]
/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()
53 xla::GetTupleElement(xla::TopK(x_rev, 3), 1); in XLA_TEST_F()
62 xla::GetTupleElement(xla::TopK(x_rev, 3), 1); in XLA_TEST_F()
78 xla::GetTupleElement(xla::TopK(x, kSize), 0); in XLA_TEST_F()
88 xla::GetTupleElement(xla::TopK(a, 5), 1); in XLA_TEST_F()
Dsorting.cc46 XlaOp values = Neg(Slice(GetTupleElement(sort_result, 0), start_indices, in TopK()
48 XlaOp indices = Slice(GetTupleElement(sort_result, 1), start_indices, in TopK()
/external/tensorflow/tensorflow/compiler/tf2xla/kernels/
Dbatch_norm_op.cc67 ctx->SetOutput(0, xla::ConvertElementType(xla::GetTupleElement(output, 0), in Compile()
69 ctx->SetOutput(1, xla::GetTupleElement(output, 1)); in Compile()
70 ctx->SetOutput(2, xla::GetTupleElement(output, 2)); in Compile()
76 ctx->SetOutput(3, xla::GetTupleElement(output, 1)); in Compile()
83 xla::XlaOp variance = xla::GetTupleElement(output, 2); in Compile()
87 ctx->SetOutput(4, xla::GetTupleElement(output, 2)); in Compile()
169 x_backprop = xla::GetTupleElement(output, 0); in Compile()
170 scale_backprop = xla::GetTupleElement(output, 1); in Compile()
171 offset_backprop = xla::GetTupleElement(output, 2); in Compile()
Dstack_ops.cc143 xla::XlaOp ta = xla::GetTupleElement(resource->value(), 0); in Compile()
144 xla::XlaOp index = xla::GetTupleElement(resource->value(), 1); in Compile()
197 xla::XlaOp ta = xla::GetTupleElement(state, 0); in Compile()
198 xla::XlaOp index = xla::GetTupleElement(state, 1); in Compile()
Dsort_ops.cc42 context->SetOutput(0, xla::GetTupleElement(result, 0)); in Compile()
43 context->SetOutput(1, xla::GetTupleElement(result, 1)); in Compile()
Dwhile_op.cc186 xla::GetTupleElement(outputs, 0); in BuildWrappedCond()
210 xla::GetTupleElement(outputs, i)); in BuildWrappedBody()
217 xla::GetTupleElement(outputs, ctx->num_outputs())); in BuildWrappedBody()
238 xla::GetTupleElement(while_result, while_result_index); in BuildWhile()
246 padded_while_outputs.push_back(xla::GetTupleElement( in BuildWhile()
533 ctx->SetOutput(i, xla::GetTupleElement(while_result, i)); in Compile()
542 xla::GetTupleElement(while_result, ctx->num_outputs()); in Compile()
562 xla::GetTupleElement(while_result, pos), builder)); in Compile()
Dtopk_op.cc52 context->SetOutput(0, xla::GetTupleElement(output_tuple, 0)); in Compile()
53 context->SetOutput(1, xla::GetTupleElement(output_tuple, 1)); in Compile()
/external/tensorflow/tensorflow/compiler/xla/service/gpu/
Dcudnn_conv_pad_for_tensor_cores_test.cc96 EXPECT_THAT(root, op::Tuple(op::Slice(op::GetTupleElement(op::CustomCall( in TEST_F()
117 EXPECT_THAT(root, op::GetTupleElement(op::Tuple( in TEST_F()
118 op::Slice(op::GetTupleElement(op::CustomCall( in TEST_F()
139 EXPECT_THAT(root, op::GetTupleElement(op::Tuple( in TEST_F()
140 op::Slice(op::GetTupleElement(op::CustomCall( in TEST_F()
161 EXPECT_THAT(root, op::GetTupleElement(op::Tuple( in TEST_F()
162 op::Slice(op::GetTupleElement(op::CustomCall( in TEST_F()
Dmulti_output_fusion_test.cc400 EXPECT_THAT(root, op::Tuple(op::GetTupleElement(), op::GetTupleElement())); in TEST_F()
427 EXPECT_THAT(root, op::Tuple(op::GetTupleElement(), op::GetTupleElement())); in TEST_F()
467 EXPECT_THAT(root, op::Tuple(op::GetTupleElement(), op::GetTupleElement(), in TEST_F()
468 op::GetTupleElement())); in TEST_F()
534 EXPECT_THAT(root, op::Tuple(op::GetTupleElement(), op::GetTupleElement(), in TEST_F()
535 op::GetTupleElement())); in TEST_F()
Dcudnn_conv_rewriter_test.cc128 op::GetTupleElement( in TEST_F()
164 op::GetTupleElement( in TEST_F()
194 op::GetTupleElement( in TEST_F()
224 op::GetTupleElement( in TEST_F()
253 op::GetTupleElement( in TEST_F()
307 op::GetTupleElement( in TEST_F()
354 op::GetTupleElement( in TEST_F()
389 op::GetTupleElement(op::CustomCall(kCudnnConvForwardCallTarget), 0)); in TEST_F()
442 op::GetTupleElement( in TEST_F()
493 op::GetTupleElement(op::CustomCall(kCudnnConvForwardCallTarget), 0)); in TEST_F()
[all …]
/external/tensorflow/tensorflow/compiler/xla/client/
Dxla_builder_test.cc513 auto gte = GetTupleElement(p0, 0); in TEST_F()
537 auto gte0 = GetTupleElement(p0, 0); in TEST_F()
538 auto gte1 = GetTupleElement(p0, 1); in TEST_F()
562 auto gte0 = GetTupleElement(p0, 0); in TEST_F()
563 auto gte1 = GetTupleElement(p0, 1); in TEST_F()
583 auto gte = GetTupleElement(p0, 0); in TEST_F()
605 auto gte0 = GetTupleElement(p0, 0); in TEST_F()
606 auto gte1 = GetTupleElement(p0, 1); in TEST_F()
626 auto gte0 = GetTupleElement(p0, 0); in TEST_F()
627 auto gte1 = GetTupleElement(p0, 1); in TEST_F()
[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()

123