Searched defs:dims_after (Results 1 – 3 of 3) sorted by relevance
156 int64_t dims_after; member
544 static Tensor reshape_indexer(const Tensor& index, int64_t dims_before, int64_t dims_after) { in reshape_indexer()556 int64_t dims_before = 0, dims_after = 0, dims_indexed = 0; in AdvancedIndex() local
3842 std::vector<int64_t> dims_before, dims_after; in TEST_F() local