Searched refs:multiples (Results 1 – 5 of 5) sorted by relevance
| /frameworks/ml/nn/common/operations/ |
| D | Tile.cpp | 71 void tileImpl(const T* inputData, const Shape& inputShape, const int32_t* multiples, T* outputData, in tileImpl() argument
73 TileOneDimension(inputShape, inputData, multiples, outputData, 0); in tileImpl()
78 bool prepare(const Shape& input, const int32_t* multiples, const Shape& multiplesShape, in prepare() argument
86 output->dimensions[i] *= multiples[i]; in prepare()
92 bool eval(const uint8_t* inputData, const Shape& inputShape, const int32_t* multiples, in eval() argument
98 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 | 1655 const RunTimeOperandInfo& multiples = operands[ins[1]]; in executeOperation() local
1661 tile::prepare(input.shape(), reinterpret_cast<const int32_t*>(multiples.buffer), in executeOperation()
1662 multiples.shape(), &outShape) && in executeOperation()
1665 reinterpret_cast<const int32_t*>(multiples.buffer), output.buffer, in executeOperation()
|
| /frameworks/ml/nn/runtime/test/ |
| D | TestValidateOperations.cpp | 1251 ANeuralNetworksOperandType multiples = { in tileTest() local
1258 OperationTestBase test(ANEURALNETWORKS_TILE, {input0, multiples}, {output0}); in tileTest()
|
| /frameworks/ml/nn/tools/api/ |
| D | types.spec | 5495 * times. The output tensor's i-th dimension has `input.dims(i) * multiples[i]`
5496 * elements, and the values of `input` are replicated `multiples[i]` times
5513 * * 1: multiples, a 1-D tensor of {@link %{OperandTypeLinkPfx}TENSOR_INT32}.
5514 * The length of multiples must be n.
|