Home
last modified time | relevance | path

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

/frameworks/ml/nn/common/operations/
DBroadcast.cpp46 constexpr uint32_t kInputTensor2 = 1; variable
457 Shape input2 = context->getInputShape(kInputTensor2); in validate()
470 const Shape& input2 = context->getInputShape(kInputTensor2); in validate()
481 Shape input2 = context->getInputShape(kInputTensor2); in prepare()
496 context->getInputBuffer<_Float16>(kInputTensor2), in executeAdd()
497 context->getInputShape(kInputTensor2), in executeAdd()
504 context->getInputBuffer<float>(kInputTensor2), in executeAdd()
505 context->getInputShape(kInputTensor2), in executeAdd()
512 context->getInputBuffer<uint8_t>(kInputTensor2), in executeAdd()
513 context->getInputShape(kInputTensor2), in executeAdd()
[all …]
DComparisons.cpp33 constexpr uint32_t kInputTensor2 = 1; variable
77 context->getInputShape(kInputTensor1), context->getInputBuffer<DataType>(kInputTensor2), in executeLessTyped()
78 context->getInputShape(kInputTensor2), context->getOutputBuffer<bool8>(kOutputTensor), in executeLessTyped()
86 context->getInputShape(kInputTensor1), context->getInputBuffer<DataType>(kInputTensor2), in executeLessEqualTyped()
87 context->getInputShape(kInputTensor2), context->getOutputBuffer<bool8>(kOutputTensor), in executeLessEqualTyped()
95 context->getInputShape(kInputTensor1), context->getInputBuffer<DataType>(kInputTensor2), in executeEqualTyped()
96 context->getInputShape(kInputTensor2), context->getOutputBuffer<bool8>(kOutputTensor), in executeEqualTyped()
104 context->getInputShape(kInputTensor1), context->getInputBuffer<DataType>(kInputTensor2), in executeNotEqualTyped()
105 context->getInputShape(kInputTensor2), context->getOutputBuffer<bool8>(kOutputTensor), in executeNotEqualTyped()
113 context->getInputShape(kInputTensor1), context->getInputBuffer<DataType>(kInputTensor2), in executeGreaterEqualTyped()
[all …]
DLogicalAndOr.cpp30 constexpr uint32_t kInputTensor2 = 1; variable
76 Shape input2 = context->getInputShape(kInputTensor2); in prepare()
85 context->getInputShape(kInputTensor1), context->getInputBuffer<bool8>(kInputTensor2), in executeAnd()
86 context->getInputShape(kInputTensor2), context->getOutputBuffer<bool8>(kOutputTensor), in executeAnd()
93 context->getInputShape(kInputTensor1), context->getInputBuffer<bool8>(kInputTensor2), in executeOr()
94 context->getInputShape(kInputTensor2), context->getOutputBuffer<bool8>(kOutputTensor), in executeOr()
DSelect.cpp31 constexpr uint32_t kInputTensor2 = 2; variable
65 context->getInputShape(kInputTensor1), context->getInputBuffer<T>(kInputTensor2), in executeTyped()
66 context->getInputShape(kInputTensor2), context->getOutputBuffer<T>(kOutputTensor), in executeTyped()
101 Shape input2 = context->getInputShape(kInputTensor2); in prepare()