Search Options

Results per page
Sort
Preferred Languages
Advance

Results 1 - 4 of 4 for MayCallFunction (0.13 sec)

  1. tensorflow/compiler/jit/xla_cluster_util.h

    // one GPU (and any number of CPUs).
    bool IsSingleGpuGraph(const Graph& g);
    
    // Returns true if it is possible (but not guaranteed) that `n` calls a
    // function.
    bool MayCallFunction(const Node& n, const FunctionLibraryDefinition* flib_def);
    
    // Returns true if `node` an operator that consumes only the shape of its input,
    // not the data itself.
    bool IsShapeConsumerOp(const Node& node);
    
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Wed Feb 21 09:53:30 UTC 2024
    - 4.6K bytes
    - Viewed (0)
  2. tensorflow/compiler/jit/resource_operation_safety_analysis.cc

      }
    
      // We conservatively assume that functions will both read and write resource
      // variables.  In the future we may consider doing some form of
      // inter-procedural analysis.
      if (MayCallFunction(n, flib_def)) {
        *out_resource_op_kind = XlaResourceOpKind::kReadWrite;
      } else {
        *out_resource_op_kind = std::nullopt;
      }
    
      return absl::OkStatus();
    }
    
    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/xla_cluster_util.cc

                                            : xla_global_jit_level.general;
      VLOG(4) << "GetGlobalJitLevelForGraph returning " << result;
      return result;
    }
    
    bool MayCallFunction(const Node& n, const FunctionLibraryDefinition* flib_def) {
      if (flib_def->Contains(n.type_string())) {
        return true;
      }
    
      // This is a conservative check: there may be nodes with a `func`
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu Feb 29 08:39:39 UTC 2024
    - 21.3K bytes
    - Viewed (0)
  4. tensorflow/compiler/jit/mark_for_compilation_pass.cc

        std::optional<DeviceId> resource_op_device;
        if (is_resource_op) {
          resource_op_device = device;
        }
    
        std::optional<int> resource_var_operation_node_id;
        if (is_resource_op || MayCallFunction(*node, flib_def_)) {
          resource_var_operation_node_id = node->id();
        }
    
        bool is_xla_compile_attr_true =
            GetNodeOrFuncAttr(node, flib_def_, kXlaCompileAttr) ||
    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