- Sort Score
- Result 10 results
- Languages All
Results 1 - 2 of 2 for CreateVariableInfoLookup (0.2 sec)
-
tensorflow/compiler/jit/variable_info_util.h
std::vector<VariableInfo>* result); std::vector<int> GetResourceVariableIndicesFromContext(OpKernelContext* ctx); Status CreateVariableInfoLookup( absl::Span<VariableInfo const> variable_args, absl::flat_hash_map<int, const VariableInfo*>& variable_info_lookup); } // namespace tensorflow
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue Feb 14 21:57:02 UTC 2023 - 4.3K bytes - Viewed (0) -
tensorflow/compiler/jit/variable_info_util.cc
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, absl::flat_hash_map<int, const VariableInfo*>& variable_info_lookup) { for (const VariableInfo& info : 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)