Search Options

Results per page
Sort
Preferred Languages
Advance

Results 21 - 23 of 23 for GraphOptimizationPassOptions (0.43 sec)

  1. tensorflow/compiler/jit/mark_for_compilation_pass.cc

      should_compile_cluster_cache_.insert({&cluster, should_compile});
      return should_compile;
    }
    
    Status MarkForCompilation(
        const GraphOptimizationPassOptions& options,
        const MarkForCompilationPassImpl::DebugOptions& debug_options) {
      Graph* graph = options.graph->get();
      FunctionLibraryDefinition* flib_def = options.flib_def;
    
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Wed Feb 21 12:19:41 UTC 2024
    - 85.3K bytes
    - Viewed (0)
  2. tensorflow/compiler/jit/encapsulate_subgraphs_pass_test.cc

      CreateSubgraphTouchingRefVar(root);
    
      auto graph = std::make_unique<Graph>(OpRegistry::Global());
      TF_ASSERT_OK(root.ToGraph(graph.get()));
    
      GraphOptimizationPassWrapper wrapper;
      GraphOptimizationPassOptions options =
          wrapper.CreateGraphOptimizationPassOptions(&graph);
    
      EncapsulateSubgraphsPass pass;
      TF_ASSERT_OK(pass.Run(options));
    
      for (const Node* node : graph->nodes()) {
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu Feb 22 08:47:20 UTC 2024
    - 113.3K bytes
    - Viewed (0)
  3. tensorflow/compiler/jit/encapsulate_subgraphs_pass.cc

          }
          n->AddAttr("index", permutation[index]);
        }
      }
      return absl::OkStatus();
    }
    
    Status EncapsulateSubgraphsPass::Run(
        const GraphOptimizationPassOptions& options) {
      VLOG(1) << "EncapsulateSubgraphsPass::Run";
      if (VLOG_IS_ON(1)) {
        DumpGraphToFile("encapsulate_subgraphs_before", **options.graph,
                        options.flib_def);
      }
    
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu Feb 22 08:47:20 UTC 2024
    - 51K bytes
    - Viewed (0)
Back to top