Home
last modified time | relevance | path

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

/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/mlir_hlo/stablehlo/dialect/
H A DStablehloOps.cpp6026 for (const auto& spatialDim : llvm::enumerate(spatialDims)) { in printConvolutionDimensions() local
6153 int64_t spatialDim; in parseConvolutionDimensions() local
/aosp_15_r20/external/android-nn-driver/
H A DConversionUtils.hpp4120 unsigned int spatialDim = rank - 2; in ConvertSpaceToBatchNd() local
/aosp_15_r20/external/armnn/shim/sl/canonical/
H A DConverter.cpp5068 unsigned int spatialDim = rank - 2; in ConvertSpaceToBatchNd() local
/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/mlir_hlo/lib/Dialect/mhlo/IR/
H A Dhlo_ops.cc8852 for (const auto& spatialDim : llvm::enumerate(spatialDims)) { in printConvolutionDimensions() local
8979 int64_t spatialDim; in parseConvolutionDimensions() local