Searched refs:kInputDimensions (Results 1 – 2 of 2) sorted by relevance
155 const int32_t* dimensionsPtr = context->getInputBuffer<int32_t>(kInputDimensions); in densify()156 inputShapes.push_back(context->getInputShape(kInputDimensions)); in densify()225 const int32_t* dimensions = context->getInputBuffer<int32_t>(kInputDimensions); in prepare()226 Shape dimensionsShape = context->getInputShape(kInputDimensions); in prepare()
34 constexpr uint32_t kInputDimensions = 4; variable