- Sort Score
- Result 10 results
- Languages All
Results 1 - 10 of 14 for input_nodes_ (0.15 sec)
-
tensorflow/cc/framework/gradients.cc
} std::unordered_set<int> stop_backprop_nodes = GetStopBackpropNodes(reachable_nodes, output_nodes); // Populate `input_nodes_` from Outputs in `inputs_`. input_nodes_.reserve(inputs_.size()); for (size_t i = 0; i < inputs_.size(); ++i) { input_nodes_.insert({inputs_[i], i}); } // TODO(andydavis) Consider a more efficient data structure for `pending_` to
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Sat Apr 13 05:57:22 UTC 2024 - 22K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/quantization/lite/quantize_model_test.cc
} class QuantizeModelTest : public testing::Test { protected: QuantizeModelTest() { input_model_ = ReadModel(internal::kConvModelWith0Plus10Weights); readonly_model_ = input_model_->GetModel(); model_ = UnPackFlatBufferModel(*readonly_model_); } std::unique_ptr<FlatBufferModel> input_model_; const Model* readonly_model_; tflite::ModelT model_;
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed Jun 12 23:15:24 UTC 2024 - 73.9K 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/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/mlir/quantization/common/quantization_lib/quantization_config.cc
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue Mar 05 07:39:40 UTC 2024 - 5.9K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/quantization/lite/quantize_weights_test.cc
QuantizeWeightsTest() {} void LoadBasicModel() { input_model_ = ReadTestModel(); model_ = input_model_->GetModel(); } void LoadSharedWeightsModel() { input_model_ = ReadSharedWeightsTestModel(); model_ = input_model_->GetModel(); } void LoadGatherTestModel() { input_model_ = ReadGatherTestModel(); model_ = input_model_->GetModel(); } void LoadCustomOpTestModel() {
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed Jun 12 23:15:24 UTC 2024 - 32.3K bytes - Viewed (0) -
tensorflow/compiler/jit/xla_compile_util_test.cc
EXPECT_EQ(identity_node->attrs().FindByString("T")->type(), DT_FLOAT); EXPECT_EQ(identity_node->num_inputs(), 1); const Node* identity_input_node = nullptr; TF_EXPECT_OK(identity_node->input_node(0, &identity_input_node)); EXPECT_EQ(identity_input_node->name(), "_arg0"); const Node* arg_node = node_name_index.at("_arg0"); EXPECT_EQ(arg_node->op_def().name(), "_Arg");
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue May 16 21:48:05 UTC 2023 - 6K 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/compilability_check_util_test.cc
NodeBuilder while_builder(kFunctionalWhileNodeName, "While", builder.opts().op_registry()); while_builder.Input({input_node, input_node}) .Attr("cond", compilable) .Attr("body", uncompilable); builder.opts().FinalizeBuilder(&while_builder); GraphDef graph_def; TF_EXPECT_OK(builder.ToGraphDef(&graph_def)); std::unique_ptr<Graph> graph(new Graph(flib_def_.get()));
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Fri Jun 10 12:32:39 UTC 2022 - 22.3K bytes - Viewed (0)