Home
last modified time | relevance | path

Searched refs:kInputFirstTensor (Results 1 – 3 of 3) sorted by relevance

/aosp_15_r20/packages/modules/NeuralNetworks/common/cpu_operations/
DPack.cpp44 const Shape firstInputTensorShape = context->getInputShape(kInputFirstTensor); in prepare()
50 context->getInputShape(kInputFirstTensor + inputTensorNum))) in prepare()
95 const Shape tensorShape = context->getInputShape(kInputFirstTensor); in packParams()
117 convertShapeToTflshape(context->getInputShape(kInputFirstTensor)); in pack()
123 inputData[inputTensorNum] = context->getInputBuffer<T>(kInputFirstTensor + inputTensorNum); in pack()
133 switch (context->getInputType(kInputFirstTensor)) { in execute()
/aosp_15_r20/packages/modules/NeuralNetworks/common/types/operations/include/
DPack.h29 constexpr uint32_t kInputFirstTensor = 1; variable
/aosp_15_r20/packages/modules/NeuralNetworks/common/types/operations/src/
DPack.cpp50 const Shape inputShape = context->getInputShape(kInputFirstTensor + inputTensorNum); in validate()