Function VecShape

tensorflow/core/kernels/random_op_test.cc:28–38

namespace {

Tensor VecShape(int64 v) {
  if (v >= std::numeric_limits<int32>::max()) {
    Tensor shape(DT_INT64, TensorShape({1}));
    shape.vec<int64>()(0) = v;
    return shape;
  } else {
    Tensor shape(DT_INT32, TensorShape({1}));
    shape.vec<int32>()(0) = v;
    return shape;
  }
}

Graph* RandomUniform(int64 n) {
  Graph* g = new Graph(OpRegistry::Global());