Method DumpEv

tensorflow/core/kernels/save_restore_v2_ops.cc:122–137

  template <typename TKey, typename TValue, typename TGlobalStep>
  void DumpEv(OpKernelContext* context, int variable_index,
      const string& tensor_name, BundleWriter& writer) {
    EmbeddingVar<TKey, TValue>* variable = nullptr;
    OP_REQUIRES_OK(context,
                   LookupResource(context,
                     HandleFromInput(context, variable_index), &variable));
    const Tensor& global_step = context->input(3);
    TGlobalStep global_step_scalar = global_step.scalar<TGlobalStep>()();
    core::ScopedUnref s(variable);
    embedding::ShrinkArgs shrink_args;
    shrink_args.global_step = global_step_scalar;
    const Tensor& prefix = context->input(0);
    const string& prefix_string = prefix.scalar<tstring>()();
    OP_REQUIRES_OK(context, variable->Save(tensor_name,
        prefix_string, &writer, shrink_args));
  }

  void Compute(OpKernelContext* context) override {
    const Tensor& prefix = context->input(0);