Searched defs:orig_sizes (Results 1 – 4 of 4) sorted by relevance
20 std::vector<int64_t> orig_sizes, in compute_out_sizes()
98 auto orig_sizes = qa.sizes().vec(); in add() local
386 auto orig_sizes = input_flat.sizes().vec(); in _unique_dim_cpu_template() local
1310 std::vector<int64_t> orig_sizes = {H, W}; in TEST() local1362 std::vector<int64_t> orig_sizes = {N, C, H, W}; in TEST() local