- Sort Score
- Result 10 results
- Languages All
Results 1 - 4 of 4 for MayCallFunction (0.13 sec)
-
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) -
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) -
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) -
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)