Function GetStack

tensorflow/core/kernels/stack.cc:127–152

};

Status GetStack(OpKernelContext* ctx, Stack** stack) {
  if (ctx->input_dtype(0) == DT_RESOURCE) {
    return LookupResource(ctx, HandleFromInput(ctx, 0), stack);
  } else {
    Tensor Tstack_handle = ctx->mutable_input(0, false);
    if (Tstack_handle.NumElements() != 2) {
      return errors::InvalidArgument(
          "Stack handle must have two elements, but had shape: ",
          Tstack_handle.shape().DebugString());
    }
    const string& container = Tstack_handle.flat<tstring>()(0);
    const string& stack_name = Tstack_handle.flat<tstring>()(1);
    string key = strings::StrCat(container, stack_name);
    ResourceMgr* rm = ctx->resource_manager();
    if (rm == nullptr) {
      return errors::Internal("No resource manager.");
    }
    auto* step_container = ctx->step_container();
    if (step_container == nullptr) {
      return errors::Internal("No step container.");
    }
    //TF_RETURN_IF_ERROR(rm->Lookup(step_container->name(), key, stack));
    TF_RETURN_IF_ERROR(step_container->Lookup(rm, key, stack));
    return Status::OK();
  }
}

std::atomic<int64> Stack::stack_counter{0};