Searched defs:probs_tensor (Results 1 – 1 of 1) sorted by relevance
340 const Tensor& probs_tensor = ctx->input(4); in Compute() local453 const Tensor& probs_tensor = ctx->input(3); in Compute() local