Function ConstructOp

tensorflow/core/graph/star_server_graph_partition_test.cc:89–120

.SetShapeFn(shape_inference::ScalarShape);

Output ConstructOp(const Scope& scope, const string& op_type,
                   const string &device,
                   const gtl::ArraySlice<Input>& inputs,
                   const gtl::ArraySlice<Node*>& control_inputs = {}) {

  if (op_type == "VariableV2") {
    auto v = ops::Variable(scope, {}, DT_FLOAT);
    v.node()->set_assigned_device_name(device);
    return v;
  }
  if (!scope.ok()) {
      return Output();
  }
  const string unique_name = scope.GetUniqueNameForOp(op_type);
  auto builder =
    NodeBuilder(unique_name, op_type, scope.graph()->op_registry());
  builder.Device(device);
  for (auto const& input : inputs) {
    builder.Input(ops::NodeOut(input.node(), input.index()));
  }
  if (!control_inputs.empty()) {
    builder.ControlInputs(control_inputs);
  }
  scope.UpdateBuilder(&builder);
  Node* ret;
  scope.UpdateStatus(builder.Finalize(scope.graph(), &ret));
  if (!scope.ok()) {
      return Output();
  }
  scope.UpdateStatus(scope.DoShapeInference(ret));
  return Output(ret);
}

Output FloatInput(const Scope& scope, const string &device) {
  return ConstructOp(scope, "FloatInput", device, {});