Searched refs:AllocateOutputWithShape (Results 1 – 1 of 1) sorted by relevance
| /external/tensorflow/tensorflow/core/kernels/ |
| D | random_op.cc | 182 static Status AllocateOutputWithShape(OpKernelContext* ctx, const Tensor& shape, in AllocateOutputWithShape() function
202 OP_REQUIRES_OK(ctx, AllocateOutputWithShape(ctx, shape, 0, &output)); in Compute()
236 OP_REQUIRES_OK(ctx, AllocateOutputWithShape(ctx, shape, 0, &output)); in Compute()
|