Search Options

Results per page
Sort
Preferred Languages
Advance

Results 1 - 10 of 29 for cpus (0.15 sec)

  1. tensorflow/c/eager/c_api_experimental.cc

    void TFE_SetLogicalCpuDevices(TFE_Context* ctx, int num_cpus,
                                  const char* prefix, TF_Status* status) {
      std::vector<std::unique_ptr<tensorflow::Device>> devices;
    
      if (prefix == nullptr || strlen(prefix) == 0)
        prefix = "/job:localhost/replica:0/task:0";
    
      tensorflow::SessionOptions sess_options;
      (*sess_options.config.mutable_device_count())["CPU"] = num_cpus;
      status->status =
    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)
  2. .bazelrc

    build:android_arm64 --config=android
    build:android_arm64 --cpu=arm64-v8a
    build:android_arm64 --fat_apk_cpu=arm64-v8a
    build:android_x86 --config=android
    build:android_x86 --cpu=x86
    build:android_x86 --fat_apk_cpu=x86
    build:android_x86_64 --config=android
    build:android_x86_64 --cpu=x86_64
    build:android_x86_64 --fat_apk_cpu=x86_64
    
    # Build everything statically for Android since all static libs are later
    Plain Text
    - Registered: Tue May 07 12:40:20 GMT 2024
    - Last Modified: Thu May 02 19:34:20 GMT 2024
    - 52.8K bytes
    - Viewed (2)
  3. tensorflow/c/eager/c_api_experimental.h

    // through TFE_ContextOptions, this method should not be called.
    TF_CAPI_EXPORT extern void TFE_SetLogicalCpuDevices(TFE_Context* ctx,
                                                        int num_cpus,
                                                        const char* prefix,
                                                        TF_Status* status);
    
    // Set configuration key and value using coordination service.
    C
    - Registered: Tue Apr 30 12:39:09 GMT 2024
    - Last Modified: Wed Feb 21 22:37:46 GMT 2024
    - 39.5K bytes
    - Viewed (0)
  4. .github/workflows/arm-ci.yml

              ./tensorflow/tools/ci_build/ci_build.sh cpu.arm64 bash tensorflow/tools/ci_build/rel/ubuntu/cpu_arm64_test.sh...
    Others
    - Registered: Tue May 07 12:40:20 GMT 2024
    - Last Modified: Wed Feb 07 17:41:21 GMT 2024
    - 2.2K bytes
    - Viewed (0)
  5. ci/official/envs/linux_x86_cuda

    TFCI_BAZEL_COMMON_ARGS="--repo_env=TF_PYTHON_VERSION=$TFCI_PYTHON_VERSION --config release_gpu_linux"
    TFCI_BAZEL_TARGET_SELECTING_CONFIG_PREFIX=linux_cuda
    TFCI_BUILD_PIP_PACKAGE_ARGS="--repo_env=WHEEL_NAME=tensorflow"
    TFCI_DOCKER_ARGS="--gpus all"
    TFCI_LIB_SUFFIX="-gpu-linux-x86_64"
    Plain Text
    - Registered: Tue Apr 30 12:39:09 GMT 2024
    - Last Modified: Fri Jan 19 00:24:30 GMT 2024
    - 1K bytes
    - Viewed (0)
  6. tensorflow/c/c_api_test.cc

      TF_DeleteStatus(s);
    }
    
    void Deallocator(void* data, size_t, void* arg) {
      tensorflow::cpu_allocator()->DeallocateRaw(data);
      *reinterpret_cast<bool*>(arg) = true;
    }
    
    TEST(CAPI, Tensor) {
      const int num_bytes = 6 * sizeof(float);
      float* values =
          reinterpret_cast<float*>(tensorflow::cpu_allocator()->AllocateRaw(
              TF_TensorDefaultAlignment(), num_bytes));
      int64_t dims[] = {2, 3};
    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)
  7. RELEASE.md

            `.predict` is available for Cloud TPUs, Cloud TPU, for all types of
            Keras models (sequential, functional and subclassing models).
        *   Automatic outside compilation is now enabled for Cloud TPUs. This allows
            `tf.summary` to be used more conveniently with Cloud TPUs.
        *   Dynamic batch sizes with DistributionStrategy and Keras are supported on
            Cloud TPUs.
    Plain Text
    - Registered: Tue May 07 12:40:20 GMT 2024
    - Last Modified: Mon Apr 29 19:17:57 GMT 2024
    - 727.7K bytes
    - Viewed (8)
  8. .github/workflows/arm-cd.yml

          - name: Build and test pip wheel
            shell: bash
            run: |
              is_nightly=0 && tf_project_name='tensorflow_cpu_aws' && ${{ github.event_name == 'schedule' }} && is_nightly=1 && tf_project_name='tf_nightly_cpu_aws'
              echo "PyPI project name:" $tf_project_name
    Others
    - Registered: Tue May 07 12:40:20 GMT 2024
    - Last Modified: Tue Mar 05 10:24:16 GMT 2024
    - 3K bytes
    - Viewed (1)
  9. ci/official/envs/linux_arm64

    TFCI_BAZEL_COMMON_ARGS="--repo_env=TF_PYTHON_VERSION=$TFCI_PYTHON_VERSION --config release_arm64_linux"
    TFCI_BAZEL_TARGET_SELECTING_CONFIG_PREFIX=linux_arm64
    # Note: this is not set to "--cpu", because that changes the package name
    # to tensorflow_cpu. These ARM builds are supposed to have the name "tensorflow"
    # despite lacking Nvidia CUDA support.
    TFCI_BUILD_PIP_PACKAGE_ARGS="--repo_env=WHEEL_NAME=tensorflow"
    TFCI_DOCKER_ENABLE=1
    Plain Text
    - Registered: Tue Apr 30 12:39:09 GMT 2024
    - Last Modified: Thu Feb 15 23:12:40 GMT 2024
    - 1.5K bytes
    - Viewed (1)
  10. ci/official/README.md

    -   Different Python versions
    -   Linux, MacOS, and Windows machines (these pool definitions are internal)
    -   x86 and arm64
    -   CPU-only, or with NVIDIA CUDA support (Linux only), or with TPUs
    
    ## How to Test Your Changes to TensorFlow
    
    You may check how your changes will affect TensorFlow by:
    
    1. Creating a PR and observing the presubmit test results
    Plain Text
    - Registered: Tue May 07 12:40:20 GMT 2024
    - Last Modified: Thu Feb 01 03:21:19 GMT 2024
    - 8K bytes
    - Viewed (0)
Back to top