to_json Class — pytorch Architecture

Source Code

aten/src/ATen/functorch/Interpreter.h lines 108–116

  template <typename T>
  friend void to_json(T& json_j, const VmapInterpreterMeta& json_t) {
    TORCH_CHECK(
      !json_t.batchSize_.is_heap_allocated(),
      "Serialization for heap-allocated SymInt is not implemented yet"
    );
    json_j["batchSize"] = json_t.batchSize_.as_int_unchecked();
    json_j["randomness"] = static_cast<int64_t>(json_t.randomness_);
  }