- Sort Score
- Result 10 results
- Languages All
Results 1 - 4 of 4 for CreateVariableInfoLookup (0.34 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) -
tensorflow/compiler/jit/get_compiler_ir.cc
args.push_back(arg); } // Build Xla Compiler Arguments from concrete_fn.captured_inputs absl::flat_hash_map<int, const VariableInfo*> variable_info_lookup; TF_RETURN_IF_ERROR( CreateVariableInfoLookup(variable_args, variable_info_lookup)); for (const VariableInfo& info : variable_args) { TF_RET_CHECK(!info.var() || info.lock_held() || info.shared_lock_held())
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Feb 22 06:59:07 UTC 2024 - 19K bytes - Viewed (0) -
tensorflow/compiler/jit/xla_launch_util.cc
if (dev_info) device_context = dev_info->default_context; } } absl::flat_hash_map<int, const VariableInfo*> variable_info_lookup; TF_CHECK_OK(CreateVariableInfoLookup(variable_args, variable_info_lookup)); for (int64_t input_num = 0; input_num < inputs.size(); ++input_num) { const Tensor* input = inputs[input_num]; XlaCompiler::Argument& arg = out[input_num];
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu May 16 00:36:08 UTC 2024 - 40.4K bytes - Viewed (0)