Home
last modified time | relevance | path

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

/frameworks/ml/nn/common/operations/
DComparisons.cpp29 constexpr uint32_t kInputTensor1 = 0; variable
70 std::less<ComparisonType>(), context->getInputBuffer<DataType>(kInputTensor1), in executeLessTyped()
71 context->getInputShape(kInputTensor1), context->getInputBuffer<DataType>(kInputTensor2), in executeLessTyped()
79 std::less_equal<ComparisonType>(), context->getInputBuffer<DataType>(kInputTensor1), in executeLessEqualTyped()
80 context->getInputShape(kInputTensor1), context->getInputBuffer<DataType>(kInputTensor2), in executeLessEqualTyped()
88 std::equal_to<ComparisonType>(), context->getInputBuffer<DataType>(kInputTensor1), in executeEqualTyped()
89 context->getInputShape(kInputTensor1), context->getInputBuffer<DataType>(kInputTensor2), in executeEqualTyped()
97 std::not_equal_to<ComparisonType>(), context->getInputBuffer<DataType>(kInputTensor1), in executeNotEqualTyped()
98 context->getInputShape(kInputTensor1), context->getInputBuffer<DataType>(kInputTensor2), in executeNotEqualTyped()
106 std::greater_equal<ComparisonType>(), context->getInputBuffer<DataType>(kInputTensor1), in executeGreaterEqualTyped()
[all …]
DBroadcast.cpp36 constexpr uint32_t kInputTensor1 = 0; variable
356 auto inputType = context->getInputType(kInputTensor1); in validate()
368 Shape input1 = context->getInputShape(kInputTensor1); in validate()
383 Shape input1 = context->getInputShape(kInputTensor1); in prepare()
395 switch (context->getInputType(kInputTensor1)) { in executeAdd()
397 return addFloat16(context->getInputBuffer<_Float16>(kInputTensor1), in executeAdd()
398 context->getInputShape(kInputTensor1), in executeAdd()
405 return addFloat32(context->getInputBuffer<float>(kInputTensor1), in executeAdd()
406 context->getInputShape(kInputTensor1), in executeAdd()
413 return addQuant8(context->getInputBuffer<uint8_t>(kInputTensor1), in executeAdd()
[all …]
DLogicalAndOr.cpp29 constexpr uint32_t kInputTensor1 = 0; variable
64 OperandType inputType = context->getInputType(kInputTensor1); in validate()
73 Shape input1 = context->getInputShape(kInputTensor1); in prepare()
82 std::logical_and<bool>(), context->getInputBuffer<bool8>(kInputTensor1), in executeAnd()
83 context->getInputShape(kInputTensor1), context->getInputBuffer<bool8>(kInputTensor2), in executeAnd()
90 std::logical_or<bool>(), context->getInputBuffer<bool8>(kInputTensor1), in executeOr()
91 context->getInputShape(kInputTensor1), context->getInputBuffer<bool8>(kInputTensor2), in executeOr()
DSelect.cpp30 constexpr uint32_t kInputTensor1 = 1; variable
61 context->getInputShape(kInputCondition), context->getInputBuffer<T>(kInputTensor1), in executeTyped()
62 context->getInputShape(kInputTensor1), context->getInputBuffer<T>(kInputTensor2), in executeTyped()
72 OperandType inputType = context->getInputType(kInputTensor1); in validate()
84 Shape input1 = context->getInputShape(kInputTensor1); in prepare()
105 switch (context->getInputType(kInputTensor1)) { in execute()