Search Options

Results per page
Sort
Preferred Languages
Advance

Results 1 - 10 of 14 for define (0.2 sec)

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

      std::vector<std::string> devices{
          "/job:localhost/replica:0/task:0/device:CPU:0",
          "/job:localhost/replica:0/task:0/device:CPU:1"};
      ParallelDevice parallel_device(devices);
      const FunctionDef assert_and_collective = FunctionDefHelper::Define(
          // Name
          "AssertAndCollective",
          // Args
          {"x: float", "condition: bool"},
          // Return values
    C++
    - Registered: Tue Apr 23 12:39:09 GMT 2024
    - Last Modified: Thu Jul 08 23:47:35 GMT 2021
    - 15.3K bytes
    - Viewed (0)
  2. tensorflow/c/c_api_experimental.cc

    TF_Library* TF_LoadPluggableDeviceLibrary(const char* library_filename,
                                              TF_Status* status) {
    #if defined(IS_MOBILE_PLATFORM) || defined(IS_SLIM_BUILD)
      status->status = tensorflow::errors::Unimplemented(
          "PluggableDevice plugin functionality is not supported on mobile");
      return nullptr;
    #else
      TF_Library* lib_handle = new TF_Library;
      static tensorflow::mutex mu(tensorflow::LINKER_INITIALIZED);
    C++
    - Registered: Tue Apr 23 12:39:09 GMT 2024
    - Last Modified: Mon Apr 15 03:35:10 GMT 2024
    - 29.4K bytes
    - Viewed (0)
  3. tensorflow/c/experimental/grappler/grappler.cc

      TF_RETURN_IF_ERROR(ValidateTPOptimizerConfigs(optimizer_configs));
    
      CGraphOptimizerRegister(
          [=]() { return new CGraphOptimizer(optimizer, params.device_type); },
          optimizer_configs, params.device_type);
    
      return absl::OkStatus();
    }
    
    }  // namespace grappler
    }  // namespace tensorflow
    
    void TF_GetNodesToPreserveListSize(const TF_GrapplerItem* item, int* num_values,
    C++
    - Registered: Tue Feb 27 12:39:08 GMT 2024
    - Last Modified: Wed Sep 06 19:12:29 GMT 2023
    - 15K bytes
    - Viewed (1)
  4. tensorflow/c/eager/custom_device_test.cc

    // A simple logging device to test custom device registration.
    #include <memory>
    
    #include "absl/strings/match.h"
    #include "tensorflow/c/c_api.h"
    #include "tensorflow/c/eager/c_api.h"
    #include "tensorflow/c/eager/c_api_experimental.h"
    #include "tensorflow/c/eager/c_api_test_util.h"
    #include "tensorflow/c/eager/custom_device_testutil.h"
    #include "tensorflow/c/tf_status.h"
    C++
    - Registered: Tue Apr 23 12:39:09 GMT 2024
    - Last Modified: Thu Aug 27 23:39:24 GMT 2020
    - 18.4K bytes
    - Viewed (0)
  5. tensorflow/c/experimental/filesystem/plugins/gcs/gcs_filesystem_test.cc

    #include "tensorflow/c/tf_status_helper.h"
    #include "tensorflow/core/platform/path.h"
    #include "tensorflow/core/platform/stacktrace_handler.h"
    #include "tensorflow/core/platform/test.h"
    
    #define ASSERT_TF_OK(x) ASSERT_EQ(TF_OK, TF_GetCode(x)) << TF_Message(x)
    #define EXPECT_TF_OK(x) EXPECT_EQ(TF_OK, TF_GetCode(x)) << TF_Message(x)
    
    static const char* content = "abcdefghijklmnopqrstuvwxyz1234567890";
    // We will work with content_view instead of content.
    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)
  6. tensorflow/c/eager/gradients.cc

    namespace internal {
    Status Reset(AbstractOperation* op_, const char* op,
                 const char* raw_device_name, ForwardOperation* forward_op_) {
      forward_op_->op_name = op;
      forward_op_->attrs.Reset(op);
      return op_->Reset(op, raw_device_name);
    }
    Status AddInput(AbstractOperation* op_, AbstractTensorHandle* input,
                    ForwardOperation* forward_op_) {
    C++
    - Registered: Tue Apr 23 12:39:09 GMT 2024
    - Last Modified: Thu Feb 15 09:49:45 GMT 2024
    - 19.3K bytes
    - Viewed (0)
  7. tensorflow/c/eager/c_api_distributed_test.cc

      const string composite_device_name =
          "/job:localhost/replica:0/task:0/device:COMPOSITE:0";
      EXPECT_EQ(TFE_TensorHandleDeviceName(packed_handle, status),
                composite_device_name);
      EXPECT_EQ(TFE_TensorHandleBackingDeviceName(packed_handle, status),
                composite_device_name);
      ASSERT_EQ(TF_GetCode(status), TF_OK) << TF_Message(status);
    
    C++
    - Registered: Tue Apr 23 12:39:09 GMT 2024
    - Last Modified: Thu Feb 15 09:49:45 GMT 2024
    - 23.5K bytes
    - Viewed (0)
  8. tensorflow/c/eager/parallel_device/parallel_device.cc

                                const char* const* underlying_devices,
                                int num_underlying_devices,
                                TFE_CustomDevice* device, void** device_info) {
      device->copy_tensor_to_device = &CopyToParallelDevice;
      device->copy_tensor_from_device = &CopyTensorFromParallelDevice;
      device->delete_device = &DeleteParallelDevice;
      device->execute = &ParallelDeviceExecute;
    C++
    - Registered: Tue Apr 23 12:39:09 GMT 2024
    - Last Modified: Wed Mar 29 22:05:31 GMT 2023
    - 18.3K bytes
    - Viewed (0)
  9. tensorflow/c/eager/parallel_device/parallel_device_lib.cc

                                      absl::optional<int64_t> step_id) const {
      for (int device_index = 0; device_index < underlying_devices_.size();
           ++device_index) {
        DeviceThread* device_thread = device_threads_[device_index].get();
        std::vector<TFE_TensorHandle*> device_inputs;
        device_inputs.reserve(inputs.size());
        for (int input_index = 0; input_index < inputs.size(); ++input_index) {
    C++
    - Registered: Tue Apr 23 12:39:09 GMT 2024
    - Last Modified: Fri Feb 09 07:47:20 GMT 2024
    - 25.4K bytes
    - Viewed (1)
  10. tensorflow/c/eager/parallel_device/parallel_device_test.cc

      const char* device_name = "/job:localhost/replica:0/task:0/device:CUSTOM:0";
      const char* first_device_name =
          "/job:localhost/replica:0/task:0/device:CPU:0";
      const char* second_device_name =
          "/job:localhost/replica:0/task:0/device:CPU:1";
      std::array<const char*, 2> underlying_devices{first_device_name,
                                                    second_device_name};
    C++
    - Registered: Tue Apr 23 12:39:09 GMT 2024
    - Last Modified: Thu Jul 08 23:47:35 GMT 2021
    - 29.3K bytes
    - Viewed (1)
Back to top