HomeSort by relevance Sort by last modified time
    Searched full:seed_shape (Results 1 - 1 of 1) sorted by null

  /external/tensorflow/tensorflow/compiler/tf2xla/kernels/
stateless_random_ops.cc 228 TensorShape seed_shape = ctx->InputShape(1); variable
229 OP_REQUIRES(ctx, seed_shape.dims() == 1 && seed_shape.dim_size(0) == 2,
231 seed_shape.DebugString()));
255 TensorShape seed_shape = ctx->InputShape(1); variable
256 OP_REQUIRES(ctx, seed_shape == TensorShape({2}),
258 seed_shape.DebugString()));

Completed in 136 milliseconds