Searched defs:outputDim (Results 1 – 8 of 8) sorted by relevance
| /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
111 if (int outputDim = operandDimsToOutputDims[i]; outputDim >= 0) { in buildTileOp() local
|
| /aosp_15_r20/external/armnn/src/armnn/layers/ |
| H A D | GatherNdLayer.cpp | 74 const auto outputDim = static_cast<unsigned int>(output_shape.size()); in InferOutputShapes() local
|
| H A D | GatherLayer.cpp | 47 const unsigned int outputDim = paramsDim - 1 + indicesDim; in InferOutputShapes() local
|
| /aosp_15_r20/packages/modules/NeuralNetworks/common/cpu_operations/ |
| D | MirrorPad.cpp | 202 const auto outputDim = getSizeOfDimension(outputShape, i); in eval() local
|
| /aosp_15_r20/external/armnn/src/armnnUtils/ |
| H A D | TensorUtils.cpp | 142 unsigned int outputDim = tensorShape.GetNumDimensions() + 1; in ExpandDims() local
|
| /aosp_15_r20/external/armnn/src/backends/backendsCommon/ |
| H A D | WorkloadData.cpp | 2405 unsigned int outputDim = in Validate() local
2693 unsigned int outputDim = outputTensorInfo.GetNumDimensions(); in Validate() local
2728 …unsigned int outputDim = inputTensorInfo.GetNumDimensions() + indicesTensorInfo.GetNumDimensions(… in Validate() local
|
| /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
|
| /aosp_15_r20/packages/modules/NeuralNetworks/runtime/test/ |
| D | TestValidateOperations.cpp | 3692 uint32_t outputDim[] = {1, 2, 3}; in sliceTest() local
4306 uint32_t inputDim[] = {1, 2, 2, 1}, outputDim[] = {1, 1, 1, 1}; in resizeNearestNeighborTest() local
|