Home
last modified time | relevance | path

Searched defs:aShape (Results 1 – 10 of 10) sorted by relevance

/aosp_15_r20/packages/modules/NeuralNetworks/common/cpu_operations/
DMaximumMinimum.cpp35 bool evalGeneric(const T* aData, const Shape& aShape, const T* bData, const Shape& bShape, in evalGeneric()
61 bool evalQuant8(const T* aData, const Shape& aShape, const T* bData, const Shape& bShape, in evalQuant8()
DPRelu.cpp46 const Shape& aShape, const T* bData, const Shape& bShape, T* outputData, in eval()
69 bool evalQuant8(const T* aData, const Shape& aShape, const T* bData, const Shape& bShape, in evalQuant8()
DSelect.cpp32 const Shape& aShape, const T* bData, const Shape& bShape, T* outputData, in compute()
DLogicalAndOr.cpp33 bool compute(const std::function<bool(bool, bool)>& func, const bool8* aData, const Shape& aShape, in compute()
DComparisons.cpp35 const Shape& aShape, const DataType* bData, const Shape& bShape, bool8* outputData, in compute()
DBroadcast.cpp211 bool executeInt32(const int32_t* aData, const Shape& aShape, const int32_t* bData, in executeInt32()
/aosp_15_r20/external/skia/tests/
H A DPathOpsSimplifyRectThreadedTest.cpp30 int aShape = state.fA & 0x03; in testSimplify4x4RectsMain() local
/aosp_15_r20/external/armnn/src/armnn/layers/
H A DConcatLayer.cpp245 const armnn::TensorShape& aShape = inputShapes[a]; in InferOutputShapes() local
/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/mlir_hlo/stablehlo/dialect/
H A DStablehloOps.cpp954 ArrayRef<int64_t> aShape = aRankedType.getShape(); in inferReturnTypeComponents() local
/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/mlir_hlo/lib/Dialect/mhlo/IR/
H A Dhlo_ops.cc952 ArrayRef<int64_t> aShape = aRankedType.getShape(); in inferReturnTypeComponents() local