Searched defs:lastDim (Results 1 – 4 of 4) sorted by relevance
385 int64_t lastDim = sz[sz.size() - 1]; in _efficient_attention_backward() local
2678 int lastDim = rank - 1; in matchAndRewrite() local
960 int64_t lastDim = aShape[aShape.size() - 1]; in inferReturnTypeComponents() local
958 int64_t lastDim = aShape[aShape.size() - 1]; in inferReturnTypeComponents() local