Search Options

Results per page
Sort
Preferred Languages
Advance

Results 1 - 4 of 4 for MarkForCompilationPass (2.31 sec)

  1. tensorflow/compiler/jit/tests/auto_clustering_test_helper.cc

      TF_RETURN_IF_ERROR(runner.AddCpus(32));
      TF_RETURN_IF_ERROR(runner.AddGpus(8));
    
      for (absl::string_view auto_clustering_pass :
           {"CloneConstantsForBetterClusteringPass", "MarkForCompilationPass",
            "IncreaseDynamismForAutoJitPass", "PartiallyDeclusterPass"}) {
        GraphDef next;
        TF_RETURN_IF_ERROR(
            runner.Run(auto_clustering_pass, std::move(graphdef), &next));
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu Feb 22 12:11:46 UTC 2024
    - 8.5K bytes
    - Viewed (0)
  2. tensorflow/compiler/jit/tests/auto_clustering_test_helper.h

                                       absl::string_view golden_summary_file_path);
    };
    
    #if defined(PLATFORM_GOOGLE)
    // Reads the GraphDef stored in graph_def_path (which must be a pbtxt file) and
    // benchmarks MarkForCompilationPass on this graphdef.
    Status BenchmarkMarkForCompilation(absl::string_view graph_def_path,
                                       benchmark::State& state);
    #endif  // PLATFORM_GOOGLE
    
    }  // namespace tensorflow
    
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Wed Sep 06 19:12:29 UTC 2023
    - 3.1K bytes
    - Viewed (0)
  3. tensorflow/compiler/jit/device_util.h

    // all operations in the cluster that XLA:CPU can compile.
    //
    // We provide the `allow_mixing_unknown_and_cpu` knob so that we can do both of
    // the following things:
    //
    // - Let MarkForCompilationPass not inject CPU-placed operations into clusters
    //   that will run on unknown devices (because the unknown XLA backend may not
    //   support every operation supported by CPU).
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Wed May 15 17:18:31 UTC 2024
    - 7.1K bytes
    - Viewed (0)
  4. tensorflow/compiler/jit/mark_for_compilation_pass.cc

        std::atomic<int64_t>* fuel = new std::atomic<int64_t>;
        *fuel = initial_value;
        return fuel;
      }();
    
      return fuel;
    }
    
    }  // anonymous namespace
    
    Status MarkForCompilationPass::Run(
        const GraphOptimizationPassOptions& options) {
      MarkForCompilationPassFlags* flags = GetMarkForCompilationPassFlags();
    
      MarkForCompilationPassImpl::DebugOptions debug_options;
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Wed Feb 21 12:19:41 UTC 2024
    - 85.3K bytes
    - Viewed (0)
Back to top