Searched refs:checkpoint (Results 1 - 25 of 101) sorted by relevance

12345

/external/perfetto/src/base/test/
H A Dtest_task_runner.cc54 void TestTaskRunner::RunUntilCheckpoint(const std::string& checkpoint, argument
57 if (checkpoints_.count(checkpoint) == 0) {
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;
80 const std::string& checkpoint) {
79 CreateCheckpoint( const std::string& checkpoint) argument
[all...]
H A Dtest_task_runner.h54 std::function<void()> CreateCheckpoint(const std::string& checkpoint);
55 void RunUntilCheckpoint(const std::string& checkpoint,
/external/tensorflow/tensorflow/core/kernels/
H A Drestore_op.cc34 preferred_shard_ = checkpoint::TensorSliceReader::kLoadAllShards;
43 RestoreTensor(context, &checkpoint::OpenTableTensorSliceReader,
60 preferred_shard_ = checkpoint::TensorSliceReader::kLoadAllShards;
69 RestoreTensor(context, &checkpoint::OpenTableTensorSliceReader,
H A Dsave_restore_tensor.h26 // 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.
H A Dsave_restore_tensor.cc40 checkpoint::TensorSliceWriter::CreateBuilderFunction builder_func,
82 checkpoint::TensorSliceWriter writer(filename_t.flat<string>()(0),
90 // checkpoint.
106 OP_REQUIRES_OK(context, checkpoint::ParseShapeAndSlice(
143 checkpoint::TensorSliceReader::OpenTableFunction open_func,
160 std::unique_ptr<checkpoint::TensorSliceReader> allocated_reader;
162 const checkpoint::TensorSliceReader* reader =
166 allocated_reader.reset(new checkpoint::TensorSliceReader(
178 "\" not found in checkpoint files ", file_pattern));
193 OP_REQUIRES_OK(context, checkpoint
[all...]
H A Dsave_op.cc34 SaveTensors(context, &checkpoint::CreateTableTensorSliceBuilder, false);
45 SaveTensors(context, &checkpoint::CreateTableTensorSliceBuilder, true);
/external/vixl/src/
H A Dpool-manager-impl.h63 // 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, argument
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) { argument
183 CheckFuturePC(T pc, T checkpoint) argument
318 T checkpoint = MaxCheckpoint<T>(); local
[all...]
/external/tensorflow/tensorflow/contrib/util/
H A Dinspect_checkpoint.cc26 tensorflow::checkpoint::TensorSliceReader reader(
27 in, tensorflow::checkpoint::OpenTableTensorSliceReader);
30 fprintf(stderr, "Unable to open the checkpoint file\n");
/external/tensorflow/tensorflow/python/training/
H A Dcheckpointable.py86 def __init__(self, checkpoint, proto_id):
87 """Specify an object within a checkpoint.
90 checkpoint: A _Checkpoint object.
93 self._checkpoint = checkpoint
106 """Set a checkpoint<->object correspondence and process slot variables.
116 checkpoint = self.checkpoint
117 current_assignment = checkpoint.object_by_proto_id.get(self._proto_id, None)
119 checkpoint.object_by_proto_id[self._proto_id] = checkpointable
121 checkpoint
189 def checkpoint(self): member in class:_CheckpointPosition
[all...]
/external/autotest/client/tests/selftest/
H A Dselftest.py16 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/
H A Dsaved_tensor_slice_util_test.cc25 namespace checkpoint { namespace in namespace:tensorflow
45 } // namespace checkpoint
H A Dtensor_slice_reader_cache.h31 namespace checkpoint { namespace in namespace:tensorflow
84 } // namespace checkpoint
H A Dtensor_slice_set.h38 namespace checkpoint { namespace in namespace:tensorflow
104 } // namespace checkpoint
H A Dtensor_slice_reader.h44 namespace checkpoint { namespace in namespace:tensorflow
56 // Abstract interface for reading data out of a tensor slice checkpoint file
186 checkpoint::TensorProtoData<T>(sts.data().data()), data);
191 } // namespace checkpoint
/external/tensorflow/tensorflow/python/util/
H A Dpy_checkpoint_reader.i27 %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
[all...]
/external/tensorflow/tensorflow/core/profiler/internal/
H A Dtfprof_stats.h57 std::unique_ptr<checkpoint::CheckpointReader> ckpt_reader);
60 std::unique_ptr<checkpoint::CheckpointReader> ckpt_reader);
112 std::unique_ptr<checkpoint::CheckpointReader> ckpt_reader_;
H A Dtfprof_tensor_test.cc46 std::unique_ptr<checkpoint::CheckpointReader> ckpt_reader(
47 new checkpoint::CheckpointReader(ckpt_path, status));
H A Dtfprof_scope.h42 explicit TFScope(checkpoint::CheckpointReader* ckpt_reader)
H A Dtfprof_show.h42 explicit TFShow(checkpoint::CheckpointReader* ckpt_reader)
121 checkpoint::CheckpointReader* ckpt_reader_;
H A Dtfprof_show_test.cc68 std::unique_ptr<checkpoint::CheckpointReader> ckpt_reader(
69 new checkpoint::CheckpointReader(ckpt_path, status));
H A Dtfprof_stats_test.cc62 std::unique_ptr<checkpoint::CheckpointReader> ckpt_reader(
63 new checkpoint::CheckpointReader(ckpt_path, status));
274 {"tensor_value"}, // Show tensor value from checkpoint.
/external/tensorflow/tensorflow/c/
H A Dcheckpoint_reader.h30 namespace checkpoint { namespace in namespace:tensorflow
35 // checkpoint::TensorSliceReader (for V1), that is more easily SWIG wrapped for
79 } // namespace checkpoint
/external/perfetto/src/ipc/
H A Dunix_socket_unittest.cc84 auto checkpoint = task_runner_.CreateCheckpoint("failure"); local
86 .WillOnce(InvokeWithoutArgs(checkpoint));
107 auto checkpoint = task_runner_.CreateCheckpoint("cli_connected"); local
110 .WillOnce(InvokeWithoutArgs(checkpoint));
230 auto checkpoint = task_runner_.CreateCheckpoint(std::to_string(i)); local
232 .WillOnce(Invoke([checkpoint](UnixSocket* s) {
234 checkpoint();
269 auto checkpoint = task_runner_.CreateCheckpoint("change_seen_by_server"); local
272 [this, tmp_fd, checkpoint, mem](UnixSocket*, UnixSocket* new_conn) {
277 .WillOnce(Invoke([checkpoint, me
293 auto checkpoint = task_runner_.CreateCheckpoint("change_seen_by_client"); local
[all...]
/external/tensorflow/tensorflow/contrib/eager/python/
H A Dcheckpointable_utils.py40 # Keyword for identifying that the next bit of a checkpoint variable name is a
45 # Where <path to variable> is a full path from the checkpoint root to the
49 # attribute in checkpoint names. Used like:
139 # where <variable name> is exactly the checkpoint name used for the original
140 # variable, including the path from the checkpoint root and the local name in
237 """Determine checkpoint keys for variables and build a serialized graph.
289 """Save a training checkpoint.
292 file_prefix: A prefix to use for the checkpoint filenames
295 root_checkpointable: A Checkpointable object to save. The checkpoint
307 The full path to the checkpoint
[all...]
/external/tensorflow/tensorflow/contrib/lookup/
H A Dlookup_ops.py313 checkpoint=True):
326 checkpoint: if True, the contents of the table are saved to and restored
334 ValueError: If checkpoint is True and no name was specified.
343 use_node_name_sharing = checkpoint and shared_name is None
365 if checkpoint:
519 checkpoint=True):
535 checkpoint: if True, the contents of the table are saved to and restored
543 ValueError: If checkpoint is True and no name was specified.
552 use_node_name_sharing = checkpoint and shared_name is None
567 if checkpoint
[all...]

Completed in 473 milliseconds

12345