Searched refs:samples_tensor (Results 1 – 1 of 1) sorted by relevance
353 Tensor* samples_tensor; in Compute() local354 OP_REQUIRES_OK(ctx, ctx->allocate_output(0, tensor_shape, &samples_tensor)); in Compute()430 maxvals_tensor.flat<T>(), rng, samples_tensor->flat<T>()); in Compute()