- Sort Score
- Result 10 results
- Languages All
Results 11 - 12 of 12 for num_inputs (0.09 sec)
-
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)