Search Options

Results per page
Sort
Preferred Languages
Advance

Results 1 - 4 of 4 for ComputeIncompatibleResourceOperationPairs (0.83 sec)

  1. tensorflow/compiler/jit/resource_operation_safety_analysis.h

    //
    // If `resource_ops_to_ignore` is set then nodes for which it returns true are
    // ignored (we pretend these nodes are not resource operations).
    Status ComputeIncompatibleResourceOperationPairs(
        const Graph& g, const FunctionLibraryDefinition* flib_def,
        const std::function<Status(const Node&, bool*)>& resource_ops_to_ignore,
        std::vector<std::pair<int, int>>* result);
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Wed Sep 06 19:12:29 UTC 2023
    - 3.2K bytes
    - Viewed (0)
  2. tensorflow/compiler/jit/resource_operation_safety_analysis.cc

      return absl::StrCat(
          "[", n.name(), ": ", n.type_string(), "(",
          XlaResourceOpInfo::XlaResourceOpKindToString(resource_op_kind), ")", "]");
    }
    }  // namespace
    
    Status ComputeIncompatibleResourceOperationPairs(
        const Graph& g, const FunctionLibraryDefinition* flib_def,
        const std::function<Status(const Node&, bool*)>& resource_ops_to_ignore,
        std::vector<std::pair<int, int>>* result) {
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Fri Feb 09 11:36:41 UTC 2024
    - 11.2K bytes
    - Viewed (0)
  3. tensorflow/compiler/jit/resource_operation_safety_analysis_test.cc

    }
    
    Status ComputeIncompatiblePairs(Graph* g,
                                    std::vector<std::pair<int, int>>* result) {
      FixupSourceAndSinkEdges(g);
      return ComputeIncompatibleResourceOperationPairs(*g, &g->flib_def(), {},
                                                       result);
    }
    
    TEST(ResourceOperationSafetyAnalysisTest, WriteRead) {
      Scope root = Scope::NewRootScope().ExitOnError();
    
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Tue Apr 28 16:53:59 UTC 2020
    - 18.7K bytes
    - Viewed (0)
  4. tensorflow/compiler/jit/mark_for_compilation_pass.cc

        return IgnoreResourceOpForSafetyAnalysis(&device_info_cache_, n, ignore);
      };
    
      std::vector<std::pair<int, int>> unsafe_resource_deps_vect;
      TF_RETURN_IF_ERROR(ComputeIncompatibleResourceOperationPairs(
          *graph_, flib_def_, ignore_resource_ops, &unsafe_resource_deps_vect));
      absl::c_copy(
          unsafe_resource_deps_vect,
          std::inserter(unsafe_resource_deps_, unsafe_resource_deps_.begin()));
    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