Lines Matching refs:kWeightsTensor

39 constexpr uint32_t kWeightsTensor = 1;  variable
250         const float weightsScale = context->getInputShape(kWeightsTensor).scale;  in validate()
283     Shape weights = context->getInputShape(kWeightsTensor);  in validate()
294     Shape weights = context->getInputShape(kWeightsTensor);  in prepare()
308                                          context->getInputBuffer<float>(kWeightsTensor),  in execute()
309                                          context->getInputShape(kWeightsTensor),  in execute()
318                                          context->getInputBuffer<_Float16>(kWeightsTensor),  in execute()
319                                          context->getInputShape(kWeightsTensor),  in execute()
328                                         context->getInputBuffer<uint8_t>(kWeightsTensor),  in execute()
329                                         context->getInputShape(kWeightsTensor),  in execute()
338                                         context->getInputBuffer<int8_t>(kWeightsTensor),  in execute()
339                                         context->getInputShape(kWeightsTensor),  in execute()