Home
last modified time | relevance | path

Searched defs:dimSize (Results 1 – 24 of 24) sorted by relevance

/aosp_15_r20/external/pytorch/aten/src/ATen/native/cuda/
H A DShape.cu158 IndexType dimSize[n]; member
181 IndexType dimSize = inputs.dimSize[blockIdx.y]; in CatArrayBatchedCopy() local
215 IndexType dimSize = inputs.dimSize[blockIdx.y]; in CatArrayBatchedCopy_contig() local
253 IndexType dimSize = inputs.dimSize[blockIdx.y]; in CatArrayBatchedCopy_aligned16_contig() local
335 int64_t dimSize = 0; in parallel_cat() local
530 int64_t dimSize = t.size(dim); in TORCH_IMPL_FUNC() local
/aosp_15_r20/external/angle/third_party/glslang/src/glslang/MachineIndependent/
H A Dreflection.cpp409 … auto dimSize = derefType.isUnsizedArray() ? 0 : derefType.getArraySizes()->getDimSize(0); in blowUpActiveAggregate() local
414 … auto dimSize = derefType.isUnsizedArray() ? 0 : derefType.getArraySizes()->getDimSize(0); in blowUpActiveAggregate() local
/aosp_15_r20/external/deqp-deps/glslang/glslang/MachineIndependent/
Dreflection.cpp409 … auto dimSize = derefType.isUnsizedArray() ? 0 : derefType.getArraySizes()->getDimSize(0); in blowUpActiveAggregate() local
414 … auto dimSize = derefType.isUnsizedArray() ? 0 : derefType.getArraySizes()->getDimSize(0); in blowUpActiveAggregate() local
/aosp_15_r20/external/tensorflow/tensorflow/compiler/mlir/lite/quantization/ir/
H A DUniformSupport.cc74 size_t dimSize = type.getDimSize(quantizationDim); in convert() local
/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/mlir_hlo/include/mlir-hlo/Dialect/mhlo/IR/
H A Dhlo_ops_base.h165 auto dimSize = rankedType.getShape()[dim]; in inferMostSpecificType() local
/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/mlir_hlo/stablehlo/dialect/
H A DBase.h211 auto dimSize = rankedType.getShape()[dim]; in inferMostSpecificType() local
H A DStablehloOps.cpp2682 auto dimSize = operandType.getDimSize(i); in verify() local
2749 auto dimSize = operandType.getDimSize(i); in verify() local
3155 int64_t dimSize = operandType.getDimSize(i); in verify() local
/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/mlir_hlo/lib/utils/
H A Dcodegen_utils.cc36 auto dimSize = b.create<memref::DimOp>(loc, memref, r); in emitNumElementsComputation() local
/aosp_15_r20/external/armnn/src/backends/reference/workloads/
H A DMirrorPad.cpp43 unsigned int dimSize = 1; in CoordToIndex() local
/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/mlir_hlo/lib/Dialect/lhlo/transforms/
H A Dlhlo_legalize_to_gpu.cc65 auto dimSize = shapedType.getDimSize(0); in matchAndRewrite() local
H A Dlhlo_elemental_utils.cc178 auto dimSize = b->create<DimOp>(loc, operandMemref, inputDim); in elementalLowerImplForBroadcastInDimOps() local
/aosp_15_r20/external/armnn/delegate/classic/src/
H A DSplit.hpp299 unsigned int dimSize = inputTensorInfo.GetShape()[dimIdx]; in VisitSplitVOperator() local
/aosp_15_r20/test/mlts/benchmark/native/
Dmulti_process_test.cpp81 [&inputElementCount](int dimSize) { inputElementCount *= dimSize; }); in readInputData()
/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/mlir_hlo/lib/Transforms/
H A Dsymbolic_shape_optimization.cc624 auto dimSize = std::get<0>(it); in concretizeOperandShape() local
/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/mlir_hlo/lib/Dialect/mhlo/transforms/
H A Dlegalize_to_linalg.cc311 Value dimSize; in extractDynamicEinsumSizes() local
2741 Value dimSize = rewriter.create<tensor::DimOp>(loc, input, en.index()); in matchAndRewrite() local
/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/mlir_hlo/lib/Dialect/mhlo/IR/
H A Dhlo_ops.cc3229 auto dimSize = operandType.getDimSize(i); in verify() local
3383 auto dimSize = operandType.getDimSize(i); in verify() local
4210 int64_t dimSize = operandType.getDimSize(i); in verify() local
5641 int64_t dimSize = ty.getDimSize(dimension()); in fold() local
7019 auto dimSize = inputTy.getShape()[dimension]; in matchAndRewrite() local
/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/mlir_hlo/lib/Dialect/gml_st/IR/
H A Dgml_st_ops.cc1330 auto dimSize = std::get<0>(dimAndIndex); in verify() local
/aosp_15_r20/external/armnn/include/armnn/
H A DDescriptors.hpp343 unsigned int dimSize = inputShape[concatenationDimension]; in CreateDescriptorForConcatenation() local
/aosp_15_r20/external/deqp-deps/glslang/glslang/Include/
DTypes.h1622 … const int dimSize = p.typeParameters->arraySizes->getDimSize(3); in TType() local
/aosp_15_r20/external/angle/third_party/glslang/src/glslang/Include/
H A DTypes.h1627 … const int dimSize = p.typeParameters->arraySizes->getDimSize(3); in TType() local
/aosp_15_r20/external/armnn/src/armnnTfLiteParser/
H A DTfLiteParser.cpp4157 unsigned int dimSize = inputTensorInfo.GetShape()[dimIdx]; in ParseSplitV() local
/aosp_15_r20/external/icu/tools/srcgen/currysrc/libs/
HDorg.eclipse.jdt.core_3.14.0.v20180528-0519.jarMETA-INF/MANIFEST.MF META-INF/ECLIPSE_.SF META-INF/ECLIPSE_ ...
/aosp_15_r20/out/soong/.intermediates/prebuilts/tools/common/m2/dialer-dagger2-compiler-deps/linux_glibc_common/local-combined/
Ddialer-dagger2-compiler-deps.jarMETA-INF/ META-INF/MANIFEST.MF META-INF/eclipse.inf META ...
/aosp_15_r20/prebuilts/tools/common/m2/repository/com/google/dagger/dagger-compiler/2.7/
HDdagger-compiler-2.7.jarMETA-INF/ META-INF/MANIFEST.MF dagger/ dagger/internal/ dagger/ ...