Searched defs:nDim (Results 1 – 6 of 6) sorted by relevance
49 size_t nDim() const { in nDim() function
394 const auto nDim = desc.nDim(); in generateKernel() local
251 const auto nDim = desc.nDim(); // NOTE: this is the compressed dim in launchFusion() local
55 const auto nDim = tuple ? PyTuple_GET_SIZE(arg) : PyList_GET_SIZE(arg); in THPUtils_unpackLongs() local
305 int64_t nDim = dense.dim(); in add_out_dense_sparse_cuda() local
6847 string data_format = "", const bool is_max_pooling = true) -> NodeDef { in TEST_P()6880 for (int nDim : test_nDims) { in TEST_P() local6972 for (int nDim : test_nDims) { in TEST_P() local