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_});
|