Home
last modified time | relevance | path

Searched refs:save_context (Results 1 – 16 of 16) sorted by relevance

/external/tensorflow/tensorflow/python/saved_model/
Dsave_context_test.py24 from tensorflow.python.saved_model import save_context
31 self.assertFalse(save_context.in_save_context())
33 save_context.get_save_options()
36 with save_context.save_context(options):
37 self.assertTrue(save_context.in_save_context())
38 self.assertTrue(save_context.get_save_options().save_debug_info)
44 self.assertFalse(save_context.in_save_context())
46 save_context.get_save_options()
49 with save_context.save_context(options):
50 self.assertTrue(save_context.in_save_context())
[all …]
Dsave_context.py53 def save_context(options): function
DBUILD298 name = "save_context",
300 "save_context.py",
311 ":save_context",
330 ":save_context",
Dsave.py52 from tensorflow.python.saved_model import save_context
793 options = save_context.get_save_options()
1271 with save_context.save_context(options):
/external/tensorflow/tensorflow/python/distribute/
Dtf_function_test.py35 from tensorflow.python.saved_model import save_context
161 with save_context.save_context(save_options.SaveOptions()):
166 with save_context.save_context(save_options.SaveOptions()):
186 with save_context.save_context(save_options.SaveOptions()):
191 with save_context.save_context(save_options.SaveOptions()):
Dvalues_util.py30 from tensorflow.python.saved_model import save_context
369 if not save_context.in_save_context():
371 options = save_context.get_save_options()
378 if ops.inside_function() and not save_context.in_save_context():
Dsharded_variable.py36 from tensorflow.python.saved_model import save_context
503 if save_context.in_save_context():
Dvalues_test.py60 from tensorflow.python.saved_model import save_context
652 with save_context.save_context(options):
934 with save_context.save_context(options):
DBUILD772 "//tensorflow/python/saved_model:save_context",
809 "//tensorflow/python/saved_model:save_context",
1153 "//tensorflow/python/saved_model:save_context",
1277 "//tensorflow/python/saved_model:save_context",
1842 "//tensorflow/python/saved_model:save_context",
Dvalues.py39 from tensorflow.python.saved_model import save_context
524 return self._packed_var is not None and not save_context.in_save_context()
/external/tensorflow/tensorflow/python/eager/
Ddef_function_test.py48 from tensorflow.python.saved_model import save_context
585 with save_context.save_context(
590 with save_context.save_context(
DBUILD617 "//tensorflow/python/saved_model:save_context",
892 "//tensorflow/python/saved_model:save_context",
Dfunction.py68 from tensorflow.python.saved_model import save_context
3229 if save_context.in_save_context():
3231 save_context.get_save_options().experimental_variable_policy)
/external/tensorflow/tensorflow/python/tpu/
Dtpu_embedding_v2.py48 from tensorflow.python.saved_model import save_context
432 if save_context.in_save_context():
861 not context.executing_eagerly() and save_context.in_save_context()):
873 not context.executing_eagerly() and save_context.in_save_context()):
/external/libaom/libaom/av1/encoder/
Dencodeframe.c1525 static AOM_INLINE void save_context(const MACROBLOCK *x, in save_context() function
1922 save_context(x, &x_ctx, mi_row, mi_col, bsize, num_planes); in rd_use_partition()
2096 save_context(x, &x_ctx, mi_row, mi_col, bsize, num_planes); in rd_use_partition()
2236 save_context(x, &x_ctx, mi_row, mi_col, bsize, 3); in nonrd_use_partition()
2335 save_context(x, &x_ctx, mi_row, mi_col, bsize, 3); in nonrd_use_partition()
2823 save_context(x, &x_ctx, mi_row, mi_col, bsize, num_planes); in rd_pick_partition()
4517 save_context(x, &sb_fp_stats->x_ctx, mi_row, mi_col, sb_size, num_planes); in backup_sb_state()
/external/libvpx/libvpx/vp9/encoder/
Dvp9_encodeframe.c2199 static void save_context(MACROBLOCK *const x, int mi_row, int mi_col, in save_context() function
2700 save_context(x, mi_row, mi_col, a, l, sa, sl, bsize); in rd_use_partition()
2863 save_context(x, mi_row, mi_col, a, l, sa, sl, bsize); in rd_use_partition()
3931 save_context(x, mi_row, mi_col, a, l, sa, sl, bsize); in rd_pick_partition()