Searched refs:shape_elements (Results 1 – 1 of 1) sorted by relevance
357 std::vector<Shape> shape_elements = { in XLA_TEST_F() local360 Shape result_shape = ShapeUtil::MakeTupleShape(shape_elements); in XLA_TEST_F()411 std::vector<Shape> shape_elements = { in XLA_TEST_F() local414 Shape result_shape = ShapeUtil::MakeTupleShape(shape_elements); in XLA_TEST_F()469 std::vector<Shape> shape_elements = {ShapeUtil::MakeShape(S32, {}), in XLA_TEST_F() local471 Shape result_shape = ShapeUtil::MakeTupleShape(shape_elements); in XLA_TEST_F()518 std::vector<Shape> shape_elements = {ShapeUtil::MakeShape(S32, {}), in XLA_TEST_F() local520 Shape result_shape = ShapeUtil::MakeTupleShape(shape_elements); in XLA_TEST_F()564 std::vector<Shape> shape_elements = {ShapeUtil::MakeShape(S32, {}), in XLA_TEST_F() local566 Shape result_shape = ShapeUtil::MakeTupleShape(shape_elements); in XLA_TEST_F()[all …]