| /aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/mlir_hlo/lib/Dialect/gml_st/transforms/ |
| H A D | fusion_interface_impl.cc | 78 if (int outputDim = operandDimsToOutputDims[i]; outputDim >= 0) { in buildPointOp() local
79 auto index = builder.create<arith::ConstantIndexOp>(loc, outputDim); in buildPointOp()
111 if (int outputDim = operandDimsToOutputDims[i]; outputDim >= 0) { in buildTileOp() local
112 auto index = builder.create<arith::ConstantIndexOp>(loc, outputDim); in buildTileOp()
|
| /aosp_15_r20/external/armnn/src/armnnUtils/ |
| H A D | TensorUtils.cpp | 142 unsigned int outputDim = tensorShape.GetNumDimensions() + 1; in ExpandDims() local
144 …if (axis < -armnn::numeric_cast<int>(outputDim) || axis > armnn::numeric_cast<int>(tensorShape.Get… in ExpandDims()
154 axis = armnn::numeric_cast<int>(outputDim) + axis; in ExpandDims()
165 return { outputDim, outputShape.data() }; in ExpandDims()
|
| /aosp_15_r20/external/armnn/src/armnnOnnxParser/test/ |
| H A D | Shape.cpp | 17 const std::string& outputDim, in ShapeMainFixture()
48 dim_value: )" + outputDim + R"( in ShapeMainFixture()
|
| /aosp_15_r20/external/armnn/src/armnn/layers/ |
| H A D | GatherLayer.cpp | 47 const unsigned int outputDim = paramsDim - 1 + indicesDim; in InferOutputShapes() local
71 return std::vector<TensorShape>({ TensorShape({outputDim, dimSizes.data()})}); in InferOutputShapes()
|
| H A D | GatherNdLayer.cpp | 74 const auto outputDim = static_cast<unsigned int>(output_shape.size()); in InferOutputShapes() local
75 return std::vector<TensorShape>({ TensorShape({outputDim, output_shape.data()})}); in InferOutputShapes()
|
| /aosp_15_r20/packages/modules/NeuralNetworks/common/cpu_operations/ |
| D | MirrorPad.cpp | 202 const auto outputDim = getSizeOfDimension(outputShape, i); in eval() local
203 NN_RET_CHECK(checkAsInt(outputDim)); in eval()
|
| /aosp_15_r20/external/armnn/src/backends/backendsCommon/ |
| H A D | WorkloadData.cpp | 2405 unsigned int outputDim = in Validate() local
2409 outputDim > 0 ? outputDim : 1, in Validate()
2693 unsigned int outputDim = outputTensorInfo.GetNumDimensions(); in Validate() local
2694 ValidateTensorNumDimensions(outputTensorInfo, descriptorName, outputDim, "output"); in Validate()
2728 …unsigned int outputDim = inputTensorInfo.GetNumDimensions() + indicesTensorInfo.GetNumDimensions(… in Validate() local
2729 ValidateTensorNumDimensions(outputTensorInfo, descriptorName, outputDim, "output"); in Validate()
|
| /aosp_15_r20/external/armnn/src/backends/reference/ |
| H A D | RefLayerSupport.cpp | 1933 …auto outputDim = input.GetNumDimensions() - armnn::numeric_cast<unsigned int>(descriptor.m_Axis.si… in IsMeanSupported() local
1935 if (outputDim > 0) in IsMeanSupported()
1937 supported &= CheckSupportRule(TensorNumDimensionsAreCorrect(output, outputDim), in IsMeanSupported()
1939 … CreateIncorrectDimensionsErrorMsg(outputDim, output.GetNumDimensions(), in IsMeanSupported()
|
| /aosp_15_r20/packages/modules/NeuralNetworks/runtime/test/ |
| D | TestValidateOperations.cpp | 3692 uint32_t outputDim[] = {1, 2, 3}; in sliceTest() local
3698 {getOpType(operandCode, 3, outputDim)}); in sliceTest()
4306 uint32_t inputDim[] = {1, 2, 2, 1}, outputDim[] = {1, 1, 1, 1}; in resizeNearestNeighborTest() local
4310 {getOpType(inputCode, 4, outputDim)}); in resizeNearestNeighborTest()
|