- Sort Score
- Result 10 results
- Languages All
Results 1 - 4 of 4 for IsIfNode (0.29 sec)
-
tensorflow/compiler/jit/extract_outside_compilation_pass.cc
if (n->type_string() == "_XlaRecvAtHost" || n->type_string() == "_XlaSendFromHost") { n->ClearAttr("device_ordinal"); n->AddAttr("device_ordinal", device_ordinal_value); } else if (n->IsIfNode()) { for (const string& attr_name : std::vector<string>{"then_branch", "else_branch"}) { NameAttrList branch_func; TF_RETURN_IF_ERROR(GetNodeAttr(n->attrs(), attr_name, &branch_func));
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue Mar 12 06:33:33 UTC 2024 - 104.7K bytes - Viewed (0) -
tensorflow/compiler/jit/compilability_check_util.cc
!IsCompilableWhile(node, lib_runtime, stack_trace, encapsulating_function, uncompilable_nodes)) { LogNotCompilable(node, "unsupported while"); return false; } if (node.IsIfNode() && !IsCompilableIf(node, lib_runtime, stack_trace, encapsulating_function, uncompilable_nodes)) { LogNotCompilable(node, "unsupported if"); return false; }
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue Mar 12 06:33:33 UTC 2024 - 30.3K bytes - Viewed (0) -
tensorflow/compiler/jit/mark_for_compilation_pass.cc
// enable interesting optimizations by themselves. int effective_cluster_size = (node->IsIdentity() || node->IsConstant()) ? 0 : 1; bool has_functional_control_flow = node->IsWhileNode() || node->IsIfNode(); std::optional<DeadnessPredicate> deadness_predicate; if (deadness_analysis_) { TF_ASSIGN_OR_RETURN( deadness_predicate,
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed Feb 21 12:19:41 UTC 2024 - 85.3K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/translate/import_model.cc
// Map Case/If/While and StatelessCase/If/While op in TensorFlow to the common // Case/If/While op in MLIR and add the differentiating attribute. if (node.IsCaseNode()) composite_control_flow_op("Case"); if (node.IsIfNode()) composite_control_flow_op("If"); if (node.IsWhileNode()) { composite_control_flow_op("While"); auto* output_shapes = node.attrs().Find("output_shapes");
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed May 01 11:17:36 UTC 2024 - 183.2K bytes - Viewed (0)