Searched refs:kInputBlockMap (Results 1 – 2 of 2) sorted by relevance
151 const int32_t* blockMap = context->getInputBuffer<int32_t>(kInputBlockMap); in densify()152 inputShapes.push_back(context->getInputShape(kInputBlockMap)); in densify()189 inputShapes[kInputBlockMap].dimensions.front()); // size of block map in densify()190 for (uint32_t i = 0; i < inputShapes[kInputBlockMap].dimensions.front(); i++) { in densify()224 const int32_t* blockMap = context->getInputBuffer<int32_t>(kInputBlockMap); in prepare()227 Shape blockMapShape = context->getInputShape(kInputBlockMap); in prepare()
32 constexpr uint32_t kInputBlockMap = 2; variable