| /external/pytorch/aten/src/ATen/native/cuda/ |
| D | Shape.cu | 158 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
|
| /external/tensorflow/tensorflow/compiler/mlir/lite/quantization/ir/ |
| D | UniformSupport.cc | 74 size_t dimSize = type.getDimSize(quantizationDim); in convert() local
|
| /external/tensorflow/tensorflow/compiler/xla/mlir_hlo/include/mlir-hlo/Dialect/mhlo/IR/ |
| D | hlo_ops_base.h | 165 auto dimSize = rankedType.getShape()[dim]; in inferMostSpecificType() local
|
| /external/tensorflow/tensorflow/compiler/xla/mlir_hlo/stablehlo/dialect/ |
| D | Base.h | 211 auto dimSize = rankedType.getShape()[dim]; in inferMostSpecificType() local
|
| D | StablehloOps.cpp | 2682 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
|
| /external/tensorflow/tensorflow/compiler/xla/mlir_hlo/lib/utils/ |
| D | codegen_utils.cc | 36 auto dimSize = b.create<memref::DimOp>(loc, memref, r); in emitNumElementsComputation() local
|
| /external/armnn/src/backends/reference/workloads/ |
| D | MirrorPad.cpp | 43 unsigned int dimSize = 1; in CoordToIndex() local
|
| /external/tensorflow/tensorflow/compiler/xla/mlir_hlo/lib/Dialect/lhlo/transforms/ |
| D | lhlo_legalize_to_gpu.cc | 65 auto dimSize = shapedType.getDimSize(0); in matchAndRewrite() local
|
| D | lhlo_elemental_utils.cc | 178 auto dimSize = b->create<DimOp>(loc, operandMemref, inputDim); in elementalLowerImplForBroadcastInDimOps() local
|
| /external/armnn/delegate/classic/src/ |
| D | Split.hpp | 299 unsigned int dimSize = inputTensorInfo.GetShape()[dimIdx]; in VisitSplitVOperator() local
|
| /external/angle/third_party/glslang/src/glslang/MachineIndependent/ |
| D | reflection.cpp | 409 … 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
|
| /external/deqp-deps/glslang/glslang/MachineIndependent/ |
| D | reflection.cpp | 409 … 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
|
| /external/tensorflow/tensorflow/compiler/xla/mlir_hlo/lib/Transforms/ |
| D | symbolic_shape_optimization.cc | 624 auto dimSize = std::get<0>(it); in concretizeOperandShape() local
|
| /external/tensorflow/tensorflow/compiler/xla/mlir_hlo/lib/Dialect/mhlo/transforms/ |
| D | legalize_to_linalg.cc | 311 Value dimSize; in extractDynamicEinsumSizes() local
2741 Value dimSize = rewriter.create<tensor::DimOp>(loc, input, en.index()); in matchAndRewrite() local
|
| /external/tensorflow/tensorflow/compiler/xla/mlir_hlo/lib/Dialect/mhlo/IR/ |
| D | hlo_ops.cc | 3229 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
|
| /external/tensorflow/tensorflow/compiler/xla/mlir_hlo/lib/Dialect/gml_st/IR/ |
| D | gml_st_ops.cc | 1330 auto dimSize = std::get<0>(dimAndIndex); in verify() local
|
| /external/armnn/include/armnn/ |
| D | Descriptors.hpp | 343 unsigned int dimSize = inputShape[concatenationDimension]; in CreateDescriptorForConcatenation() local
|
| /external/angle/third_party/glslang/src/glslang/Include/ |
| D | Types.h | 1634 … const int dimSize = p.typeParameters->arraySizes->getDimSize(3); in TType() local
|
| /external/deqp-deps/glslang/glslang/Include/ |
| D | Types.h | 1627 … const int dimSize = p.typeParameters->arraySizes->getDimSize(3); in TType() local
|
| /external/armnn/src/armnnTfLiteParser/ |
| D | TfLiteParser.cpp | 4157 unsigned int dimSize = inputTensorInfo.GetShape()[dimIdx]; in ParseSplitV() local
|