Searched refs:make_save_spec (Results 1 – 2 of 2) sorted by relevance
53 def make_save_spec(tensor, suffix): function56 specs = [make_save_spec(stamp_token, "_stamp")]57 specs += [make_save_spec(state, "_state")]58 specs += [make_save_spec(are_buckets_ready, "_are_buckets_ready")]59 specs += [make_save_spec(buckets, "buckets")]
76 def make_save_spec(tensor, suffix): function81 make_save_spec(bucket_boundaries[i], '_bucket_boundaries_' + str(i))