Search Options

Results per page
Sort
Preferred Languages
Advance

Results 1 - 4 of 4 for compile_time_const_nodes (0.27 sec)

  1. tensorflow/compiler/jit/partially_decluster_pass.cc

                                                &compile_time_const_nodes,
                                                lib_runtime, IsIntraClusterEdge));
    
      std::vector<Node*> rpo;
      GetReversePostOrder(*graph, &rpo, /*stable_comparator=*/NodeComparatorName(),
                          /*edge_filter=*/NotBackedge);
      for (Node* n : rpo) {
        if (!compile_time_const_nodes[n->id()]) {
          continue;
        }
    
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Fri Feb 09 11:36:41 UTC 2024
    - 15.7K bytes
    - Viewed (0)
  2. tensorflow/compiler/jit/compilability_check_util.cc

      std::vector<bool> const_args(arg_types.size());
      // If we can't analyze the const args. Bail out.
      TF_RETURN_IF_ERROR(
          BackwardsConstAnalysis(*((*fbody)->graph), &const_args,
                                 /*compile_time_const_nodes=*/nullptr, flr));
    
      for (size_t i = 0; i < const_args.size(); ++i) {
        if (const_args[i]) {
          constant_arg_indices->push_back(i);
        }
      }
    
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Tue Mar 12 06:33:33 UTC 2024
    - 30.3K bytes
    - Viewed (0)
  3. tensorflow/compiler/jit/mark_for_compilation_pass.cc

      FunctionLibraryRuntime* lib_runtime =
          pflr->GetFLR(ProcessFunctionLibraryRuntime::kDefaultFLRDevice);
      std::vector<bool> compile_time_const_nodes(graph_->num_node_ids(), false);
      TF_RETURN_IF_ERROR(BackwardsConstAnalysis(
          *graph_, /*compile_time_const_arg_indices=*/nullptr,
          &compile_time_const_nodes, lib_runtime));
      // Iterate over nodes in sorted order so that compiler fuel is deterministic.
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Wed Feb 21 12:19:41 UTC 2024
    - 85.3K bytes
    - Viewed (0)
  4. tensorflow/compiler/jit/encapsulate_subgraphs_pass.cc

            std::vector<bool> const_args(num_args);
            TF_RETURN_IF_ERROR(
                BackwardsConstAnalysis(**subgraph, &const_args,
                                       /*compile_time_const_nodes=*/nullptr, flr));
    
            DataTypeVector arg_types(num_args);
            TF_RETURN_IF_ERROR(GetArgTypes(**subgraph, &arg_types));
    
            // Compute a permutation of the arguments such that the constant
    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