Search Options

Results per page
Sort
Preferred Languages
Advance

Results 1 - 2 of 2 for GetNodeAttrString (0.31 sec)

  1. tensorflow/compiler/jit/encapsulate_xla_computations_pass.cc

    }
    
    /*static*/ Status EncapsulateXlaComputationsPass::BuildXlaLaunchOps(
        Graph* graph) {
      const auto is_xla_launch_node = [](const Node& node) -> absl::StatusOr<bool> {
        const string& name = GetNodeAttrString(node.attrs(), kXlaClusterIdAttr);
        return !name.empty();
      };
    
      const auto get_xla_function_info =
          [](const Node& node) -> absl::StatusOr<XlaFunctionInfo> {
        XlaFunctionInfo result;
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Tue Mar 12 06:33:33 UTC 2024
    - 15.1K bytes
    - Viewed (0)
  2. tensorflow/compiler/jit/mark_for_compilation_pass.cc

        // If global_jit_level_ is ON, respect only _XlaInternalScope.
        const string& scope =
            GetNodeAttrString(node->attrs(), kXlaInternalScopeAttr);
        if (!scope.empty()) {
          return scope;
        }
      } else {
        // If global_jit_level_ is OFF, respect only _XlaScope.
        const string& scope = GetNodeAttrString(node->attrs(), kXlaScopeAttr);
        if (!scope.empty()) {
          return scope;
        }
      }
    
    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