/external/tensorflow/tensorflow/compiler/xla/service/ |
D | while_util_test.cc | 88 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 …]
|
D | conditional_code_motion_test.cc | 78 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 …]
|
D | while_loop_constant_sinking_test.cc | 106 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()
|
D | tuple_util_test.cc | 67 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()
|
D | topk_rewriter_test.cc | 86 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()
|
D | stable_sort_expander_test.cc | 108 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()
|
D | all_reduce_combiner_test.cc | 420 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()
|
D | hlo_rematerialization_test.cc | 880 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/ |
D | sorting_test.cc | 35 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 …]
|
D | sorting.cc | 68 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()
|
D | lu_decomposition.cc | 52 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/ |
D | while_test.cc | 369 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 …]
|
D | tuple_test.cc | 119 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 …]
|
D | conditional_test.cc | 83 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 …]
|
D | outfeed_in_nested_computation_test.cc | 45 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()
|
D | local_client_execute_test.cc | 295 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/ |
D | unique_op.cc | 75 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 …]
|
D | batch_norm_op.cc | 101 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()
|
D | tensor_list_utils.cc | 156 *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 …]
|
D | stack_ops.cc | 143 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/ |
D | cudnn_pad_for_convolutions_test.cc | 96 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()
|
D | gpu_conv_rewriter_test.cc | 129 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 …]
|
D | multi_output_fusion_test.cc | 431 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/ |
D | cpu_infeed_test.cc | 213 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/ |
D | xla_builder_test.cc | 611 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 …]
|