Searched defs:input_weights (Results 1 – 2 of 2) sorted by relevance
53 void BincountTask(Tensor *input_arr, int32_t num_bins, Tensor *input_weights, Tensor *output, CpuKe… in BincountTask()97 Tensor *input_weights = ctx.Input(kThirdInputIndex); in Compute() local
78 const float *input_weights; member86 const float *input_weights; member