Function CreateConstOp

tensorflow/core/grappler/costs/utils_test.cc:31–40

namespace {

void CreateConstOp(const string& name, std::initializer_list<int64> dims,
                   NodeDef* node) {
  Tensor tensor(DT_FLOAT, TensorShape(dims));
  for (int64 i = 0; i < tensor.NumElements(); ++i)
    tensor.flat<float>()(i) = i / 10.0f;
  TF_CHECK_OK(NodeDefBuilder(name, "Const")
                  .Attr("dtype", DT_FLOAT)
                  .Attr("value", tensor)
                  .Finalize(node));
}

void CreateConstSizesOp(const string& name, const std::vector<int32>& sizes,
                        NodeDef* node) {