- Sort Score
- Result 10 results
- Languages All
Results 1 - 4 of 4 for SnapshotResourceVariables (0.63 sec)
-
tensorflow/compiler/jit/variable_info_util.h
// // We snapshot the entire set of resource variables as one atomic operation. // This models Read->* dependencies between resource variable operations. See // jit/resource_operation_safety_analysis for details. Status SnapshotResourceVariables(OpKernelContext* ctx, absl::Span<const int> variable_indices, absl::Span<VariableInfo const> variable_infos,
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
variable_ptrs.reserve(variables.size()); for (auto& var : variables) { variable_ptrs.push_back(&var); } return LockVariables(absl::MakeSpan(variable_ptrs)); } Status SnapshotResourceVariables(OpKernelContext* ctx, absl::Span<const int> variable_indices, absl::Span<VariableInfo const> variable_infos,
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/xla_compile_on_demand_op.cc
*ctx, inputs, *constant_indices_or, variable_indices, &variables, &args)); OP_REQUIRES_OK(ctx, SnapshotResourceVariables(ctx, variable_indices, variables, &variable_args)); } XlaDeviceCompiler* xla_device_compiler; xla::LocalExecutable* executable;
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/kernels/xla_ops.cc
variable_indices, &variable_infos)); TF_RETURN_IF_ERROR(LockVariables(absl::MakeSpan(variable_infos))); TF_RETURN_IF_ERROR(SnapshotResourceVariables(ctx, variable_indices, variable_infos, &result.second)); TF_ASSIGN_OR_RETURN(result.first,
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Fri May 17 22:46:36 UTC 2024 - 41.4K bytes - Viewed (0)