- Sort Score
- Result 10 results
- Languages All
Results 1 - 10 of 11 for input_edges (0.2 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/unified_api_testutil.h
TF_ExecutionContextGetTFEContext(wrap(ctx), status.get()); TF_RETURN_IF_ERROR(StatusFromTF_Status(status.get())); TFE_TensorHandle* input_eager = TestTensorHandleWithDims<T, datatype>(eager_ctx, data, dims, num_dims); *tensor = unwrap(TF_CreateAbstractTensorFromEagerTensor(input_eager, status.get())); return absl::OkStatus(); } // Return a scalar tensor handle with given value. template <class T, TF_DataType datatype>
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue Feb 27 13:57:45 UTC 2024 - 4K 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) -
tensorflow/compiler/mlir/tensorflow/translate/import_model.cc
// Collect the control operands separately, they will be held by the island. mlir::SmallVector<mlir::Value, 8> control_operands; for (const auto* input_edge : in_edges) { const Node& input_node = *input_edge->src(); if (input_node.IsSource()) { if (in_edges.size() != 1) { return errors::FailedPrecondition( "The node has other inputs besides the _Source node");
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed May 01 11:17:36 UTC 2024 - 183.2K bytes - Viewed (0) -
tensorflow/compiler/jit/deadness_analysis.cc
std::vector<Predicate*> input_preds; TF_RETURN_IF_ERROR(GetInputPreds(n, EdgeKind::kDataAndControl, &input_preds)); const Edge* pred_edge; TF_RETURN_IF_ERROR(n->input_edge(1, &pred_edge)); if (n->type_string() != "_SwitchN") { // bool pred branch selector. Predicate* true_switch; TF_RETURN_IF_ERROR(predicate_factory_.MakeSymbolPredicate(
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue Mar 12 06:33:33 UTC 2024 - 60.4K bytes - Viewed (0) -
tensorflow/compiler/jit/extract_outside_compilation_pass.cc
int num_inputs, Node* send_from_host_node) { std::vector<PartialTensorShape> results(num_inputs); for (int i = 0; i < num_inputs; i++) { const Edge* e; if (!send_from_host_node->input_edge(i, &e).ok()) { return std::nullopt; } std::vector<PartialTensorShape> shapes; if (!GetNodeAttr(e->src()->attrs(), kXlaInferredShapesAttrName, &shapes) .ok()) {
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue Mar 12 06:33:33 UTC 2024 - 104.7K bytes - Viewed (0)