- Sort Score
- Result 10 results
- Languages All
Results 1 - 10 of 27 for input_node (0.17 sec)
-
tensorflow/compiler/jit/rearrange_function_argument_pass_test.cc
ASSERT_NE(if_node, nullptr); const Node *input_node; TF_CHECK_OK(if_node->input_node(1, &input_node)); EXPECT_EQ(input_node->name(), "arg1"); TF_CHECK_OK(if_node->input_node(2, &input_node)); EXPECT_EQ(input_node->name(), "arg0"); const Node *ret0_node = node_name_index.at("ret0"); ASSERT_NE(ret0_node, nullptr); TF_CHECK_OK(ret0_node->input_node(0, &input_node)); EXPECT_EQ(input_node->name(), "if");
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Fri Feb 09 11:36:41 UTC 2024 - 10.5K bytes - Viewed (0) -
tensorflow/compiler/jit/shape_inference.cc
shape_refiner->GetContext(n); for (int i = 0; i < n->num_inputs(); i++) { const Node* input_node; if (n->input_node(i, &input_node).ok()) { auto shapes_and_types = context->input_handle_shapes_and_types(i); if (shapes_and_types) { context->set_output_handle_shapes_and_types(0,
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Fri May 31 00:41:19 UTC 2024 - 13K bytes - Viewed (0) -
tensorflow/compiler/jit/extract_outside_compilation_pass_test.cc
send_recv_nodes.push_back(n); } } EXPECT_EQ(num_send_from_host, 1); EXPECT_EQ(num_recv_at_host, 1); for (Node *n : send_recv_nodes) { Node *input_node; TF_CHECK_OK(n->input_node(n->num_inputs() - 1, &input_node)); EXPECT_EQ(input_node, key_placeholder); bool has_control_edge_to_sequencer = false; for (const Edge *e : n->out_edges()) { if (e->IsControlEdge() && e->dst() == sequencer) {
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed Sep 06 19:12:29 UTC 2023 - 41K bytes - Viewed (0) -
tensorflow/compiler/jit/increase_dynamism_for_auto_jit_pass.cc
// Create a list of all control dependencies to be copied when possibly // replacing nodes related to slice_size. Node* old_size; std::vector<const Edge*> old_size_ctrl_deps; TF_RETURN_IF_ERROR(slice->input_node(2, &old_size)); absl::c_copy_if(old_size->in_edges(), std::back_inserter(old_size_ctrl_deps), [](const Edge* e) { return e->IsControlEdge(); }); Output slice_size;
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/mlir/tensorflow/translate/import_model.cc
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"); } // We don't import the _SOURCE node. continue; } if (input_node.IsArg() && input_edge->IsControlEdge()) {
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/extract_outside_compilation_pass.cc
} if (!ret_node) { return errors::Internal("No _Retval node for loop cond function ", loop_cond_func->name()); } Node* loop_cond; TF_RETURN_IF_ERROR(ret_node->input_node(0, &loop_cond)); // Build the XlaSendToHost node. NodeDefBuilder send_loop_cond_builder( absl::StrCat("send_oc_while_cond_", while_node->name()), "XlaSendToHost");
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue Mar 12 06:33:33 UTC 2024 - 104.7K bytes - Viewed (0) -
tensorflow/compiler/jit/mark_for_compilation_pass.cc
return false; } const Node* switch_node = (*it)->src(); // Check if the Switch is driven by LoopCond. const Node* maybe_loop_cond; TF_RETURN_IF_ERROR(switch_node->input_node(1, &maybe_loop_cond)); if (!maybe_loop_cond->IsLoopCond()) { return false; } // Check if the Identity is driving any const nodes through a control edge. bool driving_any_consts =
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed Feb 21 12:19:41 UTC 2024 - 85.3K bytes - Viewed (0) -
tensorflow/c/c_api_function.cc
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Mon Apr 15 03:35:10 UTC 2024 - 13.6K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/quantization/lite/quantize_weights.cc
StringRef op_name(tflite_op_name.data(), tflite_op_name.size()); return op_name.lower(); } std::unique_ptr<tflite::ModelT> CreateMutableModelFromFile( const tflite::Model* input_model) { auto copied_model = std::make_unique<tflite::ModelT>(); input_model->UnPackTo(copied_model.get(), nullptr); return copied_model; } } // namespace // TODO(b/214314076): Support MLIR model as an input for the C++ dynamic range
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed Jun 12 23:15:24 UTC 2024 - 9.5K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/sparsity/sparsify_model_test.cc
"tensorflow/lite/testdata/sparse_tensor.bin"); tflite::ModelT input_model; input_fbm->GetModel()->UnPackTo(&input_model); // Populate input metadata auto model_metadata_buffer = std::make_unique<tflite::BufferT>(); model_metadata_buffer->data = std::vector<uint8_t>(expected_value.begin(), expected_value.end()); input_model.buffers.push_back(std::move(model_metadata_buffer));
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Mon Jun 10 20:16:40 UTC 2024 - 2.9K bytes - Viewed (0)