Home
last modified time | relevance | path

Searched defs:concatDim (Results 1 – 8 of 8) sorted by relevance

/aosp_15_r20/external/pytorch/aten/src/ATen/native/cuda/
H A DShape.cu169 const int concatDim, in CatArrayBatchedCopy()
205 const int concatDim, in CatArrayBatchedCopy_contig()
238 const int concatDim, in CatArrayBatchedCopy_aligned16_contig()
/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/mlir_hlo/lib/Dialect/thlo/IR/
H A Dthlo_ops.cc175 uint64_t concatDim = op.dimension(); in fuseConcatenateOpThroughTile() local
280 uint64_t concatDim, SmallVector<Value> &remainingOffsets, in fuseConcatenateOpThroughPointRecursively()
345 uint64_t concatDim = op.dimension(); in fuseConcatenateOpThroughPoint() local
/aosp_15_r20/external/armnn/src/backends/backendsCommon/test/layerTests/
H A DConcatTestImpl.cpp28 unsigned int concatDim) in CreateDescriptorForConcat()
48 unsigned int concatDim) in NeedPermuteForConcat()
92 unsigned int & concatDim, in Generate3dPermuteVectorForConcat()
181 unsigned int & concatDim, in PermuteInputsForConcat()
286 unsigned int concatDim, in Concatenate()
/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/mlir_hlo/lib/Dialect/mhlo/transforms/
H A Dlegalize_mhlo_to_thlo.cc52 const uint64_t concatDim = op.dimension(); in matchAndRewrite() local
/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/mlir_hlo/lib/Dialect/lhlo/transforms/
H A Dlhlo_legalize_to_affine.cc133 uint64_t concatDim = op.getDimension(); in matchAndRewrite() local
/aosp_15_r20/external/armnn/shim/sl/canonical/
H A DConversionUtils.hpp485 uint32_t concatDim) in ValidateConcatOutputShape()
H A DConverter.cpp678 int32_t concatDim; in ConvertConcatenation() local
/aosp_15_r20/external/android-nn-driver/
H A DConversionUtils.hpp547 uint32_t concatDim) in ValidateConcatOutputShape()
1997 int32_t concatDim; in ConvertConcatenation() local