Searched defs:counts_tensor (Results 1 – 1 of 1) sorted by relevance
339 const Tensor& counts_tensor = ctx->input(3); in Compute() local452 const Tensor& counts_tensor = ctx->input(2); in Compute() local