Searched refs:samples_tensor (Results 1 – 2 of 2) 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()
391 Tensor* samples_tensor; in Compute() local392 OP_REQUIRES_OK(ctx, ctx->allocate_output(0, output_shape, &samples_tensor)); in Compute()430 samples_tensor->flat<U>()); in Compute()