Function ReadVariableShapeFn

tensorflow/core/ops/resource_variable_ops.cc:32–44

namespace {

Status ReadVariableShapeFn(InferenceContext* c) {
  std::vector<ShapeAndType> shape_and_type;
  TF_RETURN_IF_ERROR(
      shape_inference::ValidateVariableResourceHandle(c, &shape_and_type));
  c->set_output(0, shape_and_type[0].shape);
  if (shape_and_type[0].dtype == DT_VARIANT && shape_and_type.size() > 1) {
    std::vector<ShapeAndType> variant_shape_and_type;
    std::copy(shape_and_type.begin() + 1, shape_and_type.end(),
              std::back_inserter(variant_shape_and_type));
    c->set_output_handle_shapes_and_types(0, variant_shape_and_type);
  }
  return Status::OK();
}

Status ReadVariablesShapeFn(InferenceContext* c) {
  int n;