Home
last modified time | relevance | path

Searched defs:nDim (Results 1 – 6 of 6) sorted by relevance

/aosp_15_r20/external/pytorch/torch/csrc/jit/codegen/fuser/
H A Dtensor_desc.h49 size_t nDim() const { in nDim() function
H A Dcodegen.cpp394 const auto nDim = desc.nDim(); in generateKernel() local
H A Dexecutor.cpp251 const auto nDim = desc.nDim(); // NOTE: this is the compressed dim in launchFusion() local
/aosp_15_r20/external/pytorch/torch/csrc/
H A Dutils.cpp55 const auto nDim = tuple ? PyTuple_GET_SIZE(arg) : PyList_GET_SIZE(arg); in THPUtils_unpackLongs() local
/aosp_15_r20/external/pytorch/aten/src/ATen/native/sparse/cuda/
H A DSparseCUDATensorMath.cu305 int64_t nDim = dense.dim(); in add_out_dense_sparse_cuda() local
/aosp_15_r20/external/tensorflow/tensorflow/compiler/tf2tensorrt/convert/
H A Dconvert_nodes_test.cc6847 string data_format = "", const bool is_max_pooling = true) -> NodeDef { in TEST_P()
6880 for (int nDim : test_nDims) { in TEST_P() local
6972 for (int nDim : test_nDims) { in TEST_P() local