Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/compiler/xla/
Dliteral_test.cc1084 Literal nested_tuple = LiteralUtil::MakeTuple({&matrix, &inner_tuple}); in TEST_F() local
1092 EXPECT_EQ(matrix, LiteralSlice(nested_tuple, {0})); in TEST_F()
1093 EXPECT_EQ(nested_tuple.Get<int32>({}, {1, 0}), 42); in TEST_F()
1094 EXPECT_EQ(nested_tuple.Get<double>({0}, {1, 1}), 23.0); in TEST_F()
1095 EXPECT_EQ(nested_tuple.Get<double>({1}, {1, 1}), 44.0); in TEST_F()
1099 TF_ASSERT_OK(nested_tuple.CopyFrom(tuple, /*dest_shape_index=*/{1}, in TEST_F()
1103 EXPECT_EQ(matrix, LiteralSlice(nested_tuple, {0})); in TEST_F()
1106 EXPECT_EQ(nested_tuple.Get<int32>({}, {1, 0}), -5); in TEST_F()
1107 EXPECT_EQ(nested_tuple.Get<double>({0}, {1, 1}), 2.0); in TEST_F()
1108 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.cc209 auto nested_tuple = builder.AddInstruction( in TEST_P() local
212 HloInstruction::CreateGetTupleElement(tuple->shape(), nested_tuple, 1)); in TEST_P()
228 HloPosition{nested_tuple, {0, 0}}, HloPosition{nested_tuple, {1, 0}}, in TEST_P()
229 HloPosition{nested_tuple, {2}}, HloPosition{gte_tuple, {0}}, in TEST_P()
240 EXPECT_THAT(analysis.GetValueDefinedAt(nested_tuple, /*index=*/{}).uses(), in TEST_P()
247 EXPECT_FALSE(analysis.GetValueDefinedAt(nested_tuple, /*index=*/{}) in TEST_P()
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.cc576 auto nested_tuple = in BuildIndependentBodyComputation() local
578 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.py607 nested_tuple = (1, (2, (3, (4, (5, 6)))))
610 self.assertEqual(pprint.pformat(nested_tuple), repr(nested_tuple))
617 self.assertEqual(pprint.pformat(nested_tuple, depth=1), lv1_tuple)
Dtest_ast.py1040 nested_tuple = (1,)
1043 nested_tuple = (nested_tuple, 2)
1048 nested_tuple, nested_frozenset)