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::__anon8e675d460111::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 | 561 nested_tuple_shape_, loop_state, 1)); in BuildIndependentBodyComputation()
609 nested_tuple_shape_, loop_state, 1)); in BuildNestedBodyComputation()
699 nested_tuple_shape_, HloOpcode::kTupleSelect, pred, tuple1, tuple2)); in BuildWhileInstruction_InitPointsToAmbiguous()
777 Shape nested_tuple_shape_ = member in xla::__anon1fa5445b0111::WhileCopyInsertionTest
780 {induction_variable_shape_, nested_tuple_shape_});
|