Search Options

Results per page
Sort
Preferred Languages
Advance

Results 1 - 5 of 5 for FixupSourceAndSinkEdges (0.31 sec)

  1. tensorflow/compiler/jit/xla_cluster_util_test.cc

      Output enter =
          ops::internal::Enter(root.WithOpName("enter"), a, "only_frame");
      Output exit = ops::internal::Exit(root.WithOpName("exit"), enter);
      Output b = ops::Add(root.WithOpName("b"), a, exit);
    
      FixupSourceAndSinkEdges(root.graph());
    
      GraphCycles cycles;
      TF_ASSERT_OK(CreateCycleDetectionGraph(root.graph(), &cycles).status());
      EXPECT_FALSE(cycles.CanContractEdge(a.node()->id(), b.node()->id()));
    }
    
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Wed Feb 21 09:53:30 UTC 2024
    - 10.8K bytes
    - Viewed (0)
  2. tensorflow/compiler/jit/build_xla_ops_pass_test.cc

        if (n->requested_device().empty()) {
          n->set_assigned_device_name(kCpuDevice);
        } else {
          n->set_assigned_device_name(n->requested_device());
        }
      }
    
      FixupSourceAndSinkEdges(graph.get());
    
      GraphOptimizationPassWrapper wrapper;
      GraphOptimizationPassOptions opt_options =
          wrapper.CreateGraphOptimizationPassOptions(&graph);
      opt_options.flib_def = &flib_def;
    
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu Feb 22 08:47:20 UTC 2024
    - 12.2K bytes
    - Viewed (0)
  3. tensorflow/compiler/jit/increase_dynamism_for_auto_jit_pass.cc

        TF_RETURN_IF_ERROR(
            RewriteSlice(g, n, *slice_inputs, *GetXlaClusterForNode(*n)));
      }
    
      if (!slices_to_rewrite.empty()) {
        // We've added constants to the graph; hook them up to _SOURCE.
        FixupSourceAndSinkEdges(g);
      }
    
      *changed = !slices_to_rewrite.empty();
    
      return absl::OkStatus();
    }
    }  // namespace
    
    Status IncreaseDynamismForAutoJitPass::Run(
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Tue Mar 12 06:33:33 UTC 2024
    - 13.8K bytes
    - Viewed (0)
  4. tensorflow/compiler/mlir/tensorflow/translate/export_graphdef.cc

      for (Operation& inst : graph_op.GetBody()) {
        TF_RETURN_IF_ERROR(exporter.AddEdge(&inst));
      }
      // Fixes the edges between the inserted nodes and special "_SOURCE" and
      // "_SINK".
      FixupSourceAndSinkEdges(graph.get());
    
      FixupInputNamesFromEdges(graph.get());
    
      TF_RETURN_IF_ERROR(
          exporter.GetControlRetNodes(graph_op.GetFetch(), control_ret_nodes));
    
      return graph;
    }
    
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Wed May 01 11:17:36 UTC 2024
    - 35.2K bytes
    - Viewed (0)
  5. tensorflow/compiler/mlir/tf2xla/api/v2/tf_executor_to_graph.cc

      for (Operation& inst : graph_op.GetBody()) {
        TF_RETURN_IF_ERROR(exporter.AddEdge(&inst));
      }
      // Fixes the edges between the inserted nodes and special "_SOURCE" and
      // "_SINK".
      FixupSourceAndSinkEdges(graph.get());
    
      FixupInputNamesFromEdges(graph.get());
    
      TF_RETURN_IF_ERROR(
          exporter.GetControlRetNodes(graph_op.GetFetch(), control_ret_nodes));
    
      return graph;
    }
    
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu Jun 06 23:04:51 UTC 2024
    - 35.2K bytes
    - Viewed (0)
Back to top