- Sort Score
- Result 10 results
- Languages All
Results 21 - 29 of 29 for contain (7.66 sec)
-
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) -
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) -
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) -
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) -
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) -
tensorflow/c/c_api_test.cc
Status TF_TensorToTensor(const TF_Tensor* src, Tensor* dst); namespace { static void ExpectHasSubstr(StringPiece s, StringPiece expected) { EXPECT_TRUE(absl::StrContains(s, expected)) << "'" << s << "' does not contain '" << expected << "'"; } // Returns the GPU device name if there is one (with arbitrary tie breaking if // there are more than one), or "" otherwise. string GPUDeviceName(TF_Session* session) {
C++ - Registered: Tue Apr 30 12:39:09 GMT 2024 - Last Modified: Mon Apr 15 03:35:10 GMT 2024 - 96.9K bytes - Viewed (3) -
tensorflow/c/checkpoint_reader.cc
} } bool CheckpointReader::HasTensor(const string& name) const { if (reader_ != nullptr) { return reader_->HasTensor(name, nullptr, nullptr); } return v2_reader_->Contains(name); } const TensorSliceReader::VarToShapeMap& CheckpointReader::GetVariableToShapeMap() const { CHECK(var_to_shape_map_); return *var_to_shape_map_; } const TensorSliceReader::VarToDataTypeMap&
C++ - Registered: Tue Apr 30 12:39:09 GMT 2024 - Last Modified: Fri Aug 25 21:29:12 GMT 2023 - 5.5K bytes - Viewed (0) -
tensorflow/c/eager/c_api_experimental_test.cc
/*isolate_session_state=*/false, init_timeout_in_ms); // Remote device on `worker2`. const char remote_device[] = "/job:localhost/replica:0/task:2/device:CPU:0"; // `ctx_0`, `ctx_1` contains `remote_device`. { const std::vector<std::string>& device_names = ListDeviceNames(ctx_0); ASSERT_TRUE(std::find(device_names.begin(), device_names.end(),
C++ - Registered: Tue Apr 30 12:39:09 GMT 2024 - Last Modified: Thu Aug 03 03:14:26 GMT 2023 - 31.5K bytes - Viewed (1) -
tensorflow/c/eager/c_api_test.cc
/*init_timeout_in_ms=*/0); // Remote device on `worker1`. const char remote_device[] = "/job:localhost/replica:0/task:1/device:CPU:0"; // `ctx_0`, `ctx_1`, `ctx_2` contains `remote_device`. { const std::vector<std::string>& device_names = ListDeviceNames(ctx_0); ASSERT_TRUE(std::find(device_names.begin(), device_names.end(),
C++ - Registered: Tue Apr 30 12:39:09 GMT 2024 - Last Modified: Thu Aug 03 20:50:20 GMT 2023 - 94.6K bytes - Viewed (1)