Home
last modified time | relevance | path

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

/external/pytorch/aten/src/ATen/native/transformers/cuda/
Dattention_backward.cu385 int64_t lastDim = sz[sz.size() - 1]; in _efficient_attention_backward() local
/external/tensorflow/tensorflow/compiler/xla/mlir_hlo/lib/Dialect/mhlo/transforms/
Dlegalize_to_linalg.cc2678 int lastDim = rank - 1; in matchAndRewrite() local
/external/tensorflow/tensorflow/compiler/xla/mlir_hlo/stablehlo/dialect/
DStablehloOps.cpp960 int64_t lastDim = aShape[aShape.size() - 1]; in inferReturnTypeComponents() local
/external/tensorflow/tensorflow/compiler/xla/mlir_hlo/lib/Dialect/mhlo/IR/
Dhlo_ops.cc958 int64_t lastDim = aShape[aShape.size() - 1]; in inferReturnTypeComponents() local