Search Options

Results per page
Sort
Preferred Languages
Advance

Results 1 - 3 of 3 for UpdateEdge (0.11 sec)

  1. tensorflow/compiler/jit/increase_dynamism_for_auto_jit_pass_test.cc

      Output slice =
          ops::Slice(root.WithOpName("slice"), input, begin, size_placeholder);
    
      Output size = ops::Const(root.WithOpName("size"), {-8, 500});
      TF_ASSERT_OK(root.graph()->UpdateEdge(/*new_src=*/size.node(),
                                            /*new_src_index=*/0,
                                            /*dst=*/slice.node(), /*dst_index=*/2));
    
      std::unique_ptr<Graph> result;
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu Feb 22 08:47:20 UTC 2024
    - 18.3K bytes
    - Viewed (0)
  2. tensorflow/compiler/jit/xla_cluster_util_test.cc

      ops::Merge merge(s.WithOpName("merge"), {init_value, init_value});
      Output next_iteration =
          ops::NextIteration(s.WithOpName("next_itr"), merge.output);
      return s.graph()->UpdateEdge(next_iteration.node(), 0, merge.output.node(),
                                   1);
    }
    
    TEST(NodesRelatedToRefVariables, Cycles) {
      Scope root = Scope::NewRootScope().ExitOnError();
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Wed Feb 21 09:53:30 UTC 2024
    - 10.8K bytes
    - Viewed (0)
  3. tensorflow/compiler/jit/build_xla_ops_pass.cc

      root.graph()->AddControlEdge(predicate_as_control.node(),
                                   identity_n.operation.node());
    
      for (int i = 0, end = int32_inputs.size(); i < end; i++) {
        TF_RETURN_IF_ERROR(root.graph()->UpdateEdge(identity_n[i].node(), i, n,
                                                    int32_inputs_input_idxs[i]));
      }
    
      return absl::OkStatus();
    }
    
    Status ReplaceNodeWithXlaCompileAndXlaRun(
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Tue Mar 12 06:33:33 UTC 2024
    - 24.3K bytes
    - Viewed (0)
Back to top