- Sort Score
- Result 10 results
- Languages All
Results 1 - 5 of 5 for num_inputs (0.13 sec)
-
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/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/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) -
tensorflow/compiler/jit/variable_info_util.cc
} return absl::OkStatus(); } std::vector<int> GetResourceVariableIndicesFromContext(OpKernelContext* ctx) { std::vector<int> out; for (int64 i = 0; i < ctx->num_inputs(); i++) { if (ctx->input(i).dtype() == DT_RESOURCE) { out.push_back(i); } } return out; } Status CreateVariableInfoLookup( absl::Span<VariableInfo const> variable_args,
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Feb 22 08:47:20 UTC 2024 - 7K bytes - Viewed (0) -
tensorflow/c/experimental/ops/gen/cpp/views/op_view.cc
const std::vector<AttrView>& OpView::Attributes() const { return argument_attrs_; } const std::vector<OpArgumentView>& OpView::AllArguments() const { return all_arguments_; } int OpView::NumInputs() const { return input_args_.size(); } int OpView::NumOutputs() const { return output_args_.size(); } ArgView OpView::OnlyInput() const { CHECK_EQ(input_args_.size(), 1); // Crash OK return input_args_.front();
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Mon Jun 03 07:02:00 UTC 2024 - 3.2K bytes - Viewed (0)