Home
last modified time | relevance | path

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

/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/
H A Dliteral_test.cc1160 Literal nested_tuple = LiteralUtil::MakeTuple({&matrix, &inner_tuple}); in TEST_F() local
1168 EXPECT_EQ(matrix, LiteralSlice(nested_tuple, {0})); in TEST_F()
1169 EXPECT_EQ(nested_tuple.Get<int32_t>({}, {1, 0}), 42); in TEST_F()
1170 EXPECT_EQ(nested_tuple.Get<double>({0}, {1, 1}), 23.0); in TEST_F()
1171 EXPECT_EQ(nested_tuple.Get<double>({1}, {1, 1}), 44.0); in TEST_F()
1175 TF_ASSERT_OK(nested_tuple.CopyFrom(tuple, /*dest_shape_index=*/{1}, in TEST_F()
1179 EXPECT_EQ(matrix, LiteralSlice(nested_tuple, {0})); in TEST_F()
1182 EXPECT_EQ(nested_tuple.Get<int32_t>({}, {1, 0}), -5); in TEST_F()
1183 EXPECT_EQ(nested_tuple.Get<double>({0}, {1, 1}), 2.0); in TEST_F()
1184 EXPECT_EQ(nested_tuple.Get<double>({1}, {1, 1}), 4.0); in TEST_F()
[all …]
/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/
H A Dtuple_simplifier_test.cc146 HloInstruction* nested_tuple = param; in TEST_F() local
148 nested_tuple = builder.AddInstruction( in TEST_F()
149 HloInstruction::CreateTuple({nested_tuple, nested_tuple})); in TEST_F()
152 HloInstruction* element = nested_tuple; in TEST_F()
H A Dhlo_dataflow_analysis_test.cc219 auto nested_tuple = builder.AddInstruction( in TEST_P() local
222 HloInstruction::CreateGetTupleElement(tuple->shape(), nested_tuple, 1)); in TEST_P()
238 HloPosition{nested_tuple, {0, 0}}, HloPosition{nested_tuple, {1, 0}}, in TEST_P()
239 HloPosition{nested_tuple, {2}}, HloPosition{gte_tuple, {0}}, in TEST_P()
250 EXPECT_THAT(analysis.GetValueDefinedAt(nested_tuple, /*index=*/{}).GetUses(), in TEST_P()
257 EXPECT_FALSE(analysis.GetValueDefinedAt(nested_tuple, /*index=*/{}) in TEST_P()
H A Ddynamic_padder_test.cc325 HloInstruction* nested_tuple = root->mutable_operand(1); in TEST_F() local
326 EXPECT_THAT(nested_tuple, in TEST_F()
H A Dlayout_assignment_test.cc267 auto nested_tuple = builder.AddInstruction( in TEST_F() local
275 Shape result_shape = nested_tuple->shape(); in TEST_F()
H A Dcopy_insertion_test.cc511 auto nested_tuple = in BuildIndependentBodyComputation() local
513 builder.AddInstruction(HloInstruction::CreateTuple({add0, nested_tuple})); in BuildIndependentBodyComputation()
/aosp_15_r20/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)
/aosp_15_r20/external/python/cpython3/Lib/test/
Dtest_pprint.py835 nested_tuple = (1, (2, (3, (4, (5, 6)))))
838 self.assertEqual(pprint.pformat(nested_tuple), repr(nested_tuple))
845 self.assertEqual(pprint.pformat(nested_tuple, depth=1), lv1_tuple)
Dtest_ast.py1871 nested_tuple = (1,)
1874 nested_tuple = (nested_tuple, 2)
1879 nested_tuple, nested_frozenset)
/aosp_15_r20/external/rust/android-crates-io/crates/axum-core/src/extract/
Dtuple.rs115 fn nested_tuple() { in nested_tuple() function