Search Options

Results per page
Sort
Preferred Languages
Advance

Results 1 - 5 of 5 for EagerContext (0.21 sec)

  1. tensorflow/c/eager/c_api.cc

      auto r = tsl::core::RefCountPtr<tensorflow::IntraProcessRendezvous>(
          new tensorflow::IntraProcessRendezvous(device_mgr.get()));
      tensorflow::EagerContext* eager_context = new tensorflow::EagerContext(
          opts->session_options.options,
          static_cast<tensorflow::ContextDevicePlacementPolicy>(
              opts->device_placement_policy),
          opts->async, device_mgr.release(),
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu May 09 08:11:23 UTC 2024
    - 44K bytes
    - Viewed (0)
  2. tensorflow/c/c_api_experimental.cc

      auto* gpu_options = config.mutable_gpu_options();
      gpu_options->set_allow_growth(gpu_memory_allow_growth);
    
      (*config.mutable_device_count())["CPU"] = num_cpu_devices;
    
      // TODO(b/113217601): This is needed for EagerContext::runner_ to use a
      // threadpool, so that we avoid the possibility of running the runner_ in the
      // threadpool of GPU event mgr, as that can trigger more callbacks to be
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Mon Apr 15 03:35:10 UTC 2024
    - 29.4K bytes
    - Viewed (0)
  3. tensorflow/compiler/mlir/tfrt/runtime_fallback/runtime_fallback_executor.cc

      // Build an ExecutionContext from the HostContext.
      auto builder = RequestContextBuilder(host_context_.get(), &resource_context_);
    
      // Get tensorflow::EagerContext for the kernel fallback.
      auto* eager_context_resource =
          resource_context_
              .GetOrCreateResource<tensorflow::tfd::EagerContextResource>(
                  tensorflow::tfd::kEagerContextResourceName);
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Wed May 08 00:18:59 UTC 2024
    - 9.1K bytes
    - Viewed (0)
  4. tensorflow/compiler/mlir/python/mlir.cc

      if (!s.ok()) {
        tsl::Set_TF_Status_from_Status(status, s);
        return "// error";
      }
    
      const std::string& function_name = functiondef.signature().name();
      EagerContext* cpp_context = ContextFromInterface(unwrap(tfe_context));
      FunctionLibraryDefinition& flib_def = *cpp_context->FuncLibDef();
      const tensorflow::FunctionDef* fdef = flib_def.Find(function_name);
      if (fdef == nullptr) {
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Fri May 03 18:16:49 UTC 2024
    - 19.3K bytes
    - Viewed (0)
  5. tensorflow/c/eager/c_api_experimental.cc

    }
    
    void TFE_ContextDisableGraphCollection(TFE_Context* ctx) {
      tensorflow::unwrap(ctx)->SetShouldStoreGraphs(false);
    }
    
    uint64_t TFE_GetContextId(TFE_Context* ctx) {
      tensorflow::EagerContext* context =
          tensorflow::ContextFromInterface(tensorflow::unwrap(ctx));
      return context->GetContextId();
    }
    
    void TFE_MonitoringCounterCellIncrementBy(TFE_MonitoringCounterCell* cell,
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu Apr 11 23:52:39 UTC 2024
    - 35.9K bytes
    - Viewed (0)
Back to top