Searched refs:nested_tuple_shape_ (Results 1 – 2 of 2) sorted by relevance
| /external/tensorflow/tensorflow/compiler/xla/ |
| D | shape_tree_test.cc | 33 nested_tuple_shape_ = ShapeUtil::MakeTupleShape( in ShapeTreeTest()
50 Shape nested_tuple_shape_; member in xla::__anon3818dd720111::ShapeTreeTest
84 TestShapeConstructor(nested_tuple_shape_, 10); in TEST_F()
117 TestInitValueConstructor(nested_tuple_shape_, 10); in TEST_F()
195 ShapeTree<int> shape_tree{nested_tuple_shape_}; in TEST_F()
203 EXPECT_TRUE(ShapeUtil::Compatible(nested_tuple_shape_, shape_tree.shape())); in TEST_F()
237 ShapeTree<int> shape_tree{nested_tuple_shape_}; in TEST_F()
310 ShapeTree<int> destination(nested_tuple_shape_, 0); in TEST_F()
329 ShapeTree<int> source(nested_tuple_shape_, 42); in TEST_F()
381 ShapeTree<int> t(&nested_tuple_shape_, 42); in TEST_F()
[all …]
|
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | copy_insertion_test.cc | 567 nested_tuple_shape_, loop_state, 1)); in BuildIndependentBodyComputation()
615 nested_tuple_shape_, loop_state, 1)); in BuildNestedBodyComputation()
705 nested_tuple_shape_, HloOpcode::kTupleSelect, pred, tuple1, tuple2)); in BuildWhileInstruction_InitPointsToAmbiguous()
783 Shape nested_tuple_shape_ = member in xla::__anon03ec66870111::WhileCopyInsertionTest
786 {induction_variable_shape_, nested_tuple_shape_});
|