Searched refs:multiples (Results 1 – 4 of 4) sorted by relevance
| /frameworks/ml/nn/common/operations/ |
| D | Tile.cpp | 65 void tileImpl(const T* inputData, const Shape& inputShape, const int32_t* multiples, T* outputData, in tileImpl() argument
67 TileOneDimension(inputShape, inputData, multiples, outputData, 0); in tileImpl()
72 bool prepare(const Shape& input, const int32_t* multiples, const Shape& multiplesShape, in prepare() argument
80 output->dimensions[i] *= multiples[i]; in prepare()
86 bool eval(const uint8_t* inputData, const Shape& inputShape, const int32_t* multiples, in eval() argument
92 tileImpl(reinterpret_cast<const dataType*>(inputData), inputShape, multiples, \ in eval()
|
| D | Tile.h | 26 bool prepare(const Shape& input, const int32_t* multiples, const Shape& multiplesShape,
29 bool eval(const uint8_t* inputData, const Shape& inputShape, const int32_t* multiples,
|
| /frameworks/ml/nn/common/ |
| D | CpuExecutor.cpp | 1665 const RunTimeOperandInfo& multiples = mOperands[ins[1]]; in executeOperation() local
1671 tile::prepare(input.shape(), reinterpret_cast<const int32_t*>(multiples.buffer), in executeOperation()
1672 multiples.shape(), &outShape) && in executeOperation()
1675 reinterpret_cast<const int32_t*>(multiples.buffer), output.buffer, in executeOperation()
|
| /frameworks/ml/nn/runtime/test/ |
| D | TestValidateOperations.cpp | 540 ANeuralNetworksOperandType multiples = { in tileTest() local
547 OperationTestBase test(ANEURALNETWORKS_TILE, {input0, multiples}, {output0}); in tileTest()
|