Search Options

Results per page
Sort
Preferred Languages
Advance

Results 11 - 20 of 22 for shield (0.66 sec)

  1. tensorflow/c/eager/parallel_device/parallel_device_lib.cc

    //
    // DeviceThread itself is thread-safe, in that StartExecute will block if there
    // is a pending execution. Since StartExecute is equivalent to grabbing a lock,
    // multiple DeviceThreads should always be accessed in the same order to avoid
    // deadlocks.
    class DeviceThread {
     public:
      // Starts a background thread waiting for `StartExecute`.
      explicit DeviceThread(const std::string& device, const bool is_async,
    C++
    - Registered: Tue Apr 30 12:39:09 GMT 2024
    - Last Modified: Fri Feb 09 07:47:20 GMT 2024
    - 25.4K bytes
    - Viewed (1)
  2. tensorflow/c/eager/parallel_device/parallel_device_test.cc

      ASSERT_EQ(TF_GetCode(status.get()), TF_OK) << TF_Message(status.get());
    
      // Un-pack the parallel tensor to verify that the operation was
      // successful. The resulting structure should be:
      //   second_device{first_device{1. * 3., 2. * 3.}, 3. * 3.}.
      std::array<TensorHandlePtr, 2> second_components;
      ExtractPerDeviceValues(context.get(), multiply_result.get(),
    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)
  3. tensorflow/c/eager/gradient_checker_test.cc

                    absl::Span<AbstractTensorHandle*> outputs) {
      return ops::Mul(ctx, inputs[0], inputs[1], &outputs[0], "Mul");
    }
    
    // TODO(vnvo2409): Add more tests from `python/ops/gradient_checker_v2_test.py`.
    // These tests should not be confused with `[*]_grad_test` which compare the
    // result of `gradient_checker` and `[*]_grad`. The tests here test the
    // functionality of `gradient_checker` by comparing the result with expected
    C++
    - Registered: Tue Apr 30 12:39:09 GMT 2024
    - Last Modified: Fri Apr 14 10:03:59 GMT 2023
    - 6.5K bytes
    - Viewed (0)
  4. tensorflow/c/eager/c_api.cc

    #endif  // !IS_MOBILE_PLATFORM
      return tensorflow::wrap(eager_context);
    }
    
    void TFE_DeleteContext(TFE_Context* ctx) {
      if (ctx == nullptr) {
        return;
      }
    
      // ctx->RefCountIsOne() should be true here.
      tensorflow::unwrap(ctx)->Release();
    }
    
    TF_DeviceList* TFE_ContextListDevices(TFE_Context* ctx, TF_Status* status) {
      TF_DeviceList* l = new TF_DeviceList;
    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)
  5. tensorflow/c/eager/c_api_experimental.cc

            tensorflow::unwrap(op_to_reset);
        op->Clear();
        status->status = op->Reset(op_or_function_name, raw_device_name);
      } else {
        TF_SetStatus(status, TF_INVALID_ARGUMENT,
                     "op_to_reset should not be nullptr");
      }
    }
    
    void TFE_ContextEnableGraphCollection(TFE_Context* ctx) {
      tensorflow::unwrap(ctx)->SetShouldStoreGraphs(true);
    }
    
    void TFE_ContextDisableGraphCollection(TFE_Context* ctx) {
    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)
  6. tensorflow/c/experimental/filesystem/plugins/gcs/gcs_filesystem.cc

      TF_SetStatusFromGCSStatus(metadata.status(), status);
      if (TF_GetCode(status) == TF_FAILED_PRECONDITION)
        TF_SetStatus(status, TF_ALREADY_EXISTS, path);
    }
    
    // TODO(vnvo2409): `RecursivelyCreateDir` should use `CreateDir` instead of the
    // default implementation. Because we could create an empty object whose
    // key is equal to the `path` and Google Cloud Console will automatically
    // display it as a directory tree.
    
    C++
    - Registered: Tue Apr 30 12:39:09 GMT 2024
    - Last Modified: Wed Aug 23 06:55:53 GMT 2023
    - 46.9K bytes
    - Viewed (0)
  7. tensorflow/c/experimental/filesystem/plugins/gcs/ram_file_block_cache_test.cc

                                                 fetcher);
      std::vector<char> out;
      // Read the second block; this should yield an OK status and a single byte.
      TF_EXPECT_OK(ReadCache(&cache, "", block_size, block_size, &out));
      EXPECT_EQ(out.size(), 1);
      // Now read the first block; this should yield an INTERNAL error because we
      // had already cached a partial block at a later position.
    C++
    - Registered: Tue Apr 23 12:39:09 GMT 2024
    - Last Modified: Fri Oct 15 03:16:57 GMT 2021
    - 23.2K bytes
    - Viewed (0)
  8. tensorflow/c/eager/c_api_experimental_test.cc

        ASSERT_EQ(TF_OK, TF_GetCode(status)) << TF_Message(status);
        TF_DeleteStatus(status);
      }
    
      // 6. Read `var` using `ctx_1`. This read should succeed since `ctx_1` was
      // created with `isolate_session_state` set to false, and update should
      // preserve it.
      {
        // Create a handle to `var`, using `ctx_1`.
        TFE_TensorHandle* var_handle =
    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)
  9. tensorflow/c/eager/c_api_test.cc

        ASSERT_EQ(TF_OK, TF_GetCode(status)) << TF_Message(status);
        TF_DeleteStatus(status);
      }
    
      // 6. Read `var` using `ctx_1`. This read should succeed since `ctx_1` was
      // created with `isolate_session_state` set to false, and update should
      // preserve it.
      {
        // Create a handle to `var`, using `ctx_1`.
        TFE_TensorHandle* var_handle =
    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)
  10. tensorflow/c/experimental/filesystem/modular_filesystem_test.cc

                                                          "a_dir", "another_dir"};
      EXPECT_EQ(children.size(), filenames.size() + dirnames.size());
      for (const auto& child : expected_children)
        EXPECT_NE(std::find(children.begin(), children.end(), child),
                  children.end());
    }
    
    TEST_P(ModularFileSystemTest, TestGetChildrenEmpty) {
      const std::string dirpath = GetURIForPath("dir");
    C++
    - Registered: Tue Apr 30 12:39:09 GMT 2024
    - Last Modified: Fri May 27 20:25:58 GMT 2022
    - 71K bytes
    - Viewed (0)
Back to top