Searched refs:total_dim_size (Results 1 – 3 of 3) sorted by relevance
171 int64 total_dim_size = 1; in TEST_F() local176 total_dim_size *= dim_sizes[i]; in TEST_F()
1479 int64_t total_dim_size = 0; // Total dimension size assigned to splits in Verify() local1496 total_dim_size += dim_val; in Verify()1500 if (!dynamic_dim_index && total_dim_size != input_dim_size) in Verify()1504 << total_dim_size << " vs " << input_dim_size; in Verify()1506 if (dynamic_dim_index && total_dim_size > input_dim_size) in Verify()1510 << total_dim_size << " vs " << input_dim_size; in Verify()
3020 int64_t total_dim_size = 0; // Total dimension size assigned to splits in matchAndRewrite() local3032 total_dim_size += dim_val; in matchAndRewrite()3044 assert(((dynamic_dim_index && total_dim_size <= input_dim_size) || in matchAndRewrite()3045 (!dynamic_dim_index && total_dim_size == input_dim_size)) && in matchAndRewrite()3050 split_sizes[*dynamic_dim_index] = input_dim_size - total_dim_size; in matchAndRewrite()