Home
last modified time | relevance | path

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

/frameworks/ml/nn/common/operations/
DComparisons.cpp32 constexpr uint32_t kInputTensor1 = 0; variable
76 std::less<ComparisonType>(), context->getInputBuffer<DataType>(kInputTensor1), in executeLessTyped()
77 context->getInputShape(kInputTensor1), context->getInputBuffer<DataType>(kInputTensor2), in executeLessTyped()
85 std::less_equal<ComparisonType>(), context->getInputBuffer<DataType>(kInputTensor1), in executeLessEqualTyped()
86 context->getInputShape(kInputTensor1), context->getInputBuffer<DataType>(kInputTensor2), in executeLessEqualTyped()
94 std::equal_to<ComparisonType>(), context->getInputBuffer<DataType>(kInputTensor1), in executeEqualTyped()
95 context->getInputShape(kInputTensor1), context->getInputBuffer<DataType>(kInputTensor2), in executeEqualTyped()
103 std::not_equal_to<ComparisonType>(), context->getInputBuffer<DataType>(kInputTensor1), in executeNotEqualTyped()
104 context->getInputShape(kInputTensor1), context->getInputBuffer<DataType>(kInputTensor2), in executeNotEqualTyped()
112 std::greater_equal<ComparisonType>(), context->getInputBuffer<DataType>(kInputTensor1), in executeGreaterEqualTyped()
[all …]
DBroadcast.cpp45 constexpr uint32_t kInputTensor1 = 0; variable
444 auto inputType = context->getInputType(kInputTensor1); in validate()
456 Shape input1 = context->getInputShape(kInputTensor1); in validate()
469 const Shape& input1 = context->getInputShape(kInputTensor1); in validate()
480 Shape input1 = context->getInputShape(kInputTensor1); in prepare()
492 switch (context->getInputType(kInputTensor1)) { in executeAdd()
494 return addFloat16(context->getInputBuffer<_Float16>(kInputTensor1), in executeAdd()
495 context->getInputShape(kInputTensor1), in executeAdd()
502 return addFloat32(context->getInputBuffer<float>(kInputTensor1), in executeAdd()
503 context->getInputShape(kInputTensor1), in executeAdd()
[all …]
DLogicalAndOr.cpp29 constexpr uint32_t kInputTensor1 = 0; variable
66 OperandType inputType = context->getInputType(kInputTensor1); in validate()
75 Shape input1 = context->getInputShape(kInputTensor1); in prepare()
84 std::logical_and<bool>(), context->getInputBuffer<bool8>(kInputTensor1), in executeAnd()
85 context->getInputShape(kInputTensor1), context->getInputBuffer<bool8>(kInputTensor2), in executeAnd()
92 std::logical_or<bool>(), context->getInputBuffer<bool8>(kInputTensor1), in executeOr()
93 context->getInputShape(kInputTensor1), context->getInputBuffer<bool8>(kInputTensor2), in executeOr()
DSelect.cpp30 constexpr uint32_t kInputTensor1 = 1; variable
64 context->getInputShape(kInputCondition), context->getInputBuffer<T>(kInputTensor1), in executeTyped()
65 context->getInputShape(kInputTensor1), context->getInputBuffer<T>(kInputTensor2), in executeTyped()
75 OperandType inputType = context->getInputType(kInputTensor1); in validate()
89 Shape input1 = context->getInputShape(kInputTensor1); in prepare()
110 switch (context->getInputType(kInputTensor1)) { in execute()