Searched refs:samples_tensor (Results 1 – 2 of 2) sorted by relevance
| /aosp_15_r20/external/tensorflow/tensorflow/core/kernels/ |
| H A D | random_binomial_op.cc | 393 Tensor* samples_tensor; in Compute() local
394 OP_REQUIRES_OK(ctx, ctx->allocate_output(0, output_shape, &samples_tensor)); in Compute()
432 samples_tensor->flat<U>()); in Compute()
505 Tensor* samples_tensor; in Compute() local
506 OP_REQUIRES_OK(ctx, ctx->allocate_output(0, output_shape, &samples_tensor)); in Compute()
518 samples_tensor->flat<U>()); in Compute()
|
| H A D | parameterized_truncated_normal_op.cc | 646 Tensor* samples_tensor; in Compute() local
647 OP_REQUIRES_OK(ctx, ctx->allocate_output(0, tensor_shape, &samples_tensor)); in Compute()
723 maxvals_tensor.flat<T>(), rng, samples_tensor->flat<T>()); in Compute()
800 Tensor* samples_tensor; in Compute() local
801 OP_REQUIRES_OK(ctx, ctx->allocate_output(0, output_shape, &samples_tensor)); in Compute()
814 maxvals_tensor.flat<T>(), philox, samples_tensor->flat<T>()); in Compute()
|