Searched refs:samples_shape (Results 1 – 3 of 3) sorted by relevance
| /external/tensorflow/tensorflow/core/kernels/ |
| D | random_poisson_op.cc | 292 TensorShape samples_shape; in Compute() local
293 OP_REQUIRES_OK(ctx, MakeShape(shape_t, &samples_shape)); in Compute()
294 const int64 num_samples = samples_shape.num_elements(); in Compute()
296 samples_shape.AppendShape(rate_t.shape()); in Compute()
299 OP_REQUIRES_OK(ctx, ctx->allocate_output(0, samples_shape, &samples_t)); in Compute()
|
| D | random_op.cc | 285 TensorShape samples_shape; in Compute() local
289 &samples_shape)); in Compute()
293 &samples_shape)); in Compute()
295 const int64 num_samples = samples_shape.num_elements(); in Compute()
297 samples_shape.AppendShape(alpha_t.shape()); in Compute()
300 OP_REQUIRES_OK(ctx, ctx->allocate_output(0, samples_shape, &samples_t)); in Compute()
|
| /external/tensorflow/tensorflow/contrib/distributions/python/ops/ |
| D | mixture.py | 349 samples_shape = static_samples_shape.as_list()
352 samples_shape = array_ops.shape(cat_samples)
371 math_ops.range(0, samples_size), samples_shape)
385 array_ops.tile(math_ops.range(0, batch_size), [n]), samples_shape)
442 array_ops.concat([samples_shape,
|