Searched refs:restore_slice (Results 1 – 5 of 5) sorted by relevance
| /external/tensorflow/tensorflow/core/kernels/ |
| D | save_restore_tensor.h | 54 int preferred_shard, bool restore_slice, int restore_index);
|
| D | save_restore_tensor.cc | 146 int preferred_shard, bool restore_slice, int restore_index) { in RestoreTensor() argument
194 if (restore_slice) { in RestoreTensor()
|
| /external/tensorflow/tensorflow/python/kernel_tests/ |
| D | save_restore_ops_test.py | 74 op = gen_io_ops.restore_slice("model", "var", "3 4 0,1:-", dtypes.float32)
|
| /external/tensorflow/tensorflow/python/ops/ |
| D | io_ops.py | 94 return gen_io_ops.restore_slice(
|
| /external/tensorflow/tensorflow/core/grappler/costs/ |
| D | graph_properties_test.cc | 831 Output restore_slice = in TEST_F() local
835 ops::Assign(s.WithOpName("init_restore_slice"), var, restore_slice); in TEST_F()
|