Search Options

Results per page
Sort
Preferred Languages
Advance

Results 1 - 1 of 1 for ReplaceLiftedArgNodePlaceholderWithArg (0.31 sec)

  2. github.com/tensorflow/tensorflow

    tensorflow/compiler/jit/extract_outside_compilation_pass.cc 

      TF_ASSIGN_OR_RETURN(Node * ret_node, function_body.graph->AddNode(ret_def));
  function_body.graph->AddEdge(arg_node, 0, ret_node, 0);

  return absl::OkStatus();
}

void ReplaceLiftedArgNodePlaceholderWithArg(
    const FunctionBody& function_body, const int original_arg_count,
    const int arg_idx, const std::vector<Node*>& lifted_arg_nodes,
    Node* arg_node) {
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Tue Mar 12 06:33:33 UTC 2024
    - 104.7K bytes
    - Viewed (0)
Back to top