- Sort Score
- Result 10 results
- Languages All
Results 1 - 10 of 40 for Nest (0.15 sec)
-
tensorflow/c/eager/dlpack_test.cc
#include "tensorflow/c/eager/dlpack.h" #include <vector> #include "absl/strings/str_join.h" #include "include/dlpack/dlpack.h" // from @dlpack #include "tensorflow/c/eager/c_api.h" #include "tensorflow/core/platform/test.h" namespace tensorflow { namespace { void TestHandleFromDLPack(TF_Status* status, TFE_Context* ctx, std::vector<int64_t> shape, std::vector<int64_t> strides) {
C++ - Registered: Tue Apr 30 12:39:09 GMT 2024 - Last Modified: Fri Jun 30 03:04:46 GMT 2023 - 4.4K bytes - Viewed (0) -
tensorflow/c/eager/c_api_debug_test.cc
==============================================================================*/ #include "tensorflow/c/eager/c_api.h" #include <string.h> #include "tensorflow/c/eager/c_api_test_util.h" #include "tensorflow/core/platform/logging.h" #include "tensorflow/core/platform/test.h" TEST(CApiDebug, ScalarCPU) { TF_Status* status = TF_NewStatus(); TFE_ContextOptions* opts = TFE_NewContextOptions(); TFE_Context* ctx = TFE_NewContext(opts, status);
C++ - Registered: Tue Apr 30 12:39:09 GMT 2024 - Last Modified: Mon Apr 06 22:10:09 GMT 2020 - 2.3K bytes - Viewed (0) -
tensorflow/c/eager/parallel_device/parallel_device_remote_test.cc
job_def->mutable_tasks()->insert( {i, tensorflow::strings::StrCat("localhost", ":", port)}); LOG(INFO) << "Picked test port: " << port << " for job: " << job_name << ", task: " << i; } return server_def; } namespace tensorflow { namespace parallel_device { TEST(PARALLEL_DEVICE, TestRemoteBasic) { std::unique_ptr<TFE_ContextOptions, decltype(&TFE_DeleteContextOptions)> opts(
C++ - Registered: Tue Apr 30 12:39:09 GMT 2024 - Last Modified: Thu Apr 27 22:09:57 GMT 2023 - 6.7K bytes - Viewed (0) -
tensorflow/c/experimental/gradients/nn_grad.cc
return absl::OkStatus(); } ~SparseSoftmaxCrossEntropyWithLogitsGradientFunction() override {} private: vector<AbstractTensorHandle*> forward_outputs_; }; // TODO(vnvo2409): Add python test class BiasAddGradientFunction : public GradientFunction { public: explicit BiasAddGradientFunction(AttrBuilder f_attrs) : forward_attrs_(f_attrs) {} Status Compute(AbstractContext* ctx,
C++ - Registered: Tue Mar 26 12:39:09 GMT 2024 - Last Modified: Wed Feb 28 13:53:47 GMT 2024 - 5.7K bytes - Viewed (0) -
tensorflow/c/eager/parallel_device/parallel_device_test.cc
ExpectScalarEq<float>(result_components[1].get(), 3.); } TEST(PARALLEL_DEVICE, TestCollectiveSync) { TestCollective(/*async=*/false); } // Note that ops on the parallel device currently don't execute // asynchronously. The test is just that we don't get deadlocks. TEST(PARALLEL_DEVICE, TestCollectiveAsync) { TestCollective(/*async=*/true); }
C++ - Registered: Tue Apr 30 12:39:09 GMT 2024 - Last Modified: Thu Jul 08 23:47:35 GMT 2021 - 29.3K bytes - Viewed (1) -
tensorflow/c/experimental/grappler/grappler_test.cc
#include "tensorflow/core/grappler/costs/graph_properties.h" #include "tensorflow/core/grappler/grappler_item.h" #include "tensorflow/core/grappler/inputs/trivial_test_graph_input_yielder.h" #include "tensorflow/core/lib/core/status_test_util.h" #include "tensorflow/core/platform/test.h" namespace tensorflow { namespace grappler { namespace { void optimize_func(void* optimizer, const TF_Buffer* graph_buf,
C++ - Registered: Tue Feb 27 12:39:08 GMT 2024 - Last Modified: Thu Apr 13 22:30:58 GMT 2023 - 11.6K bytes - Viewed (0) -
tensorflow/c/eager/gradients_test.cc
"or NotDifferentiableGradientFunction.", s.message()); ASSERT_EQ(nullptr, outputs[0]); } // TODO(b/164171226): Enable this test with tfrt after AddInputList is // supported. It is needed for IdentityN. #ifdef PLATFORM_GOOGLE INSTANTIATE_TEST_SUITE_P( UnifiedCAPI, CppGradients, ::testing::Combine(::testing::Values("graphdef", "mlir"), /*tfrt*/ ::testing::Values(false),
C++ - Registered: Tue Apr 30 12:39:09 GMT 2024 - Last Modified: Thu Feb 15 09:49:45 GMT 2024 - 7K bytes - Viewed (0) -
tensorflow/c/experimental/filesystem/plugins/gcs/gcs_filesystem_test.cc
namespace gcs = google::cloud::storage; static std::string InitializeTmpDir() { // This env should be something like `gs://bucket/path` const char* test_dir = getenv("GCS_TEST_TMPDIR"); if (test_dir != nullptr) { std::string bucket, object; TF_Status* status = TF_NewStatus(); ParseGCSPath(test_dir, true, &bucket, &object, status); if (TF_GetCode(status) != TF_OK) { TF_DeleteStatus(status); return ""; }
C++ - Registered: Tue Apr 23 12:39:09 GMT 2024 - Last Modified: Mon Aug 31 12:04:23 GMT 2020 - 24.9K bytes - Viewed (0) -
tensorflow/c/eager/c_api_distributed_test.cc
// After the graph pass is registered, it takes effect globally and can affect // other test cases. Define a static variable to switch it on and off. bool GraphErrorInjectionPass::enabled_ = false; // Test to ensure that a registered graph optimization pass is only executed // once (i.e., on the main function side) in running distributed functions. // This test creates a cluster with two workers, create a variable on the
C++ - Registered: Tue Apr 30 12:39:09 GMT 2024 - Last Modified: Thu Feb 15 09:49:45 GMT 2024 - 23.5K bytes - Viewed (0) -
tensorflow/c/experimental/next_pluggable_device/tensor_pjrt_buffer_util_test.cc
} TEST(TensorPjRtBufferUtilTest, GetPjRtCApiClientNotFound) { EXPECT_THAT( GetPjRtCApiClient(tensorflow::DeviceType(DEVICE_CPU)), StatusIs(error::NOT_FOUND, HasSubstr(absl::StrCat("PjRt client not found for device type ", DEVICE_CPU)))); } TEST(TensorPjRtBufferUtilTest, GetPjRtCApiClientIncorrectType) {
C++ - Registered: Tue Feb 27 12:39:08 GMT 2024 - Last Modified: Mon Oct 30 19:20:20 GMT 2023 - 7.2K bytes - Viewed (0)