Search Options

Results per page
Sort
Preferred Languages
Advance

Results 1 - 10 of 13 for containers (0.25 sec)

  1. tensorflow/c/c_api.cc

    static void TF_Reset_Helper(const TF_SessionOptions* opt,
                                const char** containers, int ncontainers,
                                TF_Status* status) {
      std::vector<string> container_names(ncontainers);
      for (int i = 0; i < ncontainers; ++i) {
        container_names[i] = containers[i];
      }
    
      status->status = Reset(opt->options, container_names);
    }
    
    extern "C" {
    
    C++
    - Registered: Tue Apr 30 12:39:09 GMT 2024
    - Last Modified: Mon Apr 15 03:35:10 GMT 2024
    - 102.3K bytes
    - Viewed (0)
  2. tensorflow/c/eager/c_api_unified_experimental.cc

    limitations under the License.
    ==============================================================================*/
    
    #include "tensorflow/c/eager/c_api_unified_experimental.h"
    
    #include <vector>
    
    #include "absl/container/flat_hash_map.h"
    #include "absl/strings/str_cat.h"
    #include "tensorflow/c/eager/c_api_unified_experimental_internal.h"
    #include "tensorflow/c/tf_datatype.h"
    #include "tensorflow/c/tf_status.h"
    C++
    - Registered: Tue Apr 30 12:39:09 GMT 2024
    - Last Modified: Thu Feb 15 09:49:45 GMT 2024
    - 9K bytes
    - Viewed (0)
  3. tensorflow/c/experimental/next_pluggable_device/c_api.cc

      auto cc_status =
          cc_ctx->resource_manager()->Create<tensorflow::PluginResource>(
              container_name, plugin_resource_name, cc_resource_ptr);
      status->status = cc_status;
    }
    
    void TF_LookupOrCreatePluginResource(
        TF_OpKernelContext* ctx, const char* container_name,
        const char* plugin_resource_name, void** result_plugin_resource,
        void* (*create_func)(void*), void* create_func_args,
    C++
    - Registered: Tue Feb 27 12:39:08 GMT 2024
    - Last Modified: Tue Jan 09 00:52:04 GMT 2024
    - 13.9K bytes
    - Viewed (1)
  4. tensorflow/c/eager/gradients_test.cc

    limitations under the License.
    ==============================================================================*/
    #include "tensorflow/c/eager/gradients.h"
    
    #include <memory>
    
    #include "absl/container/flat_hash_set.h"
    #include "absl/types/span.h"
    #include "tensorflow/c/eager/abstract_context.h"
    #include "tensorflow/c/eager/abstract_tensor_handle.h"
    #include "tensorflow/c/eager/c_api_experimental.h"
    C++
    - Registered: Tue Apr 30 12:39:09 GMT 2024
    - Last Modified: Thu Feb 15 09:49:45 GMT 2024
    - 7K bytes
    - Viewed (0)
  5. tensorflow/c/eager/unified_api_testutil.cc

    limitations under the License.
    ==============================================================================*/
    #include "tensorflow/c/eager/unified_api_testutil.h"
    
    #include "absl/container/flat_hash_set.h"
    #include "tensorflow/c/eager/c_api_experimental.h"
    #include "tensorflow/c/eager/c_api_test_util.h"
    #include "tensorflow/c/eager/c_api_unified_experimental.h"
    C++
    - Registered: Tue Apr 30 12:39:09 GMT 2024
    - Last Modified: Tue Feb 27 13:57:45 GMT 2024
    - 5.7K bytes
    - Viewed (0)
  6. tensorflow/c/experimental/next_pluggable_device/tensor_pjrt_buffer_util.cc

    }
    
    absl::Status ResetPjRtClient(const DeviceType& device_type) {
      ResourceMgr* rmgr = tfrt_global::GetTFGlobalResourceMgr();
      PjRtState* pjrt_state;
      TF_RETURN_IF_ERROR(rmgr->Lookup(rmgr->default_container(),
                                      kPjRtStateResourceName, &pjrt_state));
      TF_RETURN_IF_ERROR(pjrt_state->MovePjRtClientToUnused(device_type));
      return absl::OkStatus();
    }
    
    C++
    - Registered: Tue Feb 27 12:39:08 GMT 2024
    - Last Modified: Mon Oct 30 19:20:20 GMT 2023
    - 3.7K bytes
    - Viewed (0)
  7. tensorflow/c/eager/c_api_test_util.cc

      if (TF_GetCode(status) != TF_OK) return nullptr;
      TFE_OpSetAttrType(op, "dtype", TF_FLOAT);
      TFE_OpSetAttrShape(op, "shape", {}, 0, status);
      TFE_OpSetAttrString(op, "container", "localhost", 0);
      TFE_OpSetAttrString(op, "shared_name", "", 0);
      if (!device_name.empty()) {
        TFE_OpSetDevice(op, device_name.c_str(), status);
      }
      if (TF_GetCode(status) != TF_OK) return nullptr;
    C++
    - Registered: Tue Apr 30 12:39:09 GMT 2024
    - Last Modified: Wed Feb 21 22:37:46 GMT 2024
    - 23.5K bytes
    - Viewed (2)
  8. tensorflow/c/c_api_function_test.cc

      }
    
      FunctionDef fdef() {
        tensorflow::FunctionDef fdef;
        EXPECT_TRUE(GetFunctionDef(func_, &fdef));
        return fdef;
      }
    
      // logging utility
      template <class Container>
      string ToString(const Container& v) {
        std::stringstream ss;
        ss << "{";
        size_t i = 0;
        for (const auto& e : v) {
          if (i != 0) {
            ss << ", ";
          }
          ss << e.ToString();
          ++i;
    C++
    - Registered: Tue Apr 30 12:39:09 GMT 2024
    - Last Modified: Thu Jul 20 22:08:54 GMT 2023
    - 63.6K bytes
    - Viewed (6)
  9. tensorflow/c/eager/c_api_experimental.cc

    ==============================================================================*/
    
    #include "tensorflow/c/eager/c_api_experimental.h"
    
    #include <cstdint>
    #include <memory>
    #include <utility>
    #include <vector>
    
    #include "absl/container/flat_hash_map.h"
    #include "absl/strings/match.h"
    #include "absl/time/time.h"
    #include "tensorflow/c/c_api.h"
    #include "tensorflow/c/eager/c_api_internal.h"
    #include "tensorflow/c/eager/tfe_context_internal.h"
    C++
    - Registered: Tue Apr 30 12:39:09 GMT 2024
    - Last Modified: Thu Apr 11 23:52:39 GMT 2024
    - 35.9K bytes
    - Viewed (3)
  10. tensorflow/c/eager/c_api.cc

    #include "tensorflow/c/eager/c_api.h"
    
    #include <algorithm>
    #include <cstddef>
    #include <cstdint>
    #include <memory>
    #include <string>
    #include <utility>
    #include <vector>
    
    #include "absl/algorithm/container.h"
    #include "absl/memory/memory.h"
    #include "tensorflow/c/c_api.h"
    #include "tensorflow/c/c_api_internal.h"
    #include "tensorflow/c/eager/abstract_tensor_handle.h"
    #include "tensorflow/c/eager/c_api_experimental.h"
    C++
    - Registered: Tue Apr 30 12:39:09 GMT 2024
    - Last Modified: Tue Mar 12 20:00:09 GMT 2024
    - 43.9K bytes
    - Viewed (2)
Back to top