Search Options

Results per page
Sort
Preferred Languages
Advance

Results 1 - 6 of 6 for device_num (0.43 sec)

  1. tensorflow/compiler/jit/xla_device.cc

                                                     const string& device_name,
                                                     int device_ordinal) {
      return Device::BuildDeviceAttributes(
          absl::StrCat(name_prefix, "/device:", device_name, ":", device_ordinal),
          DeviceType(device_name), Bytes(16ULL << 30), DeviceLocality(),
          absl::StrCat("device: ", device_name, " device"));
    }
    
    }  // namespace
    
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Mon May 20 21:05:42 UTC 2024
    - 24.3K bytes
    - Viewed (0)
  2. tensorflow/c/experimental/stream_executor/stream_executor.cc

        OwnedTFStatus c_status(TF_NewStatus());
        SP_Stream stream_handle = static_cast<CStream*>(stream)->Handle();
        SP_DeviceMemoryBase device_mem = DeviceMemoryBaseToC(location);
        stream_executor_->mem_zero(&device_, stream_handle, &device_mem, size,
                                   c_status.get());
        return StatusFromTF_Status(c_status.get());
      }
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Fri Jun 14 07:39:19 UTC 2024
    - 27.1K bytes
    - Viewed (0)
  3. tensorflow/c/kernels.cc

    }
    
    TF_StringView TF_GetDeviceName(TF_OpKernelContext* ctx) {
      const auto& device_name =
          reinterpret_cast<tensorflow::OpKernelContext*>(ctx)->device()->name();
      TF_StringView device_name_sv;
      device_name_sv.data = device_name.data();
      device_name_sv.len = device_name.length();
      return device_name_sv;
    }
    
    #if !defined(IS_MOBILE_PLATFORM) && !defined(IS_SLIM_BUILD)
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Tue May 28 22:53:47 UTC 2024
    - 36K bytes
    - Viewed (0)
  4. tensorflow/compiler/jit/xla_tpu_device.cc

        return absl::OkStatus();
      }
    
      int device_count = platform->VisibleDeviceCount();
    
      for (int i = 0; i < device_count; ++i) {
        const string device_name = absl::StrCat("/physical_device:TPU:", i);
        devices->push_back(device_name);
      }
    
      return absl::OkStatus();
    }
    
    Status TpuNodeDeviceFactory::CreateDevices(
        const SessionOptions& session_options, const string& name_prefix,
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Tue May 28 22:53:47 UTC 2024
    - 20.9K bytes
    - Viewed (0)
  5. tensorflow/compiler/mlir/lite/python/converter_python_api.cc

        const char* node_name = opdef.name().c_str();
        const char* op_name = opdef.name().c_str();
        const char* device_name = "CPU";
        static auto fake_compute_func = [](void* kernel, TF_OpKernelContext* ctx) {
        };
    
        TF_KernelBuilder* builder =
            TF_NewKernelBuilder(op_name, device_name, /*create_func=*/nullptr,
                                fake_compute_func, /*delete_func=*/nullptr);
    
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Wed Jun 12 23:15:24 UTC 2024
    - 19.2K bytes
    - Viewed (0)
  6. tensorflow/compiler/jit/kernels/xla_ops.cc

      return launch_context;
    }
    
    Status GetTaskName(const std::string_view device_name, std::string* task_name) {
      string ignored;
      if (!DeviceNameUtils::SplitDeviceName(device_name, task_name, &ignored)) {
        return errors::InvalidArgument("Unable to parse device name: ",
                                       device_name);
      }
    
      return absl::OkStatus();
    }
    
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Fri May 17 22:46:36 UTC 2024
    - 41.4K bytes
    - Viewed (0)
Back to top