Searched refs:nested_tuple_shape_ (Results 1 – 2 of 2) sorted by relevance
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/ |
| H A D | shape_tree_test.cc | 34 nested_tuple_shape_ = ShapeUtil::MakeTupleShape( in ShapeTreeTest()
51 Shape nested_tuple_shape_; member in xla::__anon98d2a2fc0111::ShapeTreeTest
85 TestShapeConstructor(nested_tuple_shape_, 10); in TEST_F()
118 TestInitValueConstructor(nested_tuple_shape_, 10); in TEST_F()
203 ShapeTree<int> shape_tree{nested_tuple_shape_}; in TEST_F()
211 EXPECT_TRUE(ShapeUtil::Compatible(nested_tuple_shape_, shape_tree.shape())); in TEST_F()
245 ShapeTree<int> shape_tree{nested_tuple_shape_}; in TEST_F()
318 ShapeTree<int> destination(nested_tuple_shape_, 0); in TEST_F()
337 ShapeTree<int> source(nested_tuple_shape_, 42); in TEST_F()
389 ShapeTree<int> t(&nested_tuple_shape_, 42); in TEST_F()
[all …]
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/ |
| H A D | copy_insertion_test.cc | 496 nested_tuple_shape_, loop_state, 1)); in BuildIndependentBodyComputation()
544 nested_tuple_shape_, loop_state, 1)); in BuildNestedBodyComputation()
688 Shape nested_tuple_shape_ = member in xla::__anon2db0cb110111::WhileCopyInsertionTest
691 {induction_variable_shape_, nested_tuple_shape_});
|