Search Options

Results per page
Sort
Preferred Languages
Advance

Results 1 - 1 of 1 for add_enable_op (0.16 sec)

  1. tensorflow/c/eager/c_api_test_util.cc

          config->mutable_graph_options()->mutable_rewrite_options();
      rewrite_options->set_scoped_allocator_optimization(
          tensorflow::RewriterConfig::ON);
      rewrite_options->mutable_scoped_allocator_opts()->add_enable_op(
          "CollectiveReduce");
    
      if ((tensorflow::IsGoogleCudaEnabled() || tensorflow::IsBuiltWithROCm()) &&
          num_virtual_gpus > 0) {
        tensorflow::GPUOptions* gpu_options =
    Registered: Tue Nov 05 12:39:12 UTC 2024
    - Last Modified: Wed Feb 21 22:37:46 UTC 2024
    - 23.5K bytes
    - Viewed (0)
Back to top