- Sort Score
- Result 10 results
- Languages All
Results 1 - 6 of 6 for input_edges (0.22 sec)
-
tensorflow/compiler/jit/build_xla_ops_pass.cc
std::vector<const Edge*> input_edges_vector; TF_RETURN_IF_ERROR(n->input_edges(&input_edges_vector)); absl::Span<const Edge*> input_edges(input_edges_vector); absl::c_transform(input_edges.subspan(0, num_constant_inputs), std::back_inserter(result->constant_inputs), IncomingEdgeAsOutput); absl::c_transform( input_edges.subspan(num_constant_inputs, num_non_constant_inputs),
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue Mar 12 06:33:33 UTC 2024 - 24.3K bytes - Viewed (0) -
tensorflow/compiler/jit/encapsulate_xla_computations_pass.cc
TF_ASSIGN_OR_RETURN(const XlaFunctionInfo xla_function_info, get_xla_function_info(*launch)); std::vector<const Edge*> in_edges; TF_RETURN_IF_ERROR(launch->input_edges(&in_edges)); const int num_inputs = in_edges.size(); const int variable_start_index = xla_function_info.variable_start_index; const int num_variables = num_inputs - variable_start_index;
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue Mar 12 06:33:33 UTC 2024 - 15.1K bytes - Viewed (0) -
tensorflow/c/eager/c_api_unified_experimental_test.cc
TFE_TensorHandle* input_eager = TestScalarTensorHandle(eager_ctx, 2.0f); func_args.push_back(TF_CreateAbstractTensorFromEagerTensor(input_eager, s)); ASSERT_EQ(TF_OK, TF_GetCode(s)) << TF_Message(s); input_eager = TestScalarTensorHandle(eager_ctx, 3.0f); func_args.push_back(TF_CreateAbstractTensorFromEagerTensor(input_eager, s)); ASSERT_EQ(TF_OK, TF_GetCode(s)) << TF_Message(s); }
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Fri May 19 21:44:52 UTC 2023 - 39.1K bytes - Viewed (0) -
tensorflow/compiler/jit/increase_dynamism_for_auto_jit_pass.cc
const int kSliceSizeIndex = 2; const Edge* slice_input_edge; TF_RETURN_IF_ERROR(slice->input_edge(kSliceInputIndex, &slice_input_edge)); const Edge* slice_size_edge; TF_RETURN_IF_ERROR(slice->input_edge(kSliceSizeIndex, &slice_size_edge)); const Edge* slice_begin_edge; TF_RETURN_IF_ERROR(slice->input_edge(kSliceBeginIndex, &slice_begin_edge)); SliceInputs slice_inputs; slice_inputs.input =
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/jit/encapsulate_util.cc
std::map<std::pair<string, int>, Node*> placeholders; for (int i = 0, end = edges.size(); i < end; i++) { Node* dst = g->FindNodeId(edges[i].dst_node_id); const Edge* e; TF_RETURN_IF_ERROR(dst->input_edge(edges[i].dst_input, &e)); Node* src = e->src(); int src_output = e->src_output(), dst_input = e->dst_input(); g->RemoveEdge(e); // Find or create placeholder node. string new_name =
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue Mar 12 06:33:33 UTC 2024 - 15.5K bytes - Viewed (0) -
tensorflow/compiler/jit/node_matchers.cc
} } bool MatchAndExplainInput(const Node* node, int input_idx, ::testing::MatchResultListener* listener) const { const Edge* edge; if (!node->input_edge(input_idx, &edge).ok()) { if (listener->IsInterested()) { *listener << "\ncould not find incoming edge for input " << input_idx; } return false; }
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Fri Jun 03 16:15:20 UTC 2022 - 16.8K bytes - Viewed (0)