Search Options

Results per page
Sort
Preferred Languages
Advance

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

  1. tensorflow/c/c_api.cc

            "Cannot update edge, incompatible shapes: ", ic_dst->DebugString(shape),
            " and ", ic_dst->DebugString(ic_dst->input(dst.index)), ".");
        return;
      }
      status->status = graph->graph.UpdateEdge(&new_src.oper->node, new_src.index,
                                               &dst.oper->node, dst.index);
    
      if (TF_GetCode(status) == TF_OK) {
        // This modification only updates the destination node for
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Mon Apr 15 03:35:10 UTC 2024
    - 102.3K bytes
    - Viewed (0)
  2. tensorflow/c/c_api_test.cc

      EXPECT_TRUE(found_scalar_const);
      EXPECT_TRUE(found_add);
      EXPECT_TRUE(found_neg);
    
      // Clean up
      TF_DeleteGraph(graph);
      TF_DeleteStatus(s);
    }
    
    TEST(CAPI, UpdateEdge) {
      TF_Status* s = TF_NewStatus();
      TF_Graph* graph = TF_NewGraph();
    
      // Make two scalar constants.
      TF_Operation* one = ScalarConst(1, graph, s, "one");
      ASSERT_EQ(TF_OK, TF_GetCode(s)) << TF_Message(s);
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Mon Apr 15 03:35:10 UTC 2024
    - 96.9K bytes
    - Viewed (0)
  3. tensorflow/compiler/mlir/tensorflow/translate/import_model.cc

        } else if (update_inplace && edge->IsControlEdge()) {
          control_edges.push_back(edge);
        }
      }
    
      for (const auto* edge : data_edges) {
        TF_RETURN_IF_ERROR(graph_->UpdateEdge(placeholder_node, 0, edge->dst(),
                                              edge->dst_input()));
      }
    
      // TODO(lyandy): Preserve control dependencies properly by not forwarding
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Wed May 01 11:17:36 UTC 2024
    - 183.2K bytes
    - Viewed (0)
Back to top