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

/external/tensorflow/tensorflow/compiler/xla/service/
H A Dtuple_simplifier_test.cc142 HloInstruction* nested_tuple = param; local
144 nested_tuple = builder.AddInstruction(
145 HloInstruction::CreateTuple({nested_tuple, nested_tuple}));
148 HloInstruction* element = nested_tuple;
H A Dhlo_dataflow_analysis_test.cc206 auto nested_tuple = builder.AddInstruction( local
209 HloInstruction::CreateGetTupleElement(tuple->shape(), nested_tuple, 1));
224 HloPosition{nested_tuple, {0, 0}}, HloPosition{nested_tuple, {1, 0}},
225 HloPosition{nested_tuple, {2}}, HloPosition{gte_tuple, {0}},
236 EXPECT_THAT(analysis.GetValueDefinedAt(nested_tuple, /*index=*/{}).uses(),
243 EXPECT_FALSE(analysis.GetValueDefinedAt(nested_tuple, /*index=*/{})
H A Dlayout_assignment_test.cc261 // %nested_tuple = Tuple(%inner_tuple, %inner_tuple)
272 auto nested_tuple = builder.AddInstruction( local
280 Shape result_shape = nested_tuple->shape();
H A Dcopy_insertion_test.cc569 auto nested_tuple = local
571 builder.AddInstruction(HloInstruction::CreateTuple({add0, nested_tuple}));
/external/python/cpython2/Lib/test/
H A 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/tensorflow/tensorflow/compiler/xla/
H A Dliteral_util_test.cc948 auto nested_tuple = Literal::MakeTuple( local
954 // Create a tuple the same shape as the inner tuple of nested_tuple but with
960 EXPECT_EQ(*matrix, LiteralView::Create(*nested_tuple, {0}));
961 EXPECT_EQ(nested_tuple->Get<int32>({}, {1, 0}), 42);
962 EXPECT_EQ(nested_tuple->Get<double>({0}, {1, 1}), 23.0);
963 EXPECT_EQ(nested_tuple->Get<double>({1}, {1, 1}), 44.0);
965 // Overwrite the inner tuple element of nested_tuple with the contents of
967 TF_ASSERT_OK(nested_tuple->CopyFrom(*tuple, /*dest_shape_index=*/{1},
971 EXPECT_EQ(*matrix, LiteralView::Create(*nested_tuple, {0}));
974 EXPECT_EQ(nested_tuple
1302 auto nested_tuple = Literal::MakeTuple({tuple.get(), scalar.get()}); local
1324 auto nested_tuple = Literal::MakeTuple({tuple.get(), scalar.get()}); local
1347 auto nested_tuple = Literal::MakeTuple({tuple.get(), scalar.get()}); local
1371 auto nested_tuple = Literal::MakeTuple( local
1527 auto nested_tuple = Literal::MakeTuple( local
[all...]
H A Dshape_util_test.cc444 Shape nested_tuple = ShapeUtil::MakeTupleShape({tuple, matrix}); local
453 ShapeUtil::HumanString(nested_tuple));
462 ShapeUtil::HumanStringWithLayout(nested_tuple));
465 {opaque, scalar, matrix, matrix2, tuple, nested_tuple}, nested_tuple);
481 prog.add_parameter_names("nested_tuple");
488 "nested_tuple: ((opaque[], f32[], u32[1,2], s32[3,4]), u32[1,2])) -> "
/external/python/cpython3/Lib/test/
H A 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)
H A Dtest_ast.py980 nested_tuple = (1,)
983 nested_tuple = (nested_tuple, 2)
988 nested_tuple, nested_frozenset)

Completed in 232 milliseconds