- Sort Score
- Result 10 results
- Languages All
Results 1 - 10 of 23 for GraphOptimizationPassOptions (2.83 sec)
-
tensorflow/compiler/jit/test_util.h
// A helper object to create GraphOptimizationPassOptions. struct GraphOptimizationPassWrapper { explicit GraphOptimizationPassWrapper() : library(OpRegistry::Global(), FunctionDefLibrary()) { session_options.env = Env::Default(); } // Create GraphOptimizationPassOptions with a graph passed in constructor and // sensible options. GraphOptimizationPassOptions CreateGraphOptimizationPassOptions(
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Mon Oct 16 18:03:15 UTC 2023 - 3.1K bytes - Viewed (0) -
tensorflow/compiler/jit/xla_cluster_util.h
bool HasResourceInputOrOutput(const Node& node); // Determines the global jit level based on GraphOptimizationPassOptions, // --tf_xla_auto_jit and whether the graph is a single GPU graph. OptimizerOptions::GlobalJitLevel GetGlobalJitLevelForGraph( const GraphOptimizationPassOptions& options); // Returns true if `g` is a single-GPU graph. A single-GPU graph uses exactly // one GPU (and any number of CPUs).
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed Feb 21 09:53:30 UTC 2024 - 4.6K bytes - Viewed (0) -
tensorflow/compiler/jit/force_xla_constants_on_host_pass_test.cc
FunctionLibraryDefinition* flib_def, std::unique_ptr<Graph>* result) { auto graph = std::make_unique<Graph>(OpRegistry::Global()); GraphOptimizationPassOptions options; SessionOptions session_options; session_options.env = Env::Default(); options.graph = &graph; options.session_options = &session_options; options.flib_def = flib_def;
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Feb 22 08:47:20 UTC 2024 - 4.1K bytes - Viewed (0) -
tensorflow/compiler/jit/force_xla_constants_on_host_pass.cc
#include "tensorflow/compiler/jit/defs.h" #include "tensorflow/core/common_runtime/optimization_registry.h" namespace tensorflow { Status ForceXlaConstantsOnHostPass::Run( const GraphOptimizationPassOptions& options) { Graph* graph = options.graph->get(); OptimizerOptions opts; auto pflr = std::make_unique<ProcessFunctionLibraryRuntime>( nullptr, options.session_options->env, /*config=*/nullptr,
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Feb 22 08:47:20 UTC 2024 - 2.1K bytes - Viewed (0) -
tensorflow/compiler/mlir/mlir_graph_optimization_pass.h
const DeviceSet* device_set, const ConfigProto& config_proto, const Graph& graph, const FunctionLibraryDefinition& function_library) const = 0; virtual Status Run(const GraphOptimizationPassOptions& options, mlir::ModuleOp module) = 0; }; class MlirV1CompatOptimizationPassRegistry { public: // Returns the global registry of MLIR optimization passes.
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue Oct 03 22:53:34 UTC 2023 - 8.4K bytes - Viewed (0) -
tensorflow/compiler/jit/partially_decluster_pass.cc
if (options.flib_def == nullptr) { return errors::InvalidArgument( "GraphOptimizationPassOptions::flib_def must be set for " "PartiallyDeclusterPass."); } if (options.session_options == nullptr || options.session_options->env == nullptr) { return errors::InvalidArgument( "GraphOptimizationPassOptions::session_options::env must be set for " "PartiallyDeclusterPass."); }
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Fri Feb 09 11:36:41 UTC 2024 - 15.7K bytes - Viewed (0) -
tensorflow/compiler/jit/build_xla_ops_pass.cc
return absl::OkStatus(); } // Replaces `n` with a `PartitionedCall` op that calls the same function. absl::StatusOr<Node*> ReplaceFunctionCallWithPartitionedCall( const GraphOptimizationPassOptions& options, const FunctionLibraryDefinition& flib_def, Node* n, Graph* g, const NameAttrList& func, const Scope& root) { string config_string = options.session_options->config.SerializeAsString();
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue Mar 12 06:33:33 UTC 2024 - 24.3K bytes - Viewed (0) -
tensorflow/compiler/jit/encapsulate_xla_computations_pass.h
namespace tensorflow { // Encapsulates nodes marked with the _xla_compile_id attribute into // XlaLaunch operators. class EncapsulateXlaComputationsPass : public GraphOptimizationPass { public: Status Run(const GraphOptimizationPassOptions& options) override; // The following methods are public only for unit tests. // This pass has two stages: // a) first, we call EncapsulateSubgraphsPass to encapsulate all nodes
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Feb 22 06:59:07 UTC 2024 - 3.6K bytes - Viewed (0) -
tensorflow/compiler/mlir/mlir_graph_optimization_pass_test.cc
const Graph& graph, const FunctionLibraryDefinition& function_library), (const, override)); MOCK_METHOD(Status, Run, (const GraphOptimizationPassOptions& options, mlir::ModuleOp module), (override)); }; class ModifyMlirModulePass : public MlirOptimizationPass { public:
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue Feb 27 08:25:30 UTC 2024 - 16.1K bytes - Viewed (0) -
tensorflow/compiler/jit/clone_constants_for_better_clustering_test.cc
SessionOptions session_options; session_options.config.mutable_graph_options() ->mutable_optimizer_options() ->set_global_jit_level(OptimizerOptions::ON_2); GraphOptimizationPassOptions options; options.graph = &graph; options.session_options = &session_options; // Scope::ToGraph seems to drop assigned devices, probably because it goes
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Feb 22 08:47:20 UTC 2024 - 8.4K bytes - Viewed (0)