Searched refs:kInputPaddingTensor (Results 1 – 3 of 3) sorted by relevance
39 NN_RET_CHECK_EQ(context->getInputType(kInputPaddingTensor), OperandType::TENSOR_INT32); in validate()40 const Shape inputPaddingTensorShape = context->getInputShape(kInputPaddingTensor); in validate()43 << "Input tensor #" << kInputPaddingTensor << " must have 2 dimensions"; in validate()45 NN_RET_CHECK(!dim1 || dim1 == 2) << "Input tensor #" << kInputPaddingTensor in validate()72 << "Input tensor #" << kInputPaddingTensor << " first dimension " in validate()
50 NN_RET_CHECK(padPrepare(inputShape, context->getInputBuffer<int32_t>(kInputPaddingTensor), in prepare()51 context->getInputShape(kInputPaddingTensor), &outputShape)); in prepare()57 const int32_t* paddingValues = context->getInputBuffer<int32_t>(kInputPaddingTensor); in prepare()188 const int32_t* padding = context->getInputBuffer<int32_t>(kInputPaddingTensor); in eval()
30 constexpr uint32_t kInputPaddingTensor = 1; variable