| /external/tensorflow/tensorflow/security/advisory/ |
| tfsa-2018-005.md | 9 TensorFlow checkpoint meta file uses Google's [https://github.com/google/snappy](snappy)
15 A maliciously crafted checkpoint meta file could cause TensorFlow to crash or
|
| /external/perfetto/src/base/test/ |
| test_task_runner.cc | 54 void TestTaskRunner::RunUntilCheckpoint(const std::string& checkpoint,
57 if (checkpoints_.count(checkpoint) == 0) {
58 fprintf(stderr, "[TestTaskRunner] Checkpoint \"%s\" does not exist.\n",
59 checkpoint.c_str());
62 if (checkpoints_[checkpoint])
66 [this, checkpoint] {
67 if (checkpoints_[checkpoint])
69 fprintf(stderr, "[TestTaskRunner] Failed to reach checkpoint \"%s\"\n",
70 checkpoint.c_str());
75 pending_checkpoint_ = checkpoint;
[all...] |
| test_task_runner.h | 54 std::function<void()> CreateCheckpoint(const std::string& checkpoint);
55 void RunUntilCheckpoint(const std::string& checkpoint,
|
| /external/tensorflow/tensorflow/core/kernels/ |
| save_restore_tensor.h | 26 // Legacy / V1 checkpoint format.
37 checkpoint::TensorSliceWriter::CreateBuilderFunction builder_func,
53 checkpoint::TensorSliceReader::OpenTableFunction open_func,
56 // V2 checkpoint format.
58 // Invokes the V2 checkpoint read path to read tensors.
|
| restore_op.cc | 34 preferred_shard_ = checkpoint::TensorSliceReader::kLoadAllShards;
43 RestoreTensor(context, &checkpoint::OpenTableTensorSliceReader,
60 preferred_shard_ = checkpoint::TensorSliceReader::kLoadAllShards;
69 RestoreTensor(context, &checkpoint::OpenTableTensorSliceReader,
|
| save_restore_tensor.cc | 42 checkpoint::TensorSliceWriter::CreateBuilderFunction builder_func,
84 checkpoint::TensorSliceWriter writer(filename_t.flat<string>()(0),
92 // checkpoint.
108 OP_REQUIRES_OK(context, checkpoint::ParseShapeAndSlice(
145 checkpoint::TensorSliceReader::OpenTableFunction open_func,
162 std::unique_ptr<checkpoint::TensorSliceReader> allocated_reader;
164 const checkpoint::TensorSliceReader* reader = nullptr;
171 allocated_reader.reset(new checkpoint::TensorSliceReader(
183 "\" not found in checkpoint files ", file_pattern));
198 OP_REQUIRES_OK(context, checkpoint::ParseShapeAndSlice
[all...] |
| /external/vixl/src/ |
| pool-manager-impl.h | 63 // the alignment into account, which only makes the checkpoint calculations
116 // Recalculate the checkpoint before emitting the footer. The footer might
162 T PoolManager<T>::UpdateCheckpointForObject(T checkpoint,
164 checkpoint -= object->label_base_->GetPoolObjectSizeInBytes();
165 if (checkpoint > object->max_location_) checkpoint = object->max_location_;
166 checkpoint = AlignDown(checkpoint, object->alignment_);
167 return checkpoint;
176 static inline bool CheckCurrentPC(T pc, T checkpoint) {
318 T checkpoint = MaxCheckpoint<T>(); local
[all...] |
| /external/perfetto/test/ |
| test_helper.h | 67 std::string AddID(const std::string& checkpoint) {
68 return checkpoint + "." + std::to_string(instance_num_);
71 std::function<void()> CreateCheckpoint(const std::string& checkpoint) {
72 return task_runner_->CreateCheckpoint(AddID(checkpoint));
75 void RunUntilCheckpoint(const std::string& checkpoint,
77 return task_runner_->RunUntilCheckpoint(AddID(checkpoint), timeout_ms);
|
| test_helper.cc | 115 auto checkpoint = CreateCheckpoint("attach." + key); local
116 on_attach_callback_ = [&success, checkpoint](bool s) {
118 checkpoint();
139 auto checkpoint = CreateCheckpoint(checkpoint_name); local
140 endpoint_->Flush(timeout_ms, [checkpoint](bool) { checkpoint(); });
|
| /external/tensorflow/tensorflow/contrib/util/ |
| inspect_checkpoint.cc | 26 tensorflow::checkpoint::TensorSliceReader reader(
27 in, tensorflow::checkpoint::OpenTableTensorSliceReader);
30 fprintf(stderr, "Unable to open the checkpoint file\n");
|
| /external/autotest/client/tests/selftest/ |
| selftest.py | 16 def __mark(self, checkpoint):
29 logging.debug("checkpoint %d %d", current, checkpoint)
31 if (current != checkpoint):
33 "%d when %d expected" % (current, checkpoint))
|
| /external/tensorflow/tensorflow/core/util/ |
| saved_tensor_slice_util_test.cc | 25 namespace checkpoint { namespace in namespace:tensorflow
45 } // namespace checkpoint
|
| tensor_slice_reader_cache.h | 31 namespace checkpoint { namespace in namespace:tensorflow
84 } // namespace checkpoint
|
| /external/tensorflow/tensorflow/python/util/ |
| py_checkpoint_reader.i | 27 %typemap(out) const tensorflow::checkpoint::TensorSliceReader::VarToShapeMap& {
71 %typemap(out) const tensorflow::checkpoint::TensorSliceReader::VarToDataTypeMap& {
105 tensorflow::checkpoint::CheckpointReader* reader,
124 tensorflow::checkpoint::CheckpointReader* reader,
131 %unignore tensorflow::checkpoint;
132 %unignore tensorflow::checkpoint::CheckpointReader;
133 %unignore tensorflow::checkpoint::CheckpointReader::CheckpointReader;
134 %unignore tensorflow::checkpoint::CheckpointReader::~CheckpointReader;
135 %rename("debug_string") tensorflow::checkpoint::CheckpointReader::DebugString;
136 %rename("get_variable_to_shape_map") tensorflow::checkpoint::CheckpointReader::GetVariableToShapeMap
[all...] |
| /external/tensorflow/tensorflow/python/data/kernel_tests/ |
| iterator_checkpoint_test.py | 15 """Checkpoint tests for `tf.data.Iterator`."""
46 checkpoint = trackable_utils.Checkpoint(iterator=iterator)
48 save_path = checkpoint.save(checkpoint_prefix)
51 checkpoint.restore(save_path).run_restore_ops()
76 checkpoint = trackable_utils.Checkpoint(
82 save_path = checkpoint.save(checkpoint_prefix)
86 checkpoint.restore(save_path).run_restore_ops()
99 checkpoint = trackable_utils.Checkpoint(iterator=iterator
[all...] |
| /external/tensorflow/tensorflow/contrib/checkpoint/ |
| __init__.py | 35 Checkpoint management:
47 from tensorflow.contrib.checkpoint.python.containers import UniqueNameTracker
48 from tensorflow.contrib.checkpoint.python.python_state import NumpyState
49 from tensorflow.contrib.checkpoint.python.split_dependency import split_dependency
50 from tensorflow.contrib.checkpoint.python.visualize import dot_graph_from_checkpoint
|
| /external/tensorflow/tensorflow/python/training/tracking/ |
| base.py | 97 """Embeds a tensor in a checkpoint with no restore ops."""
135 """Saves Python state in a checkpoint."""
141 name: The checkpoint key to write to.
143 string. This function is run every time a checkpoint is written.
193 def __init__(self, checkpoint, proto_id):
194 """Specify an object within a checkpoint.
197 checkpoint: A _CheckpointRestoreCoordinator object.
200 self._checkpoint = checkpoint
214 """Set a checkpoint<->object correspondence and process slot variables.
224 checkpoint = self.checkpoin
397 def checkpoint(self): member in class:CheckpointPosition
[all...] |
| tracking_test.py | 110 checkpoint = util.Checkpoint(a=a)
112 checkpoint.save(os.path.join(self.get_temp_dir(), "ckpt"))
122 checkpoint = util.Checkpoint(a=a)
124 checkpoint.save(os.path.join(self.get_temp_dir(), "ckpt"))
152 checkpoint = util.Checkpoint(a=a)
153 checkpoint.save(os.path.join(self.get_temp_dir(), "ckpt"))
156 checkpoint.save(os.path.join(self.get_temp_dir(), "ckpt")
[all...] |
| util_test.py | 132 # checkpoint name will be the same.
135 # The .name attribute may be globally influenced, but the checkpoint name
179 checkpoint = trackable_utils.Checkpoint(dense=dense)
181 save_path = checkpoint.save(checkpoint_prefix)
198 checkpoint = trackable_utils.Checkpoint(x=CallsFunctionalStuff())
200 checkpoint.save(prefix)
206 checkpoint_reversed = trackable_utils.Checkpoint(
263 # go in the checkpoint, since it is never depended on
724 checkpoint = trackable_utils.Checkpoint( variable in class:CheckpointingTests.testAnonymousVarsInInit.Model
[all...] |
| /external/tensorflow/tensorflow/core/profiler/internal/ |
| tfprof_stats.h | 57 std::unique_ptr<checkpoint::CheckpointReader> ckpt_reader);
60 std::unique_ptr<checkpoint::CheckpointReader> ckpt_reader);
112 std::unique_ptr<checkpoint::CheckpointReader> ckpt_reader_;
|
| tfprof_tensor_test.cc | 46 std::unique_ptr<checkpoint::CheckpointReader> ckpt_reader(
47 new checkpoint::CheckpointReader(ckpt_path, status));
|
| /external/tensorflow/tensorflow/c/ |
| checkpoint_reader.h | 30 namespace checkpoint { namespace in namespace:tensorflow
35 // checkpoint::TensorSliceReader (for V1), that is more easily SWIG wrapped for
79 } // namespace checkpoint
|
| /external/tensorflow/tensorflow/contrib/checkpoint/python/ |
| python_state_test.py | 23 from tensorflow.contrib.checkpoint.python import python_state
39 saver = util.Checkpoint(numpy=save_state)
53 loader = util.Checkpoint(numpy=load_state)
73 saver = util.Checkpoint(numpy=save_state)
93 checkpoint = util.Checkpoint(numpy_arrays=arrays)
95 save_path = checkpoint.save(os.path.join(self.get_temp_dir(), "ckpt"))
97 checkpoint.restore(save_path)
100 second_checkpoint = util.Checkpoint(numpy_arrays=python_state.NumpyState())
|
| /external/perfetto/test/cts/ |
| heapprofd_test_cts.cc | 88 const auto checkpoint = task_runner->CreateCheckpoint(checkpoint_name); local
90 [desired_run_state, task_runner, app_name, checkpoint] {
92 std::move(checkpoint));
105 auto checkpoint = task_runner->CreateCheckpoint(checkpoint_name); local
106 task_runner->PostTask([desired_run_state, task_runner, app_name, checkpoint] {
108 std::move(checkpoint));
|
| /external/tensorflow/tensorflow/python/training/ |
| checkpoint_management_test.py | 74 # Collides with the default name of the checkpoint state file.
75 filepath = os.path.join(traindir, "checkpoint")
86 # Succeeds: the file will be named "checkpoint-<step>".
91 # Succeeds: the file will be named "checkpoint-<i>-of-<n>".
97 # Succeeds: the file will be named "checkpoint-<step>-<i>-of-<n>".
141 # Get the most recent checkpoint name from the training history file.
145 # Restore "v0" from that checkpoint.
221 os.path.join(save_dir, "checkpoint"))
239 ckpt_path = os.path.join(save_dir, "checkpoint")
249 ckpt_path = os.path.join(save_dir, "checkpoint")
[all...] |
