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