Home
last modified time | relevance | path

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

1234

/external/tensorflow/tensorflow/compiler/xla/service/
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 …]
Dconditional_code_motion_test.cc78 EXPECT_THAT(root, AllOf(op::Tuple(op::Convert(), op::GetTupleElement()))); in TEST_F()
292 op::Convert(op::Reshape(op::GetTupleElement(op::Conditional()))), in TEST_F()
293 op::Convert(op::Reshape(op::GetTupleElement(op::Conditional()))))))); in TEST_F()
358 AllOf(op::Add(op::Multiply(op::GetTupleElement(op::Conditional()), in TEST_F()
360 op::GetTupleElement(op::Conditional())))); in TEST_F()
416 op::Add(op::GetTupleElement(op::Conditional()), op::Constant()), in TEST_F()
417 op::Add(op::GetTupleElement(op::Conditional()), op::Constant())), in TEST_F()
419 op::Add(op::GetTupleElement(op::Conditional()), op::Constant()), in TEST_F()
420 op::Add(op::GetTupleElement(op::Conditional()), in TEST_F()
621 op::Convert(op::AllReduce(op::GetTupleElement(op::Conditional()))), in TEST_F()
[all …]
Dwhile_loop_constant_sinking_test.cc106 op::GetTupleElement(op::Parameter(0)), in TEST_F()
107 op::GetTupleElement(op::Parameter(0)))); in TEST_F()
146 op::Tuple(op::GetTupleElement(op::Constant(), 0), in TEST_F()
147 op::GetTupleElement(op::Parameter(0)))); in TEST_F()
195 op::GetTupleElement(op::Parameter(0)), in TEST_F()
196 op::GetTupleElement(op::Parameter(0)))); in TEST_F()
236 op::Tuple(op::GetTupleElement(), op::GetTupleElement(), in TEST_F()
237 op::GetTupleElement())); in TEST_F()
322 op::Lt(_, op::GetTupleElement(op::Constant()))); in TEST_F()
Dtuple_util_test.cc67 EXPECT_THAT(prefix, op::Tuple(op::GetTupleElement(op::Parameter(0), 0), in TEST()
68 op::GetTupleElement(op::Parameter(0), 1))); in TEST()
81 EXPECT_THAT(with_suffix, op::Tuple(op::GetTupleElement(op::Parameter(0), 0), in TEST()
82 op::GetTupleElement(op::Parameter(0), 1), in TEST()
83 op::GetTupleElement(op::Parameter(0), 2), in TEST()
Dtopk_rewriter_test.cc86 op::Tuple(op::GetTupleElement(op::CustomCall(op::Parameter(0)), 0), in TEST_F()
87 op::GetTupleElement(op::CustomCall(op::Parameter(0)), 1))); in TEST_F()
116 op::Tuple(op::GetTupleElement(op::CustomCall(op::Parameter(0)), 0), in TEST_F()
117 op::GetTupleElement(op::CustomCall(op::Parameter(0)), 1))); in TEST_F()
147 op::Tuple(op::Transpose(op::GetTupleElement( in TEST_F()
149 op::Transpose(op::GetTupleElement( 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_reduce_combiner_test.cc420 op::Add(op::Domain(op::GetTupleElement( in TEST_F()
424 op::GetTupleElement( in TEST_F()
470 auto add = op::Add(op::AllReduce(op::GetTupleElement(crs0, 0)), in TEST_F()
471 op::GetTupleElement(crs0, 1)); in TEST_F()
472 auto crs1 = op::AllReduce(add, op::GetTupleElement(crs0)); in TEST_F()
475 op::Tuple(op::GetTupleElement(crs1, 0), op::GetTupleElement(crs1, 1))); in TEST_F()
Dhlo_rematerialization_test.cc880 ASSERT_THAT(add, op::Add(op::Multiply(), op::GetTupleElement(op::Fusion()))); in TEST_F()
887 add, op::Add(op::Multiply(), op::GetTupleElement(AllOf( in TEST_F()
926 op::GetTupleElement(op::Fusion())), in TEST_F()
927 op::GetTupleElement(op::Fusion()))); in TEST_F()
936 op::Add(op::Multiply(), op::GetTupleElement(AllOf( in TEST_F()
938 op::GetTupleElement(AllOf(op::Fusion(), ::testing::Ne(fusion))))); in TEST_F()
980 op::Bitcast(op::GetTupleElement(op::Fusion())))); in TEST_F()
987 op::Bitcast(op::GetTupleElement( in TEST_F()
1027 ASSERT_THAT(add, op::Add(op::GetTupleElement( in TEST_F()
1028 op::Tuple(op::GetTupleElement(op::Fusion()), _)), in 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()
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 …]
Dsorting.cc68 XlaOp values = Slice(GetTupleElement(sort_result, 0), start_indices, in TopK()
70 XlaOp indices = Slice(GetTupleElement(sort_result, 1), start_indices, in TopK()
132 values = Slice(GetTupleElement(sort_result, 0), start_indices, in TopKWithPartitions()
134 indices = Slice(GetTupleElement(sort_result, 1), start_indices, in TopKWithPartitions()
160 XlaOp values = Slice(GetTupleElement(sort_result, 0), start_indices, in TopKWithPartitions()
162 XlaOp indices = Slice(GetTupleElement(sort_result, 1), start_indices, in TopKWithPartitions()
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/
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.cc295 auto x_0 = GetTupleElement(x, 0); in XLA_TEST_F()
296 auto x_1 = GetTupleElement(x, 1); in XLA_TEST_F()
297 auto y_0 = GetTupleElement(y, 0); in XLA_TEST_F()
298 auto y_1 = GetTupleElement(y, 1); in XLA_TEST_F()
340 auto inner_tuple = GetTupleElement(param, 0); in XLA_TEST_F()
341 auto inner_array = GetTupleElement(inner_tuple, 0); in XLA_TEST_F()
342 auto inner_vector = GetTupleElement(inner_tuple, 1); in XLA_TEST_F()
343 auto outer_vector = GetTupleElement(param, 1); in XLA_TEST_F()
377 auto element_0 = GetTupleElement(param, 0); in XLA_TEST_F()
378 auto element_1 = GetTupleElement(param, 1); in XLA_TEST_F()
[all …]
/external/tensorflow/tensorflow/compiler/tf2xla/kernels/
Dunique_op.cc75 auto j = xla::GetTupleElement(param, 2); in BuildInnerLoopCond()
76 auto output_element_size = xla::GetTupleElement(param, 3); in BuildInnerLoopCond()
87 auto input = xla::GetTupleElement(param, 0); in BuildInnerLoopBody()
88 auto target = xla::GetTupleElement(param, 1); in BuildInnerLoopBody()
89 auto j = xla::GetTupleElement(param, 2); in BuildInnerLoopBody()
90 auto output_element_size = xla::GetTupleElement(param, 3); in BuildInnerLoopBody()
91 auto output_index = xla::GetTupleElement(param, 4); in BuildInnerLoopBody()
109 auto i = xla::GetTupleElement(param, 2); in BuildOuterLoopCond()
122 auto input = xla::GetTupleElement(param, 0); in BuildOuterLoopBody()
123 auto indices = xla::GetTupleElement(param, 1); in BuildOuterLoopBody()
[all …]
Dbatch_norm_op.cc101 xla::ConvertElementType(xla::GetTupleElement(output, 0), input_type); in CompileImpl()
110 xla::XlaOp variance = xla::GetTupleElement(output, 2); in CompileImpl()
125 ctx->SetOutput(1, xla::GetTupleElement(output, 1)); in CompileImpl()
135 ctx->SetOutput(1, xla::GetTupleElement(output, 1)); in CompileImpl()
145 xla::Mul(xla::GetTupleElement(output, 1), beta)); in CompileImpl()
161 ctx->SetOutput(3, xla::GetTupleElement(output, 1)); in CompileImpl()
295 x_backprop = xla::GetTupleElement(output, 0); in Compile()
296 scale_backprop = xla::GetTupleElement(output, 1); in Compile()
297 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()
214 auto buffer = xla::GetTupleElement(list, 0); in GetLeadingDimForTensorList()
338 sub_element = xla::GetTupleElement(element, i); in GetInitializedTensorListForElement()
364 xla::XlaOp push_index = xla::GetTupleElement(list, list_tuple_size - 1); in ExecuteTensorListPushBack()
374 xla::XlaOp element_part = xla::GetTupleElement(element, i); in ExecuteTensorListPushBack()
385 xla::XlaOp list_part = xla::GetTupleElement(list, i); in ExecuteTensorListPushBack()
401 xla::XlaOp list_part = xla::GetTupleElement(list, 0); in ExecuteTensorListPushBack()
429 xla::XlaOp push_index = xla::GetTupleElement(list, list_tuple_size - 1); in ExecuteTensorListPopBack()
[all …]
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()
/external/tensorflow/tensorflow/compiler/xla/service/gpu/
Dcudnn_pad_for_convolutions_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()
233 EXPECT_THAT(root, op::Tuple(op::Slice(op::GetTupleElement(op::CustomCall( in TEST_F()
Dgpu_conv_rewriter_test.cc129 op::GetTupleElement( in TEST_F()
167 op::GetTupleElement(op::CustomCall(kCudnnConvForwardCallTarget), 0)); in TEST_F()
196 op::GetTupleElement( in TEST_F()
226 op::GetTupleElement( in TEST_F()
255 op::GetTupleElement( in TEST_F()
310 op::GetTupleElement( in TEST_F()
359 op::GetTupleElement( in TEST_F()
395 op::GetTupleElement(op::CustomCall(kCudnnConvForwardCallTarget), 0)); in TEST_F()
449 op::GetTupleElement( in TEST_F()
501 op::GetTupleElement(op::CustomCall(kCudnnConvForwardCallTarget), 0)); in TEST_F()
[all …]
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/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()
/external/tensorflow/tensorflow/compiler/xla/client/
Dxla_builder_test.cc611 auto gte = GetTupleElement(p0, 0); in TEST_F()
635 auto gte0 = GetTupleElement(p0, 0); in TEST_F()
636 auto gte1 = GetTupleElement(p0, 1); in TEST_F()
660 auto gte0 = GetTupleElement(p0, 0); in TEST_F()
661 auto gte1 = GetTupleElement(p0, 1); in TEST_F()
681 auto gte = GetTupleElement(p0, 0); in TEST_F()
703 auto gte0 = GetTupleElement(p0, 0); in TEST_F()
704 auto gte1 = GetTupleElement(p0, 1); in TEST_F()
724 auto gte0 = GetTupleElement(p0, 0); in TEST_F()
725 auto gte1 = GetTupleElement(p0, 1); in TEST_F()
[all …]

1234