Home
last modified time | relevance | path

Searched refs:kInputTensor2 (Results 1 – 9 of 9) sorted by relevance

/packages/modules/NeuralNetworks/common/cpu_operations/
DBroadcast.cpp443 Shape input2 = context->getInputShape(kInputTensor2); in prepare()
458 context->getInputBuffer<_Float16>(kInputTensor2), in executeAdd()
459 context->getInputShape(kInputTensor2), in executeAdd()
466 context->getInputBuffer<float>(kInputTensor2), in executeAdd()
467 context->getInputShape(kInputTensor2), in executeAdd()
474 context->getInputBuffer<uint8_t>(kInputTensor2), in executeAdd()
475 context->getInputShape(kInputTensor2), in executeAdd()
482 context->getInputBuffer<int8_t>(kInputTensor2), in executeAdd()
483 context->getInputShape(kInputTensor2), in executeAdd()
490 context->getInputBuffer<int32_t>(kInputTensor2), in executeAdd()
[all …]
DComparisons.cpp68 context->getInputShape(kInputTensor1), context->getInputBuffer<DataType>(kInputTensor2), in executeLessTyped()
69 context->getInputShape(kInputTensor2), context->getOutputBuffer<bool8>(kOutputTensor), in executeLessTyped()
77 context->getInputShape(kInputTensor1), context->getInputBuffer<DataType>(kInputTensor2), in executeLessEqualTyped()
78 context->getInputShape(kInputTensor2), context->getOutputBuffer<bool8>(kOutputTensor), in executeLessEqualTyped()
86 context->getInputShape(kInputTensor1), context->getInputBuffer<DataType>(kInputTensor2), in executeEqualTyped()
87 context->getInputShape(kInputTensor2), context->getOutputBuffer<bool8>(kOutputTensor), in executeEqualTyped()
95 context->getInputShape(kInputTensor1), context->getInputBuffer<DataType>(kInputTensor2), in executeNotEqualTyped()
96 context->getInputShape(kInputTensor2), context->getOutputBuffer<bool8>(kOutputTensor), in executeNotEqualTyped()
104 context->getInputShape(kInputTensor1), context->getInputBuffer<DataType>(kInputTensor2), in executeGreaterEqualTyped()
105 context->getInputShape(kInputTensor2), context->getOutputBuffer<bool8>(kOutputTensor), in executeGreaterEqualTyped()
[all …]
DLogicalAndOr.cpp59 Shape input2 = context->getInputShape(kInputTensor2); in prepare()
68 context->getInputShape(kInputTensor1), context->getInputBuffer<bool8>(kInputTensor2), in executeAnd()
69 context->getInputShape(kInputTensor2), context->getOutputBuffer<bool8>(kOutputTensor), in executeAnd()
76 context->getInputShape(kInputTensor1), context->getInputBuffer<bool8>(kInputTensor2), in executeOr()
77 context->getInputShape(kInputTensor2), context->getOutputBuffer<bool8>(kOutputTensor), in executeOr()
DSelect.cpp55 context->getInputShape(kInputTensor1), context->getInputBuffer<T>(kInputTensor2), in executeTyped()
56 context->getInputShape(kInputTensor2), context->getOutputBuffer<T>(kOutputTensor), in executeTyped()
76 Shape input2 = context->getInputShape(kInputTensor2); in prepare()
/packages/modules/NeuralNetworks/common/types/operations/include/
DLogicalAndOr.h26 constexpr uint32_t kInputTensor2 = 1; variable
DComparisons.h26 constexpr uint32_t kInputTensor2 = 1; variable
DBroadcast.h26 constexpr uint32_t kInputTensor2 = 1; variable
DSelect.h27 constexpr uint32_t kInputTensor2 = 2; variable
/packages/modules/NeuralNetworks/common/types/operations/src/
DBroadcast.cpp34 const Shape& input2 = context->getInputShape(kInputTensor2); in validate()