Home
last modified time | relevance | path

Searched defs:dims_after (Results 1 – 3 of 3) sorted by relevance

/external/pytorch/aten/src/ATen/native/
DIndexingUtils.h156 int64_t dims_after; member
DTensorAdvancedIndexing.cpp544 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
/external/tensorflow/tensorflow/compiler/tests/
Drandomized_tests.cc3842 std::vector<int64_t> dims_before, dims_after; in TEST_F() local