Search Options

Results per page
Sort
Preferred Languages
Advance

Results 1 - 4 of 4 for SnapshotResourceVariables (0.63 sec)

  1. 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)
  2. 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)
  3. 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)
  4. 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)
Back to top