Function Iota

tensorflow/core/kernels/training_ops_test.cc:76–81

}

static Node* Iota(Graph* g, int n) {
  Tensor data(DT_INT32, TensorShape({n}));
  int32* base = data.flat<int32>().data();
  for (int i = 0; i < n; ++i) base[i] = i;
  return test::graph::Constant(g, data);
}

static Node* Scalar(Graph* g, float val) {
  Tensor data(DT_FLOAT, TensorShape({}));