Searched refs:ResourceVarsSnapshot (Results 1 – 5 of 5) sorted by relevance
49 ResourceVarsSnapshot* variable_args,55 const ResourceVarsSnapshot& variable_args);
48 const ResourceVarsSnapshot& variable_args) { in Run()104 XlaCompilationCache** cache, ResourceVarsSnapshot* variable_args, in Compile()168 ResourceVarsSnapshot variable_args; in Compute()
40 using ResourceVarsSnapshot = absl::flat_hash_map<int, absl::optional<Tensor>>; variable102 ResourceVarsSnapshot* result);
180 ResourceVarsSnapshot* result) { in SnapshotResourceVariables()
80 ResourceVarsSnapshot resource_var_snapshots, int num_constant_args) in XlaExecutableClosure()95 const ResourceVarsSnapshot& resource_var_snapshots() const { in resource_var_snapshots()104 ResourceVarsSnapshot resource_var_snapshots_;376 ResourceVarsSnapshot variables; in Compute()