Home
last modified time | relevance | path

Searched defs:lastDim (Results 1 – 5 of 5) sorted by relevance

/aosp_15_r20/cts/tests/tests/transition/src/android/transition/cts/
H A DChangeBoundsTest.java398 private int validateDim(String name, String dimen, int lastDim, int newDim) { in validateDim()
/aosp_15_r20/external/pytorch/aten/src/ATen/native/transformers/cuda/
H A Dattention_backward.cu385 int64_t lastDim = sz[sz.size() - 1]; in _efficient_attention_backward() local
/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/mlir_hlo/lib/Dialect/mhlo/transforms/
H A Dlegalize_to_linalg.cc2678 int lastDim = rank - 1; in matchAndRewrite() local
/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/mlir_hlo/stablehlo/dialect/
H A DStablehloOps.cpp960 int64_t lastDim = aShape[aShape.size() - 1]; in inferReturnTypeComponents() local
/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/mlir_hlo/lib/Dialect/mhlo/IR/
H A Dhlo_ops.cc958 int64_t lastDim = aShape[aShape.size() - 1]; in inferReturnTypeComponents() local