Searched defs:x_dims (Results 1 – 2 of 2) sorted by relevance
365 def testBroadcastDynamicShape(self, x_dims, y_dims, expected_dims): argument
1206 std::vector<int64> x_dims(output_dims), y_dims(output_dims); in TEST_F() local1306 auto x_dims = RandomDims(2, kDefaultMaxRank); in TEST_F() local1328 auto x_dims = RandomDims(2, kDefaultMaxRank); in TEST_F() local3396 std::vector<int64> x_dims = RandomDims(/*min_rank=*/4, /*max_rank=*/4, in TEST_F() local