Searched defs:expanded_sizes (Results 1 – 6 of 6) sorted by relevance
19 std::vector<int64_t> expanded_sizes = {3, 5}; in TestOut2Basic() local53 std::vector<int64_t> expanded_sizes = {3, 2, 5}; in TestOut3Basic() local64 std::vector<int64_t> expanded_sizes = {3, 2, 5}; in TestOut3WithScalar() local
86 std::vector<int64_t> expanded_sizes(aligned_names.size(), 1); in aligned_size() local161 std::vector<int64_t> expanded_sizes = aligned_size( in align() local
109 VmapDimVector expanded_sizes(self_sizes.begin(), self_sizes.end()); in _remove_batch_dim() local
104 auto expanded_sizes = tensor.sym_sizes().vec(); in moveDimToFrontAndExpand() local
94 auto expanded_sizes = node->get<c10::List<int64_t>>(attr::size); in optimizeBlock() local
154 VmapDimVector expanded_sizes(self_sizes.begin(), self_sizes.end()); in _remove_batch_dim() local