- Sort Score
- Result 10 results
- Languages All
Results 1 - 10 of 12 for num_inputs (0.31 sec)
-
tensorflow/cc/gradients/functional_grad.cc
(*f.mutable_attr())[attr.first] = attr.second; } std::vector<Output> func_inputs; std::vector<DataType> input_dtypes; const int num_inputs = op.num_inputs(); func_inputs.reserve(num_inputs + grad_inputs.size()); input_dtypes.reserve(num_inputs); for (int i = 0; i < num_inputs; i++) { func_inputs.push_back(op.input(i)); input_dtypes.push_back(op.input_type(i)); }
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Fri Oct 15 20:09:06 UTC 2021 - 2.1K bytes - Viewed (0) -
tensorflow/c/experimental/gradients/grad_test_helper.cc
double abs_error) { auto num_inputs = inputs.size(); std::vector<AbstractTensorHandle*> outputs(num_inputs); auto s = RunModel(grad_model, ctx, inputs, absl::MakeSpan(outputs), /*use_function=*/use_function); ASSERT_EQ(errors::OK, s.code()) << s.message(); for (int i = 0; i < num_inputs; ++i) { if (!outputs[i]) continue;
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed Feb 28 13:53:47 UTC 2024 - 5K bytes - Viewed (0) -
tensorflow/cc/framework/ops.cc
namespace tensorflow { Operation::Operation(Node* n) : inputs_(GetInputs(n)), node_(n) {} Output Operation::input(int32_t i) const { CHECK_NOTNULL(node_); CHECK_GE(i, 0); CHECK_LT(i, node_->num_inputs()); // Handle the case where the input was unknown at the time this // Operation was constructed. if (inputs_[i].first == nullptr && inputs_[i].second == -1) { for (const Edge* e : node_->in_edges()) {
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed Aug 11 01:20:50 UTC 2021 - 3.5K bytes - Viewed (0) -
tensorflow/compiler/jit/xla_compile_util_test.cc
const Node* identity_node = node_name_index.at("identity_op"); EXPECT_EQ(identity_node->op_def().name(), "Identity"); 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");
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue May 16 21:48:05 UTC 2023 - 6K bytes - Viewed (0) -
tensorflow/cc/ops/while_loop_test.cc
namespace tensorflow { namespace { class WhileLoopTest : public ::testing::Test { protected: WhileLoopTest() : scope_(Scope::NewRootScope()) {} void Init(int num_inputs, DataType dtype = DT_INT32) { for (int i = 0; i < num_inputs; ++i) { inputs_.push_back(ops::Placeholder(scope_, dtype)); } } void CreateLoop(const ops::CondGraphBuilderFn& cond,
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 13 22:30:58 UTC 2023 - 6.4K bytes - Viewed (0) -
tensorflow/c/eager/c_api_unified_experimental.cc
absl::StatusMessageAsCStr(status)); } void TF_ExecuteOperation(TF_AbstractOp* op, int num_inputs, TF_AbstractTensor* const* inputs, TF_OutputList* o, TF_Status* s) { for (int i = 0; i < num_inputs; i++) { tsl::Set_TF_Status_from_Status(s, unwrap(op)->AddInput(unwrap(inputs[i]))); if (TF_GetCode(s) != TF_OK) { return;
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu May 09 10:15:17 UTC 2024 - 9K bytes - Viewed (0) -
tensorflow/compiler/jit/xla_kernel_creator_test.cc
ASSERT_TRUE(status.ok()) << status.ToString(); EXPECT_EQ("XTimesY", kernel_->name()); EXPECT_EQ("XTimesY", kernel_->type_string()); EXPECT_EQ(2, kernel_->num_inputs()); EXPECT_EQ(DT_FLOAT, kernel_->input_type(0)); EXPECT_EQ(DT_RESOURCE, kernel_->input_type(1)); EXPECT_EQ(DEVICE_MEMORY, kernel_->input_memory_types()[0]); EXPECT_EQ(HOST_MEMORY, kernel_->input_memory_types()[1]);
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue May 16 01:39:55 UTC 2023 - 5.7K bytes - Viewed (0) -
tensorflow/cc/gradients/data_flow_grad.cc
// grad = [[g_1, g_2], [g_3, g_4], [g_5, g_6]] // indices and data are two equal-sized lists passed // into DynamicStitch. // num_values = 2 int32_t num_values = op.num_inputs() / 2; // Stop propagation along the indices list for (int32_t i = 0; i < num_values; i++) { grad_outputs->push_back(NoGradient()); }
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Sat Jul 24 13:40:35 UTC 2021 - 5.8K bytes - Viewed (0) -
tensorflow/c/eager/c_api_unified_experimental.h
// returned through the provided TF_OutputList. // Any active tape will observe the effects of this execution. void TF_ExecuteOperation(TF_AbstractOp* op, int num_inputs, TF_AbstractTensor* const* inputs, TF_OutputList* o, TF_Status* s); // Creates a new TF_AbstractFunction from the current tracing states in the
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Sun Oct 24 11:27:00 UTC 2021 - 7K bytes - Viewed (0) -
tensorflow/compiler/mlir/quantization/tensorflow/calibrator/calibration_statistics_saver_op.cc
absl::AbortedError( "The `ids` and `calibration_methods` must have the same size.")); // Check the number and type of inputs. OP_REQUIRES(context, context->num_inputs() == ids_.size() * 3, absl::AbortedError("The number of inputs must be three times " "the size of the `ids` list.")); for (int i = 0; i < ids_.size(); ++i) {
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Mon May 13 01:31:23 UTC 2024 - 8K bytes - Viewed (0)