- Sort Score
- Result 10 results
- Languages All
Results 21 - 24 of 24 for input_model (0.12 sec)
-
tensorflow/compiler/jit/increase_dynamism_for_auto_jit_pass.cc
// Create a list of all control dependencies to be copied when possibly // replacing nodes related to slice_size. Node* old_size; std::vector<const Edge*> old_size_ctrl_deps; TF_RETURN_IF_ERROR(slice->input_node(2, &old_size)); absl::c_copy_if(old_size->in_edges(), std::back_inserter(old_size_ctrl_deps), [](const Edge* e) { return e->IsControlEdge(); }); Output slice_size;
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue Mar 12 06:33:33 UTC 2024 - 13.8K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/translate/import_model.cc
for (const auto* input_edge : in_edges) { const Node& input_node = *input_edge->src(); if (input_node.IsSource()) { if (in_edges.size() != 1) { return errors::FailedPrecondition( "The node has other inputs besides the _Source node"); } // We don't import the _SOURCE node. continue; } if (input_node.IsArg() && input_edge->IsControlEdge()) {
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed May 01 11:17:36 UTC 2024 - 183.2K bytes - Viewed (0) -
tensorflow/compiler/jit/extract_outside_compilation_pass.cc
} if (!ret_node) { return errors::Internal("No _Retval node for loop cond function ", loop_cond_func->name()); } Node* loop_cond; TF_RETURN_IF_ERROR(ret_node->input_node(0, &loop_cond)); // Build the XlaSendToHost node. NodeDefBuilder send_loop_cond_builder( absl::StrCat("send_oc_while_cond_", while_node->name()), "XlaSendToHost");
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/mark_for_compilation_pass.cc
return false; } const Node* switch_node = (*it)->src(); // Check if the Switch is driven by LoopCond. const Node* maybe_loop_cond; TF_RETURN_IF_ERROR(switch_node->input_node(1, &maybe_loop_cond)); if (!maybe_loop_cond->IsLoopCond()) { return false; } // Check if the Identity is driving any const nodes through a control edge. bool driving_any_consts =
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed Feb 21 12:19:41 UTC 2024 - 85.3K bytes - Viewed (0)