Home
last modified time | relevance | path

Searched refs:nested_tuple (Results 1 – 9 of 9) sorted by relevance

/external/tensorflow/tensorflow/compiler/xla/
Dliteral_test.cc1162 Literal nested_tuple = LiteralUtil::MakeTuple({&matrix, &inner_tuple}); in TEST_F() local
1170 EXPECT_EQ(matrix, LiteralSlice(nested_tuple, {0})); in TEST_F()
1171 EXPECT_EQ(nested_tuple.Get<int32>({}, {1, 0}), 42); in TEST_F()
1172 EXPECT_EQ(nested_tuple.Get<double>({0}, {1, 1}), 23.0); in TEST_F()
1173 EXPECT_EQ(nested_tuple.Get<double>({1}, {1, 1}), 44.0); in TEST_F()
1177 TF_ASSERT_OK(nested_tuple.CopyFrom(tuple, /*dest_shape_index=*/{1}, in TEST_F()
1181 EXPECT_EQ(matrix, LiteralSlice(nested_tuple, {0})); in TEST_F()
1184 EXPECT_EQ(nested_tuple.Get<int32>({}, {1, 0}), -5); in TEST_F()
1185 EXPECT_EQ(nested_tuple.Get<double>({0}, {1, 1}), 2.0); in TEST_F()
1186 EXPECT_EQ(nested_tuple.Get<double>({1}, {1, 1}), 4.0); in TEST_F()
[all …]
/external/tensorflow/tensorflow/compiler/xla/service/
Dtuple_simplifier_test.cc148 HloInstruction* nested_tuple = param; in TEST_F() local
150 nested_tuple = builder.AddInstruction( in TEST_F()
151 HloInstruction::CreateTuple({nested_tuple, nested_tuple})); in TEST_F()
154 HloInstruction* element = nested_tuple; in TEST_F()
Dhlo_dataflow_analysis_test.cc212 auto nested_tuple = builder.AddInstruction( in TEST_P() local
215 HloInstruction::CreateGetTupleElement(tuple->shape(), nested_tuple, 1)); in TEST_P()
231 HloPosition{nested_tuple, {0, 0}}, HloPosition{nested_tuple, {1, 0}}, in TEST_P()
232 HloPosition{nested_tuple, {2}}, HloPosition{gte_tuple, {0}}, in TEST_P()
243 EXPECT_THAT(analysis.GetValueDefinedAt(nested_tuple, /*index=*/{}).uses(), in TEST_P()
250 EXPECT_FALSE(analysis.GetValueDefinedAt(nested_tuple, /*index=*/{}) in TEST_P()
Ddynamic_padder_test.cc284 HloInstruction* nested_tuple = root->mutable_operand(1); in TEST_F() local
285 EXPECT_THAT(nested_tuple, in TEST_F()
Dlayout_assignment_test.cc301 auto nested_tuple = builder.AddInstruction( in TEST_F() local
309 Shape result_shape = nested_tuple->shape(); in TEST_F()
Dcopy_insertion_test.cc582 auto nested_tuple = in BuildIndependentBodyComputation() local
584 builder.AddInstruction(HloInstruction::CreateTuple({add0, nested_tuple})); in BuildIndependentBodyComputation()
/external/python/cpython2/Lib/test/
Dtest_pprint.py458 nested_tuple = (1, (2, (3, (4, (5, 6)))))
461 self.assertEqual(pprint.pformat(nested_tuple), repr(nested_tuple))
468 self.assertEqual(pprint.pformat(nested_tuple, depth=1), lv1_tuple)
/external/python/cpython3/Lib/test/
Dtest_pprint.py827 nested_tuple = (1, (2, (3, (4, (5, 6)))))
830 self.assertEqual(pprint.pformat(nested_tuple), repr(nested_tuple))
837 self.assertEqual(pprint.pformat(nested_tuple, depth=1), lv1_tuple)
Dtest_ast.py1662 nested_tuple = (1,)
1665 nested_tuple = (nested_tuple, 2)
1670 nested_tuple, nested_frozenset)