- Sort Score
- Result 10 results
- Languages All
Results 1 - 5 of 5 for ResourceVarsSnapshot (0.22 sec)
-
tensorflow/compiler/jit/variable_info_util.h
// Snapshot of resource variables for a TF kernel invocation, mapping from // parameter number to values at execution time. If the resource variable is not // initialized, the value will not be present. using ResourceVarsSnapshot = absl::flat_hash_map<int, std::optional<Tensor>>; // Takes a snapshot of the values of resource variable arguments, whose indices // are specified in `variable_indices` argument. We snapshot tensors that back
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/xla_compile_on_demand_op.h
DeviceCompilationProfiler** profiler, const XlaCompiler::CompilationResult** result, xla::PjRtLoadedExecutable** executable); Status Run(const ResourceVarsSnapshot& variable_args, const XlaCompiler::CompilationResult* result, const DeviceCompiler<xla::LocalExecutable, xla::LocalClient>* xla_device_compiler,
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed Sep 06 19:12:29 UTC 2023 - 3.2K bytes - Viewed (0) -
tensorflow/compiler/jit/kernels/xla_ops.cc
return compilation_result_; } const ResourceVarsSnapshot& resource_var_snapshots() const { return resource_var_snapshots_; } int num_constant_args() const { return num_constant_args_; } private: ClientType* client_; ExecutableType* executable_; const XlaCompiler::CompilationResult* compilation_result_; ResourceVarsSnapshot resource_var_snapshots_; int num_constant_args_;
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Fri May 17 22:46:36 UTC 2024 - 41.4K bytes - Viewed (0) -
tensorflow/compiler/jit/xla_compile_on_demand_op.cc
static_cast<Device*>(ctx.device()))); return absl::OkStatus(); } } // namespace Status XlaCompileOnDemandOp::Run(const ResourceVarsSnapshot& variable_args, const XlaCompiler::CompilationResult* result, const XlaDeviceCompiler* xla_device_compiler,
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Feb 29 08:39:39 UTC 2024 - 13.4K bytes - Viewed (0) -
tensorflow/compiler/jit/variable_info_util.cc
absl::Span<const int> variable_indices, absl::Span<VariableInfo const> variable_infos, ResourceVarsSnapshot* result) { for (int i = 0, end = variable_indices.size(); i < end; i++) { Var* var = variable_infos[i].var(); (*result)[variable_indices[i]] = var ? std::make_optional(*var->tensor()) : std::nullopt;
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Feb 22 08:47:20 UTC 2024 - 7K bytes - Viewed (0)