SYMBOL INDEX (1253 symbols across 163 files) FILE: csrc/loader/chunk_source/chunk_source.h function namespace (line 10) | namespace lczero { FILE: csrc/loader/chunk_source/chunk_source_view.h function namespace (line 12) | namespace lczero { FILE: csrc/loader/chunk_source/debug_chunk_source.cc type lczero (line 13) | namespace lczero { type training (line 14) | namespace training { FILE: csrc/loader/chunk_source/debug_chunk_source.h function namespace (line 11) | namespace lczero { FILE: csrc/loader/chunk_source/rawfile_chunk_source.cc type lczero (line 12) | namespace lczero { type training (line 13) | namespace training { FILE: csrc/loader/chunk_source/rawfile_chunk_source.h function namespace (line 9) | namespace lczero { FILE: csrc/loader/chunk_source/tar_chunk_source.cc type lczero (line 19) | namespace lczero { type training (line 20) | namespace training { type TarHeader (line 22) | struct TarHeader { function ParseOctal (line 43) | uint64_t ParseOctal(const std::array& octal) { function ReadExact (line 52) | bool ReadExact(int fd, off_t offset, void* buffer, size_t size) { function ReadGzipPrefix (line 64) | std::optional ReadGzipPrefix(int fd, off_t offset, size... FILE: csrc/loader/chunk_source/tar_chunk_source.h function namespace (line 12) | namespace lczero { FILE: csrc/loader/data_loader.cc type lczero (line 15) | namespace lczero { type training (line 16) | namespace training { function DataLoaderConfig (line 17) | DataLoaderConfig DataLoader::ParseConfig(const std::string& serializ... function TensorTuple (line 102) | TensorTuple DataLoader::GetNext(std::string_view alias) { FILE: csrc/loader/data_loader.h function namespace (line 18) | namespace lczero { FILE: csrc/loader/data_loader_metrics.cc type lczero (line 11) | namespace lczero { type training (line 12) | namespace training { function ProtoT (line 16) | ProtoT* FindByName(std::vector* entries, absl::string_view n... function UpdateFrom (line 25) | void UpdateFrom(QueueMetricProto& dest, const QueueMetricProto& src) { function UpdateFrom (line 34) | void UpdateFrom(CountMetricProto& dest, const CountMetricProto& src) { function UpdateFrom (line 39) | void UpdateFrom(GaugeMetricProto& dest, const GaugeMetricProto& src) { function UpdateFrom (line 45) | void UpdateFrom(StageMetricProto& dest, const StageMetricProto& src) { function UpdateFrom (line 108) | void UpdateFrom(DataLoaderMetricsProto& dest, FILE: csrc/loader/data_loader_metrics.h function namespace (line 13) | namespace lczero { FILE: csrc/loader/data_loader_test.cc type lczero (line 7) | namespace lczero { type training (line 8) | namespace training { function TEST (line 10) | TEST(DataLoaderTest, AllowsNoOutputsConfigured) { function TEST (line 19) | TEST(DataLoaderTest, ThrowsOnDuplicateStageName) { FILE: csrc/loader/frame_type.h function namespace (line 32) | namespace lczero { FILE: csrc/loader/loader_main.cpp type lczero (line 25) | namespace lczero { type training (line 26) | namespace training { function Run (line 28) | void Run() { function main (line 115) | int main(int argc, char* argv[]) { FILE: csrc/loader/pybind_module.cc type lczero (line 27) | namespace lczero { type training (line 28) | namespace training { function SerializePyProto (line 32) | std::string SerializePyProto(const py::handle& obj, const char* expe... function ProtoT (line 45) | ProtoT ParsePyProto(const py::handle& obj, const char* expected_type) { function MakePythonProto (line 51) | py::object MakePythonProto(const char* module_name, const char* mess... function tensor_to_numpy (line 62) | py::array tensor_to_numpy(std::unique_ptr tensor) { function tensor_tuple_to_numpy_tuple (line 75) | py::tuple tensor_tuple_to_numpy_tuple(TensorTuple tensor_tuple) { function PYBIND11_MODULE (line 83) | PYBIND11_MODULE(_lczero_training, m) { FILE: csrc/loader/stages/chunk_rescorer.cc type lczero (line 11) | namespace lczero { type training (line 12) | namespace training { function V6ToV7 (line 15) | void V6ToV7(std::span data, float theta = 5.0f / 6.0f) { function StageMetricProto (line 158) | StageMetricProto ChunkRescorer::FlushMetrics() { FILE: csrc/loader/stages/chunk_rescorer.h function namespace (line 23) | namespace lczero { FILE: csrc/loader/stages/chunk_rescorer_test.cc type lczero (line 12) | namespace lczero { type training (line 13) | namespace training { class PassthroughStage (line 18) | class PassthroughStage : public Stage { method PassthroughStage (line 20) | explicit PassthroughStage(Queue* queue) : queue_(queue) {} method Start (line 22) | void Start() override {} method Stop (line 23) | void Stop() override {} method StageMetricProto (line 24) | StageMetricProto FlushMetrics() override { return StageMetricProto... method QueueBase (line 25) | QueueBase* GetOutput(std::string_view name = "") override { method SetInputs (line 29) | void SetInputs(absl::Span inputs) override { class ChunkRescorerTest (line 41) | class ChunkRescorerTest : public ::testing::Test { method SetUp (line 43) | void SetUp() override { method TrainingChunk (line 54) | TrainingChunk MakeChunk(std::vector frames, function TEST_F (line 69) | TEST_F(ChunkRescorerTest, HandlesInputQueueClosure) { FILE: csrc/loader/stages/chunk_source_loader.cc type lczero (line 12) | namespace lczero { type training (line 13) | namespace training { function CreateChunkSourceFromFile (line 15) | std::unique_ptr CreateChunkSourceFromFile( function StageMetricProto (line 163) | StageMetricProto ChunkSourceLoader::FlushMetrics() { FILE: csrc/loader/stages/chunk_source_loader.h function namespace (line 19) | namespace lczero { FILE: csrc/loader/stages/chunk_source_loader_test.cc type lczero (line 10) | namespace lczero { type training (line 11) | namespace training { class PassthroughStage (line 16) | class PassthroughStage : public Stage { method PassthroughStage (line 18) | explicit PassthroughStage(Queue* queue) : queue_(queue) {} method Start (line 20) | void Start() override {} method Stop (line 21) | void Stop() override {} method StageMetricProto (line 22) | StageMetricProto FlushMetrics() override { return StageMetricProto... method QueueBase (line 23) | QueueBase* GetOutput(std::string_view name = "") override { method SetInputs (line 27) | void SetInputs(absl::Span inputs) override { function TEST (line 39) | TEST(ChunkSourceLoaderTest, ProcessesFiles) { function TEST (line 73) | TEST(ChunkSourceLoaderTest, HandlesPhases) { function TEST (line 105) | TEST(ChunkSourceLoaderTest, PassesThroughInitialScanComplete) { function TEST (line 136) | TEST(ChunkSourceLoaderTest, SentinelBarrierWithMultipleThreads) { FILE: csrc/loader/stages/chunk_source_splitter.cc type lczero (line 11) | namespace lczero { type training (line 12) | namespace training { function QueueBase (line 83) | QueueBase* ChunkSourceSplitter::GetOutput(std::string_view name) { function StageMetricProto (line 93) | StageMetricProto ChunkSourceSplitter::FlushMetrics() { FILE: csrc/loader/stages/chunk_source_splitter.h function namespace (line 22) | namespace lczero { FILE: csrc/loader/stages/chunk_source_splitter_test.cc type lczero (line 16) | namespace lczero { type training (line 17) | namespace training { class FixedCountChunkSource (line 21) | class FixedCountChunkSource : public ChunkSource { method FixedCountChunkSource (line 23) | FixedCountChunkSource(std::string sort_key, size_t count) method GetChunkSortKey (line 27) | std::string GetChunkSortKey() const override { return key_; } method GetChunkCount (line 28) | size_t GetChunkCount() const override { return count_; } method GetChunkData (line 29) | std::optional> GetChunkData(size_t) override { class PassthroughStage (line 38) | class PassthroughStage : public Stage { method PassthroughStage (line 40) | explicit PassthroughStage(Queue* queue) : queue_(queue) {} method Start (line 42) | void Start() override {} method Stop (line 43) | void Stop() override {} method StageMetricProto (line 44) | StageMetricProto FlushMetrics() override { return StageMetricProto... method QueueBase (line 45) | QueueBase* GetOutput(std::string_view) override { return queue_; } method SetInputs (line 46) | void SetInputs(absl::Span inputs) override { function TEST (line 58) | TEST(ChunkSourceSplitterTest, SplitsByHashAndWeight) { function TEST (line 122) | TEST(ChunkSourceSplitterTest, BroadcastsInitialScanComplete) { FILE: csrc/loader/stages/chunk_unpacker.cc type lczero (line 26) | namespace lczero { type training (line 27) | namespace training { function PickSampledPositions (line 35) | std::vector PickSampledPositions(int32_t n, double p, function SampleProbabilisticSequence (line 70) | std::vector SampleProbabilisticSequence( function GenerateRunSeed (line 98) | uint32_t GenerateRunSeed() { function QueueBase (line 173) | QueueBase* ChunkUnpacker::GetOutput(std::string_view name) { function FramesToProbabilities (line 188) | std::vector FramesToProbabilities(std::span ... function StageMetricProto (line 271) | StageMetricProto ChunkUnpacker::FlushMetrics() { FILE: csrc/loader/stages/chunk_unpacker.h function namespace (line 22) | namespace lczero { FILE: csrc/loader/stages/chunk_unpacker_test.cc type lczero (line 18) | namespace lczero { type training (line 19) | namespace training { class PassthroughStage (line 24) | class PassthroughStage : public Stage { method PassthroughStage (line 26) | explicit PassthroughStage(Queue* queue) : queue_(queue) {} method Start (line 28) | void Start() override {} method Stop (line 29) | void Stop() override {} method StageMetricProto (line 30) | StageMetricProto FlushMetrics() override { return StageMetricProto... method QueueBase (line 31) | QueueBase* GetOutput(std::string_view name = "") override { method SetInputs (line 35) | void SetInputs(absl::Span inputs) override { class ChunkUnpackerTest (line 47) | class ChunkUnpackerTest : public ::testing::Test { method SetUp (line 49) | void SetUp() override { method FrameType (line 56) | FrameType CreateTestFrame(uint32_t version) { method TrainingChunk (line 64) | TrainingChunk MakeChunk(std::vector frames, function TEST_F (line 79) | TEST_F(ChunkUnpackerTest, UnpacksSingleFrame) { function TEST_F (line 95) | TEST_F(ChunkUnpackerTest, UnpacksMultipleFrames) { function TEST_F (line 126) | TEST_F(ChunkUnpackerTest, UnpacksMultipleChunks) { function TEST_F (line 160) | TEST_F(ChunkUnpackerTest, HandlesEmptyChunk) { function TEST_F (line 176) | TEST_F(ChunkUnpackerTest, HandlesQueueClosure) { function TEST (line 188) | TEST(PickSampledPositionsTest, Deterministic) { function TEST (line 198) | TEST(PickSampledPositionsTest, FullBucketFirstRound) { function TEST (line 207) | TEST(PickSampledPositionsTest, DisjointBuckets) { function TEST (line 227) | TEST(PickSampledPositionsTest, PartialBucketElementsAreReturned) { function TEST (line 256) | TEST(PickSampledPositionsTest, PartialBucketCompletedSize) { FILE: csrc/loader/stages/file_path_provider.cc type lczero (line 24) | namespace lczero { type training (line 25) | namespace training { function ShouldSkipName (line 29) | bool ShouldSkipName(std::string_view name) { function ShouldSkipPathEntry (line 33) | bool ShouldSkipPathEntry(const FilePathProvider::Path& path) { function StageMetricProto (line 85) | StageMetricProto FilePathProvider::FlushMetrics() { type inotify_event (line 200) | struct inotify_event type inotify_event (line 201) | struct inotify_event type inotify_event (line 219) | struct inotify_event type epoll_event (line 272) | struct epoll_event type epoll_event (line 288) | struct epoll_event type inotify_event (line 319) | struct inotify_event type inotify_event (line 320) | struct inotify_event type inotify_event (line 324) | struct inotify_event type inotify_event (line 331) | struct inotify_event FILE: csrc/loader/stages/file_path_provider.h function FilePathProviderMessageType (line 29) | enum class FilePathProviderMessageType { FILE: csrc/loader/stages/file_path_provider_main.cc function main (line 15) | int main(int argc, char* argv[]) { FILE: csrc/loader/stages/file_path_provider_test.cc type lczero (line 12) | namespace lczero { type training (line 13) | namespace training { function FilePathProviderConfig (line 17) | FilePathProviderConfig MakeConfig(const std::filesystem::path& direc... function RelativeTo (line 24) | std::string RelativeTo(const std::filesystem::path& base, class FilePathProviderTest (line 31) | class FilePathProviderTest : public ::testing::Test { method SetUp (line 33) | void SetUp() override { method TearDown (line 42) | void TearDown() override { method CreateFile (line 48) | void CreateFile(const std::filesystem::path& path, method CreateDirectory (line 55) | void CreateDirectory(const std::filesystem::path& path) { method DrainInitialScan (line 59) | std::vector DrainInitialScan( method AwaitNextFile (line 78) | FilePathProvider::File AwaitNextFile(Queue... method FilePathProviderConfig (line 92) | FilePathProviderConfig Config() const { return MakeConfig(test_dir... function TEST_F (line 97) | TEST_F(FilePathProviderTest, ConstructorCreatesQueue) { function TEST_F (line 113) | TEST_F(FilePathProviderTest, InitialScanFindsVisibleFiles) { function TEST_F (line 136) | TEST_F(FilePathProviderTest, InitialScanSkipsHiddenEntries) { function TEST_F (line 160) | TEST_F(FilePathProviderTest, DetectsNewVisibleFile) { function TEST_F (line 174) | TEST_F(FilePathProviderTest, DetectsFilesInPreExistingSubdirectory) { function TEST_F (line 191) | TEST_F(FilePathProviderTest, IgnoresHiddenFileEvents) { function TEST_F (line 207) | TEST_F(FilePathProviderTest, SkipsHiddenDirectoryRecursion) { function TEST_F (line 223) | TEST_F(FilePathProviderTest, HandlesEmptyDirectory) { FILE: csrc/loader/stages/join_stage.cc type lczero (line 10) | namespace lczero { type training (line 11) | namespace training { function StageMetricProto (line 71) | StageMetricProto JoinStage::FlushMetrics() { class JoinStage (line 86) | class JoinStage FILE: csrc/loader/stages/join_stage.h function namespace (line 18) | namespace lczero { FILE: csrc/loader/stages/join_stage_test.cc type lczero (line 12) | namespace lczero { type training (line 13) | namespace training { class JoinStageTest (line 15) | class JoinStageTest : public ::testing::Test { method SetUp (line 17) | void SetUp() override { config_.mutable_output()->set_queue_capaci... method FrameType (line 19) | FrameType CreateTestFrame(uint32_t version) { function TEST_F (line 30) | TEST_F(JoinStageTest, JoinsTwoInputs) { function TEST_F (line 63) | TEST_F(JoinStageTest, JoinsThreeInputs) { function TEST_F (line 98) | TEST_F(JoinStageTest, HandlesEmptyInputs) { function TEST_F (line 118) | TEST_F(JoinStageTest, FlushesMetrics) { FILE: csrc/loader/stages/position_sampling.cc type lczero (line 5) | namespace lczero { type training (line 6) | namespace training { function ComputePositionSamplingWeight (line 8) | float ComputePositionSamplingWeight(const FrameType& frame, FILE: csrc/loader/stages/position_sampling.h function namespace (line 6) | namespace lczero { FILE: csrc/loader/stages/shuffling_chunk_pool.cc type lczero (line 27) | namespace lczero { type training (line 28) | namespace training { function QueueBase (line 90) | QueueBase* ShufflingChunkPool::GetOutput(std::string_view name) { type ShufflingChunkPool::ChunkData (line 478) | struct ShufflingChunkPool::ChunkData { function ABSL_EXCLUSIVE_LOCKS_REQUIRED (line 659) | ABSL_EXCLUSIVE_LOCKS_REQUIRED(chunk_sources_mutex_) { function StageMetricProto (line 719) | StageMetricProto ShufflingChunkPool::FlushMetrics() { FILE: csrc/loader/stages/shuffling_chunk_pool.h function namespace (line 31) | namespace lczero { FILE: csrc/loader/stages/shuffling_chunk_pool_test.cc type lczero (line 22) | namespace lczero { type training (line 23) | namespace training { class PassthroughStage (line 28) | class PassthroughStage : public Stage { method PassthroughStage (line 30) | explicit PassthroughStage(Queue* queue) : queue_(queue) {} method Start (line 32) | void Start() override {} method Stop (line 33) | void Stop() override {} method StageMetricProto (line 34) | StageMetricProto FlushMetrics() override { return StageMetricProto... method QueueBase (line 35) | QueueBase* GetOutput(std::string_view name = "") override { method SetInputs (line 39) | void SetInputs(absl::Span inputs) override { class MockChunkSource (line 52) | class MockChunkSource : public ChunkSource { method MockChunkSource (line 54) | MockChunkSource(const std::string& sort_key, size_t chunk_count) method GetChunkSortKey (line 57) | std::string GetChunkSortKey() const override { return sort_key_; } method GetChunkCount (line 58) | size_t GetChunkCount() const override { return chunk_count_; } method GetChunkData (line 60) | std::optional> GetChunkData(size_t index) o... class InvalidChunkSource (line 75) | class InvalidChunkSource : public ChunkSource { method InvalidChunkSource (line 77) | explicit InvalidChunkSource(std::string sort_key) method GetChunkSortKey (line 80) | std::string GetChunkSortKey() const override { return sort_key_; } method GetChunkCount (line 81) | size_t GetChunkCount() const override { return 2; } method GetChunkData (line 83) | std::optional> GetChunkData(size_t index) o... class ShufflingChunkPoolTest (line 99) | class ShufflingChunkPoolTest : public ::testing::Test { method SetUp (line 101) | void SetUp() override { method TearDown (line 107) | void TearDown() override { method AddMockChunkSourceToQueue (line 113) | void AddMockChunkSourceToQueue(const std::string& sort_key, method MarkInitialScanComplete (line 123) | void MarkInitialScanComplete() { method CloseInputQueue (line 130) | void CloseInputQueue() { method ShufflingChunkPoolConfig (line 134) | ShufflingChunkPoolConfig MakeConfig(int chunk_pool_size, function TEST_F (line 150) | TEST_F(ShufflingChunkPoolTest, ConstructorCreatesOutputQueue) { function TEST_F (line 180) | TEST_F(ShufflingChunkPoolTest, HandlesEmptyInputQueue) { function TEST_F (line 208) | TEST_F(ShufflingChunkPoolTest, FlushMetricsHandlesEmptyChunkSources) { function TEST_F (line 236) | TEST_F(ShufflingChunkPoolTest, FlushMetricsReportsWindowAndTotalCoun... function TEST_F (line 282) | TEST_F(ShufflingChunkPoolTest, ProcessesInitialScanChunkSources) { function TEST_F (line 306) | TEST_F(ShufflingChunkPoolTest, OutputWorkerProducesChunks) { function TEST_F (line 342) | TEST_F(ShufflingChunkPoolTest, DropsInvalidChunks) { function TEST_F (line 390) | TEST_F(ShufflingChunkPoolTest, NewChunkSourceProcessing) { function TEST_F (line 421) | TEST_F(ShufflingChunkPoolTest, ChunkWindowManagement) { function TEST_F (line 446) | TEST_F(ShufflingChunkPoolTest, ChunkSorting) { function TEST_F (line 470) | TEST_F(ShufflingChunkPoolTest, StreamShufflerResetWhenExhausted) { function TEST_F (line 530) | TEST_F(ShufflingChunkPoolTest, HanseMetrics_NoRejection_CacheAndResh... function TEST_F (line 576) | TEST_F(ShufflingChunkPoolTest, ExplicitClose) { function TEST_F (line 610) | TEST_F(ShufflingChunkPoolTest, CloseStopsOutputWorkers) { function TEST_F (line 646) | TEST_F(ShufflingChunkPoolTest, CloseIsIdempotent) { function TEST_F (line 666) | TEST_F(ShufflingChunkPoolTest, DestructorCallsClose) { function TEST_F (line 696) | TEST_F(ShufflingChunkPoolTest, InputQueueClosureDoesNotCloseOutputQu... function TEST_F (line 726) | TEST_F(ShufflingChunkPoolTest, BasicAnchorFunctionality) { function TEST_F (line 753) | TEST_F(ShufflingChunkPoolTest, ResetAnchor) { function TEST_F (line 776) | TEST_F(ShufflingChunkPoolTest, AnchorCounterIncrement) { function TEST_F (line 810) | TEST_F(ShufflingChunkPoolTest, AnchorCounterResetDuringInitialLoad) { FILE: csrc/loader/stages/shuffling_frame_sampler.cc type lczero (line 12) | namespace lczero { type training (line 13) | namespace training { function StageMetricProto (line 94) | StageMetricProto ShufflingFrameSampler::FlushMetrics() { FILE: csrc/loader/stages/shuffling_frame_sampler.h function namespace (line 21) | namespace lczero { FILE: csrc/loader/stages/shuffling_frame_sampler_test.cc type lczero (line 10) | namespace lczero { type training (line 11) | namespace training { class PassthroughStage (line 16) | class PassthroughStage : public Stage { method PassthroughStage (line 18) | explicit PassthroughStage(Queue* queue) : queue_(queue) {} method Start (line 20) | void Start() override {} method Stop (line 21) | void Stop() override {} method StageMetricProto (line 22) | StageMetricProto FlushMetrics() override { return StageMetricProto... method QueueBase (line 23) | QueueBase* GetOutput(std::string_view name = "") override { method SetInputs (line 27) | void SetInputs(absl::Span inputs) override { class ShufflingFrameSamplerTest (line 39) | class ShufflingFrameSamplerTest : public ::testing::Test { method SetUp (line 41) | void SetUp() override { method FrameType (line 47) | FrameType CreateTestFrame(uint32_t version) { function TEST_F (line 59) | TEST_F(ShufflingFrameSamplerTest, OutputsNoFramesWithSmallInput) { function TEST_F (line 88) | TEST_F(ShufflingFrameSamplerTest, OutputsFramesWithLargeInput) { function TEST_F (line 124) | TEST_F(ShufflingFrameSamplerTest, HandlesEmptyInput) { function TEST_F (line 136) | TEST_F(ShufflingFrameSamplerTest, HandlesExactReservoirSize) { function TEST_F (line 166) | TEST_F(ShufflingFrameSamplerTest, PreservesFrameData) { FILE: csrc/loader/stages/simple_chunk_extractor.cc type lczero (line 10) | namespace lczero { type training (line 11) | namespace training { function StageMetricProto (line 89) | StageMetricProto SimpleChunkExtractor::FlushMetrics() { FILE: csrc/loader/stages/simple_chunk_extractor.h function namespace (line 20) | namespace lczero { FILE: csrc/loader/stages/simple_chunk_extractor_test.cc type lczero (line 17) | namespace lczero { type training (line 18) | namespace training { class MockChunkSource (line 22) | class MockChunkSource : public ChunkSource { method MockChunkSource (line 24) | MockChunkSource(std::string sort_key, size_t chunk_count) method GetChunkSortKey (line 32) | std::string GetChunkSortKey() const override { return sort_key_; } method GetChunkCount (line 33) | size_t GetChunkCount() const override { return chunk_count_; } method GetChunkData (line 35) | std::optional> GetChunkData(size_t index) o... class SimpleChunkExtractorTest (line 46) | class SimpleChunkExtractorTest : public ::testing::Test { method SetUp (line 48) | void SetUp() override { method TearDown (line 67) | void TearDown() override { class DummyStage (line 75) | class DummyStage : public Stage { method DummyStage (line 77) | explicit DummyStage(QueueBase* queue) : queue_(queue) {} method Start (line 78) | void Start() override {} method Stop (line 79) | void Stop() override {} method StageMetricProto (line 80) | StageMetricProto FlushMetrics() override { return {}; } method QueueBase (line 81) | QueueBase* GetOutput(std::string_view name = "") override { function TEST_F (line 95) | TEST_F(SimpleChunkExtractorTest, ProcessesSingleSource) { function TEST_F (line 137) | TEST_F(SimpleChunkExtractorTest, ProcessesMultipleSources) { function TEST_F (line 179) | TEST_F(SimpleChunkExtractorTest, SkipsNonFileMessages) { function TEST_F (line 210) | TEST_F(SimpleChunkExtractorTest, MetricsAreRecorded) { FILE: csrc/loader/stages/stage.cc type lczero (line 5) | namespace lczero { type training (line 6) | namespace training { function QueueBase (line 20) | QueueBase* StageRegistry::GetStageOutput(std::string_view stage_name... FILE: csrc/loader/stages/stage.h function namespace (line 17) | namespace lczero { FILE: csrc/loader/stages/stage_factory.cc type lczero (line 17) | namespace lczero { type training (line 18) | namespace training { function CountStageConfigs (line 22) | int CountStageConfigs(const StageConfig& config) { function CreateStage (line 37) | std::unique_ptr CreateStage(const StageConfig& config) { FILE: csrc/loader/stages/stage_factory.h function namespace (line 8) | namespace lczero { FILE: csrc/loader/stages/stage_factory_test.cc type lczero (line 7) | namespace lczero { type training (line 8) | namespace training { function TEST (line 10) | TEST(StageFactoryTest, CreatesFilePathProviderStage) { function TEST (line 20) | TEST(StageFactoryTest, ThrowsWhenNoStageConfigSet) { function TEST (line 26) | TEST(StageFactoryTest, ThrowsWhenMultipleStageConfigsSet) { FILE: csrc/loader/stages/tensor_generator.cc type lczero (line 18) | namespace lczero { type training (line 19) | namespace training { function TensorTuple (line 85) | TensorTuple TensorGenerator::ConvertFramesToTensors( function StageMetricProto (line 202) | StageMetricProto TensorGenerator::FlushMetrics() { FILE: csrc/loader/stages/tensor_generator.h function namespace (line 21) | namespace lczero { FILE: csrc/loader/stages/tensor_generator_test.cc type lczero (line 15) | namespace lczero { type training (line 16) | namespace training { class PassthroughStage (line 21) | class PassthroughStage : public Stage { method PassthroughStage (line 23) | explicit PassthroughStage(Queue* queue) : queue_(queue) {} method Start (line 25) | void Start() override {} method Stop (line 26) | void Stop() override {} method StageMetricProto (line 27) | StageMetricProto FlushMetrics() override { return StageMetricProto... method QueueBase (line 28) | QueueBase* GetOutput(std::string_view name = "") override { method SetInputs (line 32) | void SetInputs(absl::Span inputs) override { class TensorGeneratorTest (line 44) | class TensorGeneratorTest : public ::testing::Test { method SetUp (line 46) | void SetUp() override { method FrameType (line 53) | FrameType CreateTestFrame() { method VerifyTensorTuple (line 91) | void VerifyTensorTuple(const TensorTuple& tensors, method VerifyTensorData (line 126) | void VerifyTensorData(const TensorTuple& tensors, function TEST_F (line 200) | TEST_F(TensorGeneratorTest, GeneratesCorrectTensorShapes) { function TEST_F (line 217) | TEST_F(TensorGeneratorTest, GeneratesCorrectTensorData) { function TEST_F (line 235) | TEST_F(TensorGeneratorTest, HandlesMultipleBatches) { function TEST_F (line 270) | TEST_F(TensorGeneratorTest, HandlesDifferentBatchSizes) { function TEST_F (line 288) | TEST_F(TensorGeneratorTest, HandlesEmptyInput) { function TEST_F (line 300) | TEST_F(TensorGeneratorTest, VerifiesPlanesConversion) { function TEST_F (line 339) | TEST_F(TensorGeneratorTest, VerifiesQDConversion) { FILE: csrc/loader/stages/training_chunk.h function namespace (line 10) | namespace lczero { FILE: csrc/tools/dump_chunk_main.cc type lczero (line 28) | namespace lczero { type training (line 29) | namespace training { function DumpChunk (line 36) | void DumpChunk(const std::string& path, int64_t max_entries, function main (line 80) | int main(int argc, char** argv) { FILE: csrc/tools/filter_chunks_main.cc function CollectTarFiles (line 48) | std::vector CollectTarFiles(const fs::path& directory) { function ParsePlaneValues (line 62) | std::vector ParsePlaneValues(absl::string_view value_list) { function PlanesMatch (line 89) | bool PlanesMatch(const FrameType& entry, absl::Span expe... function FindMatchingFrameIndex (line 95) | std::optional FindMatchingFrameIndex( function WriteChunk (line 103) | void WriteChunk(const fs::path& output_dir, absl::string_view base_name, function ProcessTar (line 138) | void ProcessTar(const fs::path& tar_path, const fs::path& output_dir, function main (line 170) | int main(int argc, char** argv) { FILE: csrc/tools/position_weight_stats_main.cc type WeightedPosition (line 43) | struct WeightedPosition { function CollectTarFiles (line 48) | std::vector CollectTarFiles(const fs::path& directory) { function CollectWeights (line 62) | std::vector CollectWeights(const fs::path& tar_path, function PrintHistogram (line 99) | void PrintHistogram(const std::vector& sorted_weights) { function PrintPercentiles (line 139) | void PrintPercentiles(const std::vector& sorted_weights) { function PrintStatistics (line 149) | void PrintStatistics(const std::vector& weights) { function main (line 165) | int main(int argc, char** argv) { FILE: csrc/tools/rescore_chunk_main.cc function ReadChunkFrames (line 44) | std::vector ReadChunkFrames(const fs::path& path) { function WriteChunkFrames (line 54) | void WriteChunkFrames(const fs::path& path, function BuildOutputPath (line 63) | fs::path BuildOutputPath(const fs::path& input_path) { function main (line 73) | int main(int argc, char** argv) { FILE: csrc/tools/result_distribution_main.cc type ChunkResult (line 41) | enum class ChunkResult { kWin, kDraw, kLoss } type ResultCounts (line 43) | struct ResultCounts { class CsvWriter (line 49) | class CsvWriter { method CsvWriter (line 51) | CsvWriter(std::ostream* output, absl::Mutex* mutex) method Write (line 54) | void Write(absl::string_view basename, const ResultCounts& counts) con... function DetermineChunkResult (line 79) | std::optional DetermineChunkResult(absl::string_view chunk_... function ResultCounts (line 113) | ResultCounts CountResultsInTar(const fs::path& tar_path) { function main (line 150) | int main(int argc, char** argv) { FILE: csrc/tools/startpos_policy_distribution_main.cc function MatchesStartPosition (line 56) | bool MatchesStartPosition(const FrameType& data) { function CollectTarFiles (line 62) | std::vector CollectTarFiles(const fs::path& directory) { function WriteHeader (line 76) | void WriteHeader(std::ostream& output) { function WriteRow (line 82) | void WriteRow(std::ostream& output, absl::string_view sort_key, size_t i... function ProcessTarFile (line 91) | void ProcessTarFile(const fs::path& tar_path, std::ostream& output) { function main (line 119) | int main(int argc, char** argv) { FILE: csrc/utils/gz.cc type lczero (line 9) | namespace lczero { type training (line 10) | namespace training { function GunzipBuffer (line 12) | std::string GunzipBuffer(std::string_view buffer) { FILE: csrc/utils/gz.h function namespace (line 7) | namespace lczero { FILE: csrc/utils/metrics/exponential_aggregator.h type class (line 18) | enum class function GetBucketIndex (line 147) | static size_t GetBucketIndex(TimePeriod period) { function one_tick (line 318) | auto one_tick = [&](Metric& carry) { FILE: csrc/utils/metrics/group.h function namespace (line 7) | namespace lczero { FILE: csrc/utils/metrics/load_metric.h function namespace (line 11) | namespace lczero { function UpdateFrom (line 81) | inline void UpdateFrom(LoadMetricProto& dest, const LoadMetricProto& src) { function class (line 88) | class LoadMetricPauser { function DoNotResume (line 101) | void DoNotResume() { should_resume_ = false; } FILE: csrc/utils/metrics/load_metric_test.cc type lczero (line 11) | namespace lczero { type training (line 12) | namespace training { class LoadMetricTest (line 14) | class LoadMetricTest : public ::testing::Test { method SetUp (line 18) | void SetUp() override { start_time_ = Clock::now(); } function TEST_F (line 23) | TEST_F(LoadMetricTest, BasicLoadMetricProto) { function TEST_F (line 44) | TEST_F(LoadMetricTest, LoadMetricUpdaterBasic) { function TEST_F (line 75) | TEST_F(LoadMetricTest, LoadMetricUpdaterFlush) { function TEST_F (line 97) | TEST_F(LoadMetricTest, LoadMetricProtoMerging) { function TEST_F (line 126) | TEST_F(LoadMetricTest, LoadMetricProtoMoveSemantics) { function TEST_F (line 166) | TEST_F(LoadMetricTest, LoadUtilizationTracking) { class LoadMetricProtoIntegrationTest (line 214) | class LoadMetricProtoIntegrationTest : public ::testing::Test { method SetUp (line 220) | void SetUp() override { function TEST_F (line 233) | TEST_F(LoadMetricProtoIntegrationTest, RecordMetricsWithUpdater) { function TEST_F (line 253) | TEST_F(LoadMetricProtoIntegrationTest, MultipleRecordMetrics) { function TEST_F (line 276) | TEST_F(LoadMetricProtoIntegrationTest, AdvanceTest) { function TEST_F (line 300) | TEST_F(LoadMetricTest, LoadStartStopReturnValues) { function main (line 322) | int main(int argc, char** argv) { FILE: csrc/utils/metrics/printer.h function namespace (line 8) | namespace lczero { FILE: csrc/utils/metrics/statistics_metric.h function namespace (line 7) | namespace lczero { FILE: csrc/utils/metrics/stats_test.cc type lczero (line 12) | namespace lczero { class CounterMetric (line 14) | class CounterMetric { method CounterMetric (line 16) | CounterMetric() : count_(0) {} method CounterMetric (line 17) | CounterMetric(int count) : count_(count) {} method Reset (line 19) | void Reset() { count_ = 0; } method MergeFrom (line 21) | void MergeFrom(const CounterMetric& other) { count_ += other.count_; } method Print (line 23) | void Print(MetricPrinter& printer) const { method count (line 29) | int count() const { return count_; } method set_count (line 30) | void set_count(int count) { count_ = count; } class AverageMetric (line 36) | class AverageMetric { method AverageMetric (line 38) | AverageMetric() : sum_(0), count_(0) {} method AverageMetric (line 39) | AverageMetric(double sum, int count) : sum_(sum), count_(count) {} method Reset (line 41) | void Reset() { method MergeFrom (line 46) | void MergeFrom(const AverageMetric& other) { method Print (line 51) | void Print(MetricPrinter& printer) const { method average (line 61) | double average() const { return count_ > 0 ? sum_ / count_ : 0.0; } method add_sample (line 62) | void add_sample(double value) { method sum (line 67) | double sum() const { return sum_; } method count (line 68) | int count() const { return count_; } class MaxMetric (line 75) | class MaxMetric { method MaxMetric (line 77) | MaxMetric() : max_value_(0), has_value_(false) {} method MaxMetric (line 78) | MaxMetric(double max_value) : max_value_(max_value), has_value_(true... method Reset (line 80) | void Reset() { method MergeFrom (line 85) | void MergeFrom(const MaxMetric& other) { method Print (line 94) | void Print(MetricPrinter& printer) const { method max_value (line 105) | double max_value() const { return max_value_; } method has_value (line 106) | bool has_value() const { return has_value_; } method set_value (line 107) | void set_value(double value) { class OptionalValueMetric (line 120) | class OptionalValueMetric { method OptionalValueMetric (line 122) | OptionalValueMetric() : value_(std::nullopt) {} method OptionalValueMetric (line 123) | OptionalValueMetric(int value) : value_(value) {} method Reset (line 125) | void Reset() { value_ = std::nullopt; } method MergeFrom (line 127) | void MergeFrom(const OptionalValueMetric& other) { method Print (line 134) | void Print(MetricPrinter& printer) const { method value (line 145) | std::optional value() const { return value_; } method has_value (line 146) | bool has_value() const { return value_.has_value(); } method set_value (line 147) | void set_value(int value) { value_ = value; } class MetricGroupTest (line 154) | class MetricGroupTest : public ::testing::Test { function TEST_F (line 160) | TEST_F(MetricGroupTest, InitialState) { function TEST_F (line 167) | TEST_F(MetricGroupTest, GetMutable) { function TEST_F (line 183) | TEST_F(MetricGroupTest, Reset) { function TEST_F (line 197) | TEST_F(MetricGroupTest, MergeFromGroup) { function TEST_F (line 218) | TEST_F(MetricGroupTest, MergeFromSingleMetric) { function TEST_F (line 232) | TEST_F(MetricGroupTest, Print) { class MetricPrinterTest (line 251) | class MetricPrinterTest : public ::testing::Test {} function TEST_F (line 253) | TEST_F(MetricPrinterTest, StringMetricPrinter) { function TEST_F (line 265) | TEST_F(MetricPrinterTest, MultipleGroups) { function TEST_F (line 280) | TEST_F(MetricPrinterTest, EmptyGroup) { function TEST_F (line 290) | TEST_F(MetricPrinterTest, SizeTOverload) { function TEST_F (line 302) | TEST_F(MetricPrinterTest, MetricToStringFunction) { class ExponentialAggregatorTest (line 310) | class ExponentialAggregatorTest : public ::testing::Test { method SetUp (line 317) | void SetUp() override { function TEST_F (line 328) | TEST_F(ExponentialAggregatorTest, RecordMetrics) { function TEST_F (line 347) | TEST_F(ExponentialAggregatorTest, MultipleUpdatesLiveMetrics) { function TEST_F (line 364) | TEST_F(ExponentialAggregatorTest, Advance) { function TEST_F (line 383) | TEST_F(ExponentialAggregatorTest, MultipleAdvances) { function TEST_F (line 407) | TEST_F(ExponentialAggregatorTest, MultipleAdvancesThreeTicks) { function TEST_F (line 427) | TEST_F(ExponentialAggregatorTest, AggregationTest) { function TEST_F (line 514) | TEST_F(ExponentialAggregatorTest, ActualVsRequestedTimeCoverage) { function TEST_F (line 559) | TEST_F(ExponentialAggregatorTest, ExactDurationTest) { function main (line 591) | int main(int argc, char** argv) { FILE: csrc/utils/queue.h function namespace (line 14) | namespace lczero { FILE: csrc/utils/queue_test.cc type lczero (line 14) | namespace lczero { class QueueTest (line 16) | class QueueTest : public ::testing::Test { method SetUp (line 18) | void SetUp() override {} function TEST_F (line 23) | TEST_F(QueueTest, ConstructorCreatesEmptyQueue) { function TEST_F (line 29) | TEST_F(QueueTest, SinglePutGet) { function TEST_F (line 42) | TEST_F(QueueTest, MovePutGet) { function TEST_F (line 56) | TEST_F(QueueTest, MultiplePutGet) { function TEST_F (line 74) | TEST_F(QueueTest, CircularBufferBehavior) { function TEST_F (line 95) | TEST_F(QueueTest, BatchPutConstSpan) { function TEST_F (line 110) | TEST_F(QueueTest, BatchPutMoveSpan) { function TEST_F (line 129) | TEST_F(QueueTest, BatchPutEmptySpan) { function TEST_F (line 140) | TEST_F(QueueTest, BatchGet) { function TEST_F (line 158) | TEST_F(QueueTest, BatchGetZeroCount) { function TEST_F (line 172) | TEST_F(QueueTest, CapacityOne) { function TEST_F (line 187) | TEST_F(QueueTest, CreateProducerOnClosedQueue) { function TEST_F (line 199) | TEST_F(QueueTest, GetOnClosedQueue) { function TEST_F (line 209) | TEST_F(QueueTest, BatchGetOnClosedQueue) { function TEST_F (line 221) | TEST_F(QueueTest, SingleProducerSingleConsumer) { function TEST_F (line 256) | TEST_F(QueueTest, MultipleProducersMultipleConsumers) { function TEST_F (line 301) | TEST_F(QueueTest, BlockingBehaviorOnFullQueue) { function TEST_F (line 330) | TEST_F(QueueTest, BlockingBehaviorOnEmptyQueue) { function TEST_F (line 356) | TEST_F(QueueTest, ProducerDestructionUnblocksWaitingGet) { function TEST_F (line 389) | TEST_F(QueueTest, GetFromClosedQueueWithElements) { function TEST_F (line 411) | TEST_F(QueueTest, BatchGetFromClosedQueueWithElements) { function TEST_F (line 439) | TEST_F(QueueTest, ProducerTokenMechanism) { function TEST_F (line 472) | TEST_F(QueueTest, ProducerMoveSemantics) { function TEST_F (line 502) | TEST_F(QueueTest, PutOnClosedQueueThrowsException) { function TEST_F (line 519) | TEST_F(QueueTest, PutOnClosedQueueAfterProducerDestruction) { function TEST_F (line 533) | TEST_F(QueueTest, BatchPutOnClosedQueueThrowsException) { function TEST_F (line 549) | TEST_F(QueueTest, PublicCloseMethod) { function TEST_F (line 570) | TEST_F(QueueTest, CloseUnblocksWaitingSinglePut) { function TEST_F (line 602) | TEST_F(QueueTest, CloseUnblocksWaitingBatchPut) { function TEST_F (line 636) | TEST_F(QueueTest, WaitForRoomAtLeast) { function TEST_F (line 685) | TEST_F(QueueTest, WaitForRoomAtMost) { function TEST_F (line 721) | TEST_F(QueueTest, WaitForSizeAtLeast) { function TEST_F (line 753) | TEST_F(QueueTest, WaitForSizeAtMost) { function TEST_F (line 796) | TEST_F(QueueTest, WaitFunctionsEdgeCases) { function TEST_F (line 825) | TEST_F(QueueTest, BatchPutAtCapacityWorks) { function TEST_F (line 835) | TEST_F(QueueTest, BatchGetAtCapacityWorks) { function TEST_F (line 851) | TEST_F(QueueTest, LargeRangePutGetGradual) { function TEST_F (line 878) | TEST_F(QueueTest, LargeRangePutMove) { function TEST_F (line 908) | TEST_F(QueueTest, LargeRangeGetGradual) { function TEST_F (line 933) | TEST_F(QueueTest, LargeRangePutGetConcurrent) { function TEST_F (line 991) | TEST_F(QueueTest, GradualOperationsWithQueueClosure) { function TEST_F (line 1025) | TEST_F(QueueTest, GetTotalPutCountBasic) { function TEST_F (line 1048) | TEST_F(QueueTest, GetTotalPutCountBatch) { function TEST_F (line 1066) | TEST_F(QueueTest, GetTotalPutCountReset) { function TEST_F (line 1088) | TEST_F(QueueTest, GetTotalPutCountThreadSafe) { function TEST_F (line 1116) | TEST_F(QueueTest, GetTotalPutCountBatchThreadSafe) { function TEST_F (line 1145) | TEST_F(QueueTest, GetTotalPutCountWithMoveSemantics) { function TEST_F (line 1163) | TEST_F(QueueTest, GetTotalPutCountEmptyBatch) { function TEST_F (line 1180) | TEST_F(QueueTest, DropNewBasicBehavior) { function TEST_F (line 1205) | TEST_F(QueueTest, DropNewBatchBehavior) { function TEST_F (line 1228) | TEST_F(QueueTest, DropNewThreadSafety) { function TEST_F (line 1264) | TEST_F(QueueTest, KeepNewestBasicBehavior) { function TEST_F (line 1293) | TEST_F(QueueTest, KeepNewestBatchBehavior) { function TEST_F (line 1317) | TEST_F(QueueTest, KeepNewestLargeBatch) { function TEST_F (line 1336) | TEST_F(QueueTest, GetTotalGetCountBasic) { function TEST_F (line 1355) | TEST_F(QueueTest, GetTotalGetCountBatch) { function TEST_F (line 1375) | TEST_F(QueueTest, GetTotalGetCountReset) { function TEST_F (line 1390) | TEST_F(QueueTest, GetTotalDropCountBasic) { function TEST_F (line 1407) | TEST_F(QueueTest, GetTotalDropCountKeepNewest) { function TEST_F (line 1422) | TEST_F(QueueTest, GetTotalDropCountReset) { function TEST_F (line 1437) | TEST_F(QueueTest, MaybeGetOnEmptyQueue) { function TEST_F (line 1443) | TEST_F(QueueTest, MaybeGetOnNonEmptyQueue) { function TEST_F (line 1456) | TEST_F(QueueTest, MaybeGetMultipleValues) { function TEST_F (line 1481) | TEST_F(QueueTest, MaybeGetWithMoveOnlyType) { function TEST_F (line 1494) | TEST_F(QueueTest, MaybeGetUpdatesGetCount) { function TEST_F (line 1516) | TEST_F(QueueTest, StopTokenCancelsPut) { function TEST_F (line 1532) | TEST_F(QueueTest, StopTokenCancelsGet) { function TEST_F (line 1544) | TEST_F(QueueTest, StopTokenCancelsBatchPut) { function TEST_F (line 1562) | TEST_F(QueueTest, StopTokenCancelsBatchGet) { function TEST_F (line 1573) | TEST_F(QueueTest, StopTokenCancelsWaitForRoomAtLeast) { function TEST_F (line 1590) | TEST_F(QueueTest, StopTokenCancelsWaitForSizeAtLeast) { FILE: csrc/utils/stream_shuffler.cc type lczero (line 3) | namespace lczero { type training (line 4) | namespace training { FILE: csrc/utils/stream_shuffler.h function namespace (line 11) | namespace lczero { FILE: csrc/utils/stream_shuffler_test.cc type lczero (line 10) | namespace lczero { type training (line 11) | namespace training { class StreamShufflerTest (line 13) | class StreamShufflerTest : public ::testing::Test { method SetUp (line 15) | void SetUp() override { shuffler_.SetBucketSize(4); } function TEST_F (line 20) | TEST_F(StreamShufflerTest, EmptyRangeReturnsNullopt) { function TEST_F (line 26) | TEST_F(StreamShufflerTest, SingleItemRange) { function TEST_F (line 37) | TEST_F(StreamShufflerTest, BasicRangeGeneration) { function TEST_F (line 54) | TEST_F(StreamShufflerTest, HeadAdvancesByBucketMultiples) { function TEST_F (line 78) | TEST_F(StreamShufflerTest, HeadAdvancesByNonMultiples) { function TEST_F (line 101) | TEST_F(StreamShufflerTest, TailAdvancesByBucketMultiples) { function TEST_F (line 129) | TEST_F(StreamShufflerTest, TailAdvancesByNonMultiples) { function TEST_F (line 157) | TEST_F(StreamShufflerTest, BothBoundsSlideSimultaneously) { function TEST_F (line 187) | TEST_F(StreamShufflerTest, ComplexSlidingWindow) { function TEST_F (line 223) | TEST_F(StreamShufflerTest, UniquenessAcrossMultipleBuckets) { function TEST_F (line 238) | TEST_F(StreamShufflerTest, TailCatchesUpToHead) { function TEST_F (line 251) | TEST_F(StreamShufflerTest, ResetAllowsIterationRestart) { FILE: csrc/utils/tensor.h function namespace (line 12) | namespace lczero { FILE: csrc/utils/tensor_test.cc type lczero (line 8) | namespace lczero { function TEST (line 11) | TEST(TypedTensorTest, ConstructorAndBasicProperties) { function TEST (line 36) | TEST(TypedTensorTest, PyFormatForDifferentTypes) { function TEST (line 50) | TEST(TypedTensorTest, ElementAccess) { function TEST (line 70) | TEST(TypedTensorTest, ConstElementAccess) { function TEST (line 84) | TEST(TypedTensorTest, SliceAccess) { function TEST (line 115) | TEST(TypedTensorTest, ConstSliceAccess) { function TEST (line 129) | TEST(TypedTensorTest, ElementAccessWrongDimensions) { function TEST (line 136) | TEST(TypedTensorTest, SliceAccessTooManyDimensions) { function TEST (line 142) | TEST(TypedTensorTest, OneDimensionalTensor) { FILE: csrc/utils/thread_pool.h function namespace (line 15) | namespace lczero { function num_pending_tasks (line 83) | size_t num_pending_tasks() const; function ThreadPool (line 121) | inline ThreadPool::ThreadPool(size_t initial_threads, function ThreadPool (line 131) | inline ThreadPool::~ThreadPool() { Shutdown(); } function WorkerLoop (line 137) | inline void ThreadPool::WorkerLoop() { function WorkerEntryPoint (line 161) | inline void ThreadPool::WorkerEntryPoint() { function WaitAll (line 175) | inline void ThreadPool::WaitAll() { function WaitForAvailableThread (line 182) | inline void ThreadPool::WaitForAvailableThread() { function WaitForPendingTasksBelow (line 189) | inline void ThreadPool::WaitForPendingTasksBelow(size_t threshold) { function StartWorkerThread (line 196) | inline void ThreadPool::StartWorkerThread() function Shutdown (line 216) | inline void ThreadPool::Shutdown() { FILE: csrc/utils/training_data_printer.cc type lczero (line 12) | namespace lczero { type training (line 13) | namespace training { function PrintFloatArray (line 15) | void PrintFloatArray(const float* data, size_t size, absl::string_vi... function PrintUint64Array (line 32) | void PrintUint64Array(const uint64_t* data, size_t size, absl::strin... function DecodeInvarianceInfo (line 49) | std::string DecodeInvarianceInfo(uint8_t invariance_info) { function TrainingDataToFen (line 59) | std::string TrainingDataToFen(const FrameType& entry) { function PrintTrainingDataEntry (line 73) | void PrintTrainingDataEntry(const FrameType& entry, FILE: csrc/utils/training_data_printer.h function namespace (line 13) | namespace lczero { FILE: scripts/diff.py function get_sorted_chunk_ids (line 8) | def get_sorted_chunk_ids(dirs): function main (line 17) | def main(argv): FILE: scripts/fixorder.py function get_sorted_chunk_ids (line 8) | def get_sorted_chunk_ids(dirs): function main (line 17) | def main(argv): FILE: scripts/initsplit.py function get_sorted_chunk_ids (line 8) | def get_sorted_chunk_ids(dirs): function main (line 17) | def main(argv): FILE: scripts/inittrainingname.py function get_sorted_chunk_ids (line 8) | def get_sorted_chunk_ids(dirs): function main (line 22) | def main(argv): FILE: scripts/pack.py function get_uncompressed_size (line 15) | def get_uncompressed_size(filename): function get_sorted_chunk_ids (line 21) | def get_sorted_chunk_ids(dirs): function pack (line 30) | def pack(ids): function main (line 50) | def main(): FILE: scripts/purge.py function get_sorted_chunk_ids (line 8) | def get_sorted_chunk_ids(dirs): function main (line 17) | def main(argv): FILE: scripts/shuffle.py function split (line 16) | def split(a, n): function positions (line 21) | def positions(chunk): function shuffle (line 30) | def shuffle(files): FILE: scripts/unpack.py function unpack (line 13) | def unpack(filepath): function main (line 42) | def main(): FILE: src/lczero_training/_lczero_training.pyi class TensorBase (line 11) | class TensorBase: method shape (line 12) | def shape(self) -> List[int]: ... method strides (line 13) | def strides(self) -> List[int]: ... method element_size (line 14) | def element_size(self) -> int: ... method py_format (line 15) | def py_format(self) -> str: ... class DataLoader (line 17) | class DataLoader: method __init__ (line 18) | def __init__(self, config: DataLoaderConfig | bytes) -> None: ... method add_stages (line 19) | def add_stages(self, config: DataLoaderConfig | bytes) -> None: ... method send_control_message (line 20) | def send_control_message( method start (line 23) | def start(self) -> None: ... method get_next (line 24) | def get_next(self, alias: str = "") -> Tuple[np.ndarray, ...]: ... method maybe_get_next (line 25) | def maybe_get_next( method stop (line 28) | def stop(self) -> None: ... method get_bucket_metrics (line 29) | def get_bucket_metrics( method get_aggregate_ending_now (line 32) | def get_aggregate_ending_now( FILE: src/lczero_training/commands/backfill_metrics.py function _build_parser (line 9) | def _build_parser() -> argparse.ArgumentParser: function main (line 45) | def main(argv: list[str] | None = None) -> int: FILE: src/lczero_training/commands/common.py function configure_root_logging (line 13) | def configure_root_logging(level: int | str = logging.INFO) -> None: function parse_log_level (line 32) | def parse_log_level(level: int | str) -> int: function add_logging_arguments (line 46) | def add_logging_arguments(parser: argparse.ArgumentParser) -> None: FILE: src/lczero_training/commands/daemon.py function _build_parser (line 9) | def _build_parser() -> argparse.ArgumentParser: function main (line 15) | def main(argv: list[str] | None = None) -> int: FILE: src/lczero_training/commands/dataloader_viz.py function _build_parser (line 11) | def _build_parser() -> argparse.ArgumentParser: function main (line 30) | def main(argv: list[str] | None = None) -> int: FILE: src/lczero_training/commands/describe_training.py function _build_parser (line 8) | def _build_parser() -> argparse.ArgumentParser: function main (line 34) | def main(argv: list[str] | None = None) -> int: FILE: src/lczero_training/commands/jax2leela.py function _build_parser (line 19) | def _build_parser() -> argparse.ArgumentParser: function jax2leela (line 49) | def jax2leela( function main (line 116) | def main(argv: list[str] | None = None) -> int: FILE: src/lczero_training/commands/leela2jax.py function _build_parser (line 5) | def _build_parser() -> argparse.ArgumentParser: function main (line 50) | def main(argv: list[str] | None = None) -> int: FILE: src/lczero_training/commands/migrate_checkpoint.py function _build_parser (line 8) | def _build_parser() -> argparse.ArgumentParser: function main (line 64) | def main(argv: list[str] | None = None) -> int: FILE: src/lczero_training/commands/overfit.py function _build_parser (line 8) | def _build_parser() -> argparse.ArgumentParser: function main (line 39) | def main(argv: list[str] | None = None) -> int: FILE: src/lczero_training/commands/test_dataloader.py function _build_parser (line 8) | def _build_parser() -> argparse.ArgumentParser: function main (line 34) | def main(argv: list[str] | None = None) -> int: FILE: src/lczero_training/commands/train.py function _build_parser (line 27) | def _build_parser() -> argparse.ArgumentParser: function train (line 38) | def train(config_filename: str) -> None: function main (line 124) | def main(argv: list[str] | None = None) -> int: FILE: src/lczero_training/commands/training_eval.py function _build_parser (line 8) | def _build_parser() -> argparse.ArgumentParser: function main (line 59) | def main(argv: list[str] | None = None) -> int: FILE: src/lczero_training/commands/training_init.py function _build_parser (line 8) | def _build_parser() -> argparse.ArgumentParser: function main (line 63) | def main(argv: list[str] | None = None) -> int: FILE: src/lczero_training/commands/tui.py function _build_parser (line 11) | def _build_parser() -> argparse.ArgumentParser: function _amain (line 17) | async def _amain(args: argparse.Namespace) -> None: function main (line 22) | def main(argv: list[str] | None = None) -> int: FILE: src/lczero_training/commands/tune_lr.py function _build_parser (line 8) | def _build_parser() -> argparse.ArgumentParser: function main (line 76) | def main(argv: list[str] | None = None) -> int: FILE: src/lczero_training/commands/weights_tool.py function _build_parser (line 11) | def _build_parser() -> argparse.ArgumentParser: function main (line 49) | def main(argv: list[str] | None = None) -> int: FILE: src/lczero_training/convert/jax_to_leela.py class JaxToLeela (line 19) | class JaxToLeela(LeelaPytreeWeightsVisitor): method embedding_block (line 20) | def embedding_block( method tensor (line 33) | def tensor( method encoder_tower (line 55) | def encoder_tower( class LeelaExportOptions (line 64) | class LeelaExportOptions: function jax_to_leela (line 71) | def jax_to_leela( function _split_version (line 96) | def _split_version(version_str: str) -> tuple[int, int, int]: function _make_format (line 102) | def _make_format() -> net_pb2.Format: FILE: src/lczero_training/convert/leela_pytree_visitor.py class LeelaPytreeWeightsVisitor (line 9) | class LeelaPytreeWeightsVisitor: method __init__ (line 10) | def __init__(self, nnx_state: nnx.State, leela_net: net_pb2.Net) -> None: method run (line 14) | def run(self) -> None: method embedding_block (line 32) | def embedding_block( method encoder_tower (line 63) | def encoder_tower( method encoder_block (line 81) | def encoder_block( method mha (line 89) | def mha(self, nnx_dict: nnx.State, weights: net_pb2.Weights.MHA) -> None: method smolgen (line 96) | def smolgen( method layernorm (line 105) | def layernorm( method policy_heads (line 114) | def policy_heads( method policy_head (line 130) | def policy_head( method value_head (line 139) | def value_head( method movesleft_head (line 156) | def movesleft_head( method ffn (line 163) | def ffn(self, nnx_dict: nnx.State, ffn: net_pb2.Weights.FFN) -> None: method matmul (line 167) | def matmul( method tensor (line 179) | def tensor( FILE: src/lczero_training/convert/leela_to_jax.py class LeelaImportOptions (line 25) | class LeelaImportOptions: function fix_older_weights_file (line 30) | def fix_older_weights_file(file: net_pb2.Net) -> None: class LeelaToJax (line 84) | class LeelaToJax(LeelaPytreeWeightsVisitor): method embedding_block (line 85) | def embedding_block( method tensor (line 96) | def tensor( function leela_to_jax (line 113) | def leela_to_jax( function leela_to_jax_files (line 130) | def leela_to_jax_files( FILE: src/lczero_training/convert/leela_to_modelconfig.py function _defaultactivation_to_activation (line 4) | def _defaultactivation_to_activation( function leela_to_modelconfig (line 13) | def leela_to_modelconfig( FILE: src/lczero_training/daemon/daemon.py class TrainingDaemon (line 20) | class TrainingDaemon: method __init__ (line 25) | def __init__(self, memory_profile_dir: str | None = None) -> None: method _setup_logging (line 44) | def _setup_logging(self) -> None: method _setup_signal_handling (line 56) | def _setup_signal_handling(self) -> None: method _signal_handler_thread (line 62) | def _signal_handler_thread(self) -> None: method _shutdown (line 67) | def _shutdown(self, signum: int) -> None: method _metrics_main (line 72) | async def _metrics_main(self) -> None: method _metrics_task (line 76) | async def _metrics_task(self) -> None: method run (line 116) | def run(self) -> None: method on_start_training (line 128) | def on_start_training(self, payload: StartTrainingPayload) -> None: method on_start_training_immediately (line 131) | def on_start_training_immediately( FILE: src/lczero_training/daemon/metrics.py class CachedBatch (line 31) | class CachedBatch: function load_batch_from_npz (line 38) | def load_batch_from_npz(npz_filename: str) -> BatchTuple: class _TrainingBatchMetric (line 59) | class _TrainingBatchMetric(_Metric): method __init__ (line 62) | def __init__(self, config: MetricConfig, logger: TensorboardLogger): method log (line 70) | def log(self, hook_data: StepHookData, graphdef: nnx.GraphDef) -> None: class _EvaluatingMetric (line 74) | class _EvaluatingMetric(_Metric, ABC): method __init__ (line 77) | def __init__( method get_batch (line 89) | def get_batch(self) -> BatchTuple: method log (line 92) | def log(self, hook_data: StepHookData, graphdef: nnx.GraphDef) -> None: method _evaluate (line 97) | def _evaluate( function _make_eval_jit (line 122) | def _make_eval_jit(graphdef: nnx.GraphDef, loss_fn: LczeroLoss) -> _Eval... function evaluate_batch (line 144) | def evaluate_batch( class _DataLoaderMetric (line 176) | class _DataLoaderMetric(_EvaluatingMetric): method __init__ (line 179) | def __init__( method get_batch (line 195) | def get_batch(self) -> BatchTuple: method log (line 198) | def log(self, hook_data: StepHookData, graphdef: nnx.GraphDef) -> None: class _NpzMetric (line 216) | class _NpzMetric(_EvaluatingMetric): method __init__ (line 219) | def __init__( method get_batch (line 231) | def get_batch(self) -> BatchTuple: method log (line 234) | def log(self, hook_data: StepHookData, graphdef: nnx.GraphDef) -> None: class Metrics (line 245) | class Metrics: method __init__ (line 248) | def __init__( method on_step (line 294) | def on_step(self, hook_data: StepHookData, graphdef: nnx.GraphDef) -> ... method close (line 304) | def close(self) -> None: FILE: src/lczero_training/daemon/metrics_base.py class _Metric (line 12) | class _Metric(ABC): method __init__ (line 15) | def __init__(self, config: MetricConfig, logger: TensorboardLogger): method should_log (line 19) | def should_log( method log (line 29) | def log(self, hook_data: StepHookData, graphdef: nnx.GraphDef) -> None: FILE: src/lczero_training/daemon/pipeline.py function _read_config_file (line 44) | def _read_config_file(config_filepath: str) -> RootConfig: function _make_dataloader (line 53) | def _make_dataloader(config: DataLoaderConfig) -> DataLoader: function _configure_file_logging (line 58) | def _configure_file_logging(config: RootConfig) -> None: function _log_jax_system_info (line 73) | def _log_jax_system_info() -> None: class _TrainingCycleState (line 93) | class _TrainingCycleState: class TrainingPipeline (line 106) | class TrainingPipeline: method __init__ (line 116) | def __init__( method start_training_immediately (line 211) | def start_training_immediately(self) -> None: method run (line 217) | def run(self) -> None: method _export_network (line 239) | def _export_network(self) -> bytes | None: method _save_network (line 263) | def _save_network(self, network_bytes: bytes) -> None: method _upload_network (line 277) | def _upload_network(self, network_bytes: bytes) -> None: method _step_hook (line 319) | def _step_hook(self, hook_data: StepHookData) -> None: method _train_one_network (line 325) | def _train_one_network(self) -> None: method _save_checkpoint (line 360) | def _save_checkpoint(self) -> None: method stop (line 368) | def stop(self) -> None: method get_data_loader (line 373) | def get_data_loader(self) -> DataLoader: method _wait_for_chunks (line 376) | def _wait_for_chunks(self) -> None: method get_training_schedule_data (line 397) | def get_training_schedule_data( method _send_chunk_pool_control (line 434) | def _send_chunk_pool_control( method _reset_chunk_anchor (line 443) | def _reset_chunk_anchor(self) -> tuple[str, int]: method _chunks_since_anchor (line 452) | def _chunks_since_anchor(self) -> int: method _set_chunk_anchor (line 460) | def _set_chunk_anchor(self, anchor: str) -> None: method _load_config (line 465) | def _load_config(self, config_filepath: str) -> RootConfig: FILE: src/lczero_training/daemon/protocol/communicator.py function _to_serializable (line 17) | def _to_serializable(obj: Any) -> Any: function _unwrap_optional (line 39) | def _unwrap_optional(t: Any) -> Any: function _is_protobuf (line 47) | def _is_protobuf(cls: type) -> bool: function _from_serializable (line 55) | def _from_serializable(cls: type, data: Any) -> Any: class Communicator (line 88) | class Communicator: method __init__ (line 89) | def __init__( method send (line 104) | def send(self, payload_instance: Any) -> None: method _dispatch (line 123) | def _dispatch(self, line: str) -> None: method run (line 140) | def run(self) -> None: class AsyncCommunicator (line 152) | class AsyncCommunicator: method __init__ (line 153) | def __init__( method send (line 175) | async def send(self, payload_instance: Any) -> None: method _dispatch (line 196) | async def _dispatch(self, line: str) -> None: method run (line 213) | async def run(self) -> None: FILE: src/lczero_training/daemon/protocol/messages.py class TrainingStage (line 13) | class TrainingStage(Enum): class TrainingScheduleData (line 19) | class TrainingScheduleData: class StartTrainingPayload (line 36) | class StartTrainingPayload: class StartTrainingImmediatelyPayload (line 42) | class StartTrainingImmediatelyPayload: class TrainingStatusPayload (line 51) | class TrainingStatusPayload: FILE: src/lczero_training/daemon/protocol/registry.py function register (line 12) | def register(event_type: str) -> Callable[[type], type]: FILE: src/lczero_training/daemon/rms_metrics.py function compute_rms (line 18) | def compute_rms(state_subtree: nnx.State) -> jax.Array: function extract_attention_components (line 26) | def extract_attention_components(model: LczeroModel) -> dict[str, Any]: function collect_rms_metrics (line 58) | def collect_rms_metrics(model: LczeroModel) -> dict[str, Any]: class _RmsMetric (line 103) | class _RmsMetric(_Metric): method __init__ (line 106) | def __init__(self, config: MetricConfig, logger: TensorboardLogger): method log (line 109) | def log(self, hook_data: StepHookData, graphdef: nnx.GraphDef) -> None: FILE: src/lczero_training/dataloader/__init__.py function make_dataloader (line 10) | def make_dataloader(config: DataLoaderConfig) -> DataLoader: FILE: src/lczero_training/model/embedding.py class Embedding (line 11) | class Embedding(nnx.Module): method __init__ (line 14) | def __init__( method __call__ (line 54) | def __call__(self, x: jax.Array) -> jax.Array: class MaGating (line 70) | class MaGating(nnx.Module): method __init__ (line 73) | def __init__(self, feature_shape: tuple[int, ...], *, rngs: nnx.Rngs): method __call__ (line 81) | def __call__(self, x: jax.Array) -> jax.Array: class Gating (line 85) | class Gating(nnx.Module): method __init__ (line 86) | def __init__( method __call__ (line 99) | def __call__(self, inputs: jax.Array) -> jax.Array: FILE: src/lczero_training/model/encoder.py class EncoderTower (line 15) | class EncoderTower(nnx.Module): method __init__ (line 16) | def __init__( method __call__ (line 49) | def __call__(self, x: jax.Array) -> jax.Array: class EncoderBlock (line 53) | class EncoderBlock(nnx.Module): method __init__ (line 56) | def __init__( method __call__ (line 87) | def __call__(self, x: jax.Array) -> jax.Array: class MultiHeadAttention (line 94) | class MultiHeadAttention(nnx.Module): method __init__ (line 97) | def __init__( method __call__ (line 153) | def __call__(self, x: jax.Array) -> jax.Array: class Smolgen (line 180) | class Smolgen(nnx.Module): method __init__ (line 183) | def __init__( method __call__ (line 218) | def __call__(self, x: jax.Array) -> jax.Array: FILE: src/lczero_training/model/loss_function.py function _compute_q_from_wdl (line 24) | def _compute_q_from_wdl(wdl_logits: jax.Array) -> jax.Array: class LossBase (line 31) | class LossBase: method __init__ (line 32) | def __init__( method __call__ (line 46) | def __call__( class RegularizationLoss (line 54) | class RegularizationLoss: method __init__ (line 57) | def __init__(self, config: RegularizationLossConfig) -> None: method __call__ (line 62) | def __call__(self, model: LczeroModel) -> jax.Array: class LczeroLoss (line 77) | class LczeroLoss: method __init__ (line 85) | def __init__(self, config: LossConfig) -> None: method __call__ (line 131) | def __call__( class ValueLoss (line 181) | class ValueLoss(LossBase): method __init__ (line 182) | def __init__(self, config: ValueLossConfig) -> None: method __call__ (line 186) | def __call__( class PolicyLoss (line 209) | class PolicyLoss(LossBase): method __init__ (line 210) | def __init__(self, config: PolicyLossConfig): method _apply_temperature_and_normalize (line 235) | def _apply_temperature_and_normalize( method _compute_optimistic_weight (line 251) | def _compute_optimistic_weight( method __call__ (line 278) | def __call__( class MovesLeftLoss (line 323) | class MovesLeftLoss(LossBase): method __init__ (line 324) | def __init__(self, config: MovesLeftLossConfig) -> None: method __call__ (line 328) | def __call__( class ValueErrorLoss (line 351) | class ValueErrorLoss(LossBase): method __init__ (line 352) | def __init__(self, config: ValueErrorLossConfig) -> None: method __call__ (line 357) | def __call__( class ValueCategoricalLoss (line 386) | class ValueCategoricalLoss(LossBase): method __init__ (line 387) | def __init__(self, config: ValueCategoricalLossConfig) -> None: method __call__ (line 391) | def __call__( FILE: src/lczero_training/model/model.py class ModelPrediction (line 21) | class ModelPrediction: class LczeroModel (line 35) | class LczeroModel(nnx.Module): method __init__ (line 36) | def __init__(self, config: model_config_pb2.ModelConfig, *, rngs: nnx.... method __call__ (line 109) | def __call__(self, x: jax.Array) -> ModelPrediction: FILE: src/lczero_training/model/movesleft_head.py class MovesLeftHead (line 9) | class MovesLeftHead(nnx.Module): method __init__ (line 10) | def __init__( method __call__ (line 36) | def __call__(self, x: jax.Array) -> jax.Array: FILE: src/lczero_training/model/policy_head.py class PolicyHead (line 13) | class PolicyHead(nnx.Module): method __init__ (line 14) | def __init__( method __call__ (line 58) | def __call__(self, x: jax.Array) -> jax.Array: FILE: src/lczero_training/model/shared.py class Ffn (line 10) | class Ffn(nnx.Module): method __init__ (line 11) | def __init__( method __call__ (line 40) | def __call__(self, x: jax.Array) -> jax.Array: FILE: src/lczero_training/model/utils.py function get_activation (line 11) | def get_activation( function get_dtype (line 26) | def get_dtype(dtype: XlaShapeProto.Type) -> jnp.dtype: FILE: src/lczero_training/model/value_head.py class ValueHead (line 11) | class ValueHead(nnx.Module): method __init__ (line 12) | def __init__( method __call__ (line 50) | def __call__( method predict (line 66) | def predict(self, x: jax.Array) -> jax.Array: FILE: src/lczero_training/tests/test_protobuf.py function test_protobuf_import (line 4) | def test_protobuf_import() -> None: function test_protobuf_functionality (line 29) | def test_protobuf_functionality() -> None: FILE: src/lczero_training/tests/test_protocol_registry.py function clear_registry (line 16) | def clear_registry() -> Any: function test_basic_registration (line 25) | def test_basic_registration() -> None: function test_duplicate_event_type (line 39) | def test_duplicate_event_type() -> None: function test_duplicate_class (line 57) | def test_duplicate_class() -> None: function test_non_class_registration (line 74) | def test_non_class_registration() -> None: function test_multiple_registrations (line 83) | def test_multiple_registrations() -> None: function test_registry_persistence (line 115) | def test_registry_persistence() -> None: FILE: src/lczero_training/tests/test_weights_tool.py function test_weights_arithmetic (line 13) | def test_weights_arithmetic() -> None: function test_policy_head_replacement (line 54) | def test_policy_head_replacement() -> None: function test_policy_head_map_assignment (line 101) | def test_policy_head_map_assignment() -> None: function test_noop_arithmetic (line 158) | def test_noop_arithmetic() -> None: function test_list_item_assignment (line 195) | def test_list_item_assignment() -> None: FILE: src/lczero_training/tools/weight_codecs.py function decode_linear16 (line 8) | def decode_linear16( function encode_linear16 (line 18) | def encode_linear16(arr: np.ndarray) -> tuple[bytes, float, float]: function decode_float16 (line 34) | def decode_float16(params: bytes, shape: tuple[int, ...]) -> np.ndarray: function encode_float16 (line 41) | def encode_float16(arr: np.ndarray) -> tuple[bytes, float, float]: function decode_bfloat16 (line 47) | def decode_bfloat16(params: bytes, shape: tuple[int, ...]) -> np.ndarray: function encode_bfloat16 (line 55) | def encode_bfloat16(arr: np.ndarray) -> tuple[bytes, float, float]: function decode_layer (line 63) | def decode_layer( function encode_layer (line 87) | def encode_layer(arr: np.ndarray, encoding: int) -> tuple[bytes, float, ... FILE: src/lczero_training/tools/weight_wrappers.py class LayerWrapper (line 14) | class LayerWrapper: method __init__ (line 19) | def __init__( method value (line 33) | def value(self) -> np.ndarray: method value (line 44) | def value(self, arr: np.ndarray) -> None: method commit (line 49) | def commit(self, encoding: int) -> None: method __add__ (line 63) | def __add__(self, other: "LayerWrapper") -> "LayerWrapper": method __sub__ (line 70) | def __sub__(self, other: "LayerWrapper") -> "LayerWrapper": method __mul__ (line 77) | def __mul__(self, scalar: float) -> "LayerWrapper": method __rmul__ (line 84) | def __rmul__(self, scalar: float) -> "LayerWrapper": class ListWrapper (line 88) | class ListWrapper: method __init__ (line 93) | def __init__(self, proto_list: Any, parent: "NetWrapper") -> None: method __len__ (line 102) | def __len__(self) -> int: method __getitem__ (line 105) | def __getitem__(self, idx: int) -> Any: method __setitem__ (line 111) | def __setitem__(self, idx: int, value: Any) -> None: method __iter__ (line 132) | def __iter__(self) -> Iterator[Any]: class NetWrapper (line 137) | class NetWrapper: method __init__ (line 142) | def __init__( method _detect_encoding (line 157) | def _detect_encoding(self) -> int: method __getattr__ (line 163) | def __getattr__(self, name: str) -> Any: method _wrap_field (line 180) | def _wrap_field(self, value: Any) -> Any: method __setattr__ (line 191) | def __setattr__(self, name: str, value: Any) -> None: method save (line 213) | def save(self, path: str, encoding: int | None = None) -> None: method _commit_all (line 228) | def _commit_all(self, encoding: int) -> None: method __add__ (line 242) | def __add__(self, other: "NetWrapper") -> "NetWrapper": method __sub__ (line 253) | def __sub__(self, other: "NetWrapper") -> "NetWrapper": method __mul__ (line 264) | def __mul__(self, scalar: float) -> "NetWrapper": method __rmul__ (line 275) | def __rmul__(self, scalar: float) -> "NetWrapper": method _add_weights (line 278) | def _add_weights(self, lhs: "NetWrapper", rhs: "NetWrapper") -> None: method _sub_weights (line 316) | def _sub_weights(self, lhs: "NetWrapper", rhs: "NetWrapper") -> None: method _mul_weights (line 354) | def _mul_weights(self, source: "NetWrapper", scalar: float) -> None: FILE: src/lczero_training/tools/weights_tool.py function load_weights (line 10) | def load_weights(path: str) -> NetWrapper: function save_weights (line 24) | def save_weights( FILE: src/lczero_training/training/backfill_metrics.py function _load_config (line 28) | def _load_config(config_path: str) -> RootConfig: function _validate_and_get_metrics (line 36) | def _validate_and_get_metrics( function _load_and_migrate_checkpoint (line 64) | def _load_and_migrate_checkpoint( function backfill_metrics (line 75) | def backfill_metrics( FILE: src/lczero_training/training/dataloader_probe.py function _stop_loader (line 18) | def _stop_loader(loader: DataLoader) -> None: function _store_batches (line 23) | def _store_batches(path: str, batches: list) -> None: function probe_dataloader (line 33) | def probe_dataloader( FILE: src/lczero_training/training/describe.py function describe (line 17) | def describe( FILE: src/lczero_training/training/eval.py class DiffRecord (line 50) | class DiffRecord: function _tensor_to_list (line 61) | def _tensor_to_list(obj: Any) -> Any: function _bin_counts (line 73) | def _bin_counts(values: np.ndarray) -> Dict[str, Any]: function _format_bound (line 88) | def _format_bound(value: float) -> str: function _format_stats (line 95) | def _format_stats(stats: Dict[str, Any]) -> str: function _collect_diff_statistics (line 108) | def _collect_diff_statistics( class Dumper (line 125) | class Dumper: method __init__ (line 128) | def __init__( method dump_tensors (line 142) | def dump_tensors(self, tensors: dict, prefix: str) -> None: method dump_structured (line 159) | def dump_structured(self, batch: dict, outputs: dict, losses: dict) ->... method _dump_to_shelve (line 176) | def _dump_to_shelve(self, key: str, data: dict) -> None: method _dump_to_json (line 182) | def _dump_to_json(self, key: str, data: dict) -> None: method close (line 194) | def close(self) -> None: class OnnxComparator (line 199) | class OnnxComparator: method __init__ (line 202) | def __init__(self, onnx_model_path: str): method compare (line 223) | def compare( method log_summary (line 262) | def log_summary(self) -> None: method _log_diff_stats (line 280) | def _log_diff_stats( method _align_onnx_outputs (line 328) | def _align_onnx_outputs( method _reshape_output (line 375) | def _reshape_output( class Evaluation (line 393) | class Evaluation: method __init__ (line 396) | def __init__(self, loss_fn: LczeroLoss): method run (line 399) | def run( method _process_sample (line 427) | def _process_sample( method _loss_for_grad (line 502) | def _loss_for_grad( method _model_for_output (line 508) | def _model_for_output( function from_dataloader (line 514) | def from_dataloader( function _load_model_from_checkpoint (line 522) | def _load_model_from_checkpoint(config: RootConfig) -> LczeroModel: function _get_dataloader_config (line 545) | def _get_dataloader_config( function eval (line 564) | def eval( FILE: src/lczero_training/training/init.py function _load_lc0_model_state (line 25) | def _load_lc0_model_state( function init (line 62) | def init( FILE: src/lczero_training/training/lr_schedule.py function _create_rule_fn (line 9) | def _create_rule_fn(rule: LrSchedule) -> Callable: function make_lr_schedule (line 103) | def make_lr_schedule(schedules: Sequence[LrSchedule]) -> optax.Schedule: FILE: src/lczero_training/training/migrate_checkpoint.py function _str_to_key_path (line 14) | def _str_to_key_path(path_str: str) -> tuple[str, ...]: function _load_new_state (line 18) | def _load_new_state( function load_checkpoint (line 29) | def load_checkpoint( function get_checkpoint_steps (line 52) | def get_checkpoint_steps( function _load_old_state (line 83) | def _load_old_state( function load_migration_rules (line 89) | def load_migration_rules(rules_file: str | None) -> List[Tuple[Any, Any]]: function _format_value (line 121) | def _format_value(value: Any) -> str: function _format_path_diff (line 127) | def _format_path_diff( class Migration (line 143) | class Migration: method __init__ (line 144) | def __init__(self, old_state: Any, new_state: Any): method _apply_move_rule (line 171) | def _apply_move_rule( method _apply_ignore_rule (line 202) | def _apply_ignore_rule(self, from_path: Tuple[str, ...]) -> None: method _apply_keep_rule (line 210) | def _apply_keep_rule(self, to_path: Tuple[str, ...]) -> None: method apply_rules (line 218) | def apply_rules(self, rules: List[Tuple[Any, Any]]) -> None: method run (line 227) | def run(self, rules: List[Tuple[Any, Any]]) -> Any: function _save_checkpoint (line 258) | def _save_checkpoint( function _dump_paths (line 291) | def _dump_paths(paths: Iterable[Tuple[str, ...]], field: str) -> None: function migrate_checkpoint (line 299) | def migrate_checkpoint( FILE: src/lczero_training/training/optimizer.py function update_optimizer_step (line 17) | def update_optimizer_step( function make_gradient_transformation (line 37) | def make_gradient_transformation( FILE: src/lczero_training/training/overfit.py function _stop_loader (line 32) | def _stop_loader(loader: DataLoader) -> None: function _prepare_batch (line 37) | def _prepare_batch(batch_tuple: tuple) -> TrainingBatch: function _make_eval_step (line 46) | def _make_eval_step(graphdef: nnx.GraphDef, loss_fn: LczeroLoss) -> Any: function overfit (line 69) | def overfit( FILE: src/lczero_training/training/state.py class TrainingSample (line 27) | class TrainingSample: class TrainingBatch (line 51) | class TrainingBatch: method from_tuple (line 71) | def from_tuple( class JitTrainingState (line 87) | class JitTrainingState: method replace (line 97) | def replace(self, **changes: Any) -> "JitTrainingState": class TrainingState (line 103) | class TrainingState: method replace (line 109) | def replace(self, **changes: Any) -> "TrainingState": method with_updated_step (line 113) | def with_updated_step(self, step: int) -> "TrainingState": method new_from_config (line 128) | def new_from_config( FILE: src/lczero_training/training/tensorboard.py function _to_ndarray (line 18) | def _to_ndarray(value: Any) -> np.ndarray: function _to_scalar (line 25) | def _to_scalar(value: Any) -> float | None: function _flatten_metrics (line 36) | def _flatten_metrics( function _to_step (line 51) | def _to_step(step: Any) -> int: class TensorboardLogger (line 55) | class TensorboardLogger: method __init__ (line 58) | def __init__(self, logdir: str) -> None: method log (line 61) | def log(self, step: int, metrics: MetricsDict) -> None: method close (line 68) | def close(self) -> None: FILE: src/lczero_training/training/test_lr_schedule.py function _sched (line 10) | def _sched( function _val (line 16) | def _val(s: Callable[[jnp.ndarray], jnp.ndarray], t: int | float) -> float: function test_rule_selection_by_starting_step (line 20) | def test_rule_selection_by_starting_step() -> None: function test_default_constant_transition_and_tail (line 38) | def test_default_constant_transition_and_tail() -> None: function test_linear_then_hold (line 51) | def test_linear_then_hold() -> None: function test_looping_constant_segments (line 66) | def test_looping_constant_segments() -> None: function test_zero_duration_is_skipped (line 80) | def test_zero_duration_is_skipped() -> None: function test_chain_zero_durations_then_linear (line 97) | def test_chain_zero_durations_then_linear() -> None: function test_cosine (line 116) | def test_cosine() -> None: function test_before_first_rule_uses_earliest_first_lr (line 130) | def test_before_first_rule_uses_earliest_first_lr() -> None: FILE: src/lczero_training/training/training.py class StepHookData (line 30) | class StepHookData: function from_dataloader (line 45) | def from_dataloader( class Training (line 52) | class Training: method __init__ (line 61) | def __init__( method _swa_tree_map (line 164) | def _swa_tree_map( method update_swa (line 174) | def update_swa( method maybe_update_swa (line 206) | def maybe_update_swa( method _validate_and_prepare_batch (line 227) | def _validate_and_prepare_batch( method _log_step_metrics (line 258) | def _log_step_metrics( method _execute_step_hook (line 275) | def _execute_step_hook( method run (line 295) | def run( FILE: src/lczero_training/training/tune_lr.py function _prepare_batch (line 28) | def _prepare_batch(batch_tuple: tuple) -> Dict: function _make_optimizer_with_schedule (line 37) | def _make_optimizer_with_schedule( function _make_eval_step (line 68) | def _make_eval_step( function _plot_results (line 87) | def _plot_results(results: List[Tuple[float, float]], plot_output: str) ... function tune_lr (line 102) | def tune_lr( FILE: src/lczero_training/training/utils.py function make_weights_mask (line 8) | def make_weights_mask( FILE: src/lczero_training/tui/app.py class HeaderBar (line 33) | class HeaderBar(Static): method compose (line 36) | def compose(self) -> ComposeResult: class JAXTrainingPane (line 41) | class JAXTrainingPane(Static): method compose (line 44) | def compose(self) -> ComposeResult: class TrainingTuiApp (line 53) | class TrainingTuiApp(App): method add_arguments (line 66) | def add_arguments(parser: argparse.ArgumentParser) -> None: method __init__ (line 102) | def __init__(self, args: Optional[argparse.Namespace] = None) -> None: method on_load (line 122) | async def on_load(self) -> None: method compose (line 160) | def compose(self) -> ComposeResult: method _monitor_daemon_process (line 183) | async def _monitor_daemon_process(self) -> None: method on_mount (line 198) | def on_mount(self) -> None: method _send_start_training (line 204) | async def _send_start_training(self) -> None: method _command_start_training_immediately (line 209) | async def _command_start_training_immediately(self) -> None: method action_quit (line 216) | async def action_quit(self) -> None: # type: ignore method get_system_commands (line 225) | def get_system_commands(self, screen: Screen) -> Iterable[SystemCommand]: method on_training_status (line 235) | async def on_training_status(self, payload: TrainingStatusPayload) -> ... FILE: src/lczero_training/tui/data_pipeline_pane.py class DataPipelinePane (line 26) | class DataPipelinePane(Container): method __init__ (line 29) | def __init__(self, **kwargs: Any) -> None: method compose (line 38) | def compose(self) -> ComposeResult: method _friendly_title (line 42) | def _friendly_title(self, stage_key: str) -> str: method _ensure_stage_widget (line 47) | def _ensure_stage_widget( method _ensure_queue_widgets (line 67) | def _ensure_queue_widgets( method _ensure_statistics_widgets (line 97) | def _ensure_statistics_widgets( method _mount_widgets (line 124) | def _mount_widgets( method _ensure_rows (line 132) | def _ensure_rows( method update_metrics (line 157) | def update_metrics( FILE: src/lczero_training/tui/dataloader_widgets.py function _find_stage_metric (line 14) | def _find_stage_metric( function _collect_metric_names (line 26) | def _collect_metric_names( function _find_load_metric (line 48) | def _find_load_metric( function _find_count_metric (line 60) | def _find_count_metric( function _find_gauge_metric (line 72) | def _find_gauge_metric( function _find_statistics_metric (line 84) | def _find_statistics_metric( function _get_queue_metric (line 96) | def _get_queue_metric( function format_si (line 117) | def format_si(value: int, precision: int = 1) -> str: function format_full_number (line 135) | def format_full_number(value: int) -> str: function _format_load (line 141) | def _format_load( function _format_count (line 155) | def _format_count( function _format_gauge (line 171) | def _format_gauge( function _format_statistics (line 184) | def _format_statistics( function _average_queue_fullness (line 205) | def _average_queue_fullness( function _canonical_stage_name (line 220) | def _canonical_stage_name( class BaseRowWidget (line 234) | class BaseRowWidget(Widget): method __init__ (line 239) | def __init__( method compose (line 257) | def compose(self) -> ComposeResult: method on_mount (line 261) | def on_mount(self) -> None: method add_content_widget (line 276) | def add_content_widget(self, widget: Widget) -> None: method _update_name (line 281) | def _update_name( class StageWidget (line 292) | class StageWidget(BaseRowWidget): method __init__ (line 295) | def __init__( method _ensure_chip (line 309) | def _ensure_chip(self, key: str, default_text: str, classes: str) -> S... method _update_last_chunk_chip (line 317) | def _update_last_chunk_chip( method _update_anchor_chip (line 333) | def _update_anchor_chip( method update_metrics (line 342) | def update_metrics( class StatisticsRowWidget (line 404) | class StatisticsRowWidget(BaseRowWidget): method __init__ (line 407) | def __init__( method compose (line 425) | def compose(self) -> ComposeResult: method on_mount (line 429) | def on_mount(self) -> None: method update_metrics (line 437) | def update_metrics( class QueueWidget (line 454) | class QueueWidget(BaseRowWidget): method __init__ (line 457) | def __init__( method compose (line 490) | def compose(self) -> ComposeResult: method on_mount (line 495) | def on_mount(self) -> None: method update_metrics (line 504) | def update_metrics( FILE: src/lczero_training/tui/log_pane.py class StreamingLogPane (line 9) | class StreamingLogPane(RichLog): method __init__ (line 12) | def __init__( method on_mount (line 28) | def on_mount(self) -> None: method _write_banner (line 34) | def _write_banner(self) -> None: method _write_to_file (line 50) | def _write_to_file(self, line: str) -> None: method _read_stream (line 58) | async def _read_stream(self) -> None: FILE: src/lczero_training/tui/training_widgets.py class TimeProgressWidget (line 7) | class TimeProgressWidget(Static): method __init__ (line 10) | def __init__(self, label: str, *, id: str | None = None) -> None: method compose (line 14) | def compose(self) -> ComposeResult: method update_progress (line 19) | def update_progress( function format_time_duration (line 46) | def format_time_duration(seconds: float) -> str: class TrainingScheduleWidget (line 63) | class TrainingScheduleWidget(Static): method compose (line 66) | def compose(self) -> ComposeResult: method update_training_schedule (line 73) | def update_training_schedule( FILE: tf/attention_policy_map.py function make_map (line 39) | def make_map(): function make_pos_enc (line 96) | def make_pos_enc(): FILE: tf/chunkparsefunc.py function parse_function (line 21) | def parse_function(planes, probs, winner, q, plies_left): FILE: tf/chunkparser.py function reverse_expand_bits (line 82) | def reverse_expand_bits(plane): class ChunkDataSrc (line 88) | class ChunkDataSrc: method __init__ (line 89) | def __init__(self, items): method next (line 92) | def next(self): function chunk_reader (line 98) | def chunk_reader(chunk_filenames, chunk_filename_queue): class ChunkParser (line 121) | class ChunkParser: method __init__ (line 123) | def __init__(self, method shutdown (line 141) | def shutdown(self): method parse (line 153) | def parse(self): method sequential (line 156) | def sequential(self): class ChunkParserInner (line 160) | class ChunkParserInner: method __init__ (line 161) | def __init__(self, parent, chunks, expected_input_format, shuffle_size, method init_structs (line 240) | def init_structs(self): method convert_v6_to_tuple (line 250) | def convert_v6_to_tuple(self, content): method sample_record (line 405) | def sample_record(self, chunkdata): method single_file_gen (line 462) | def single_file_gen(self, filename): method sequential_gen (line 489) | def sequential_gen(self): method sequential (line 494) | def sequential(self): method task (line 501) | def task(self, chunk_filename_queue, writer): method v6_gen (line 512) | def v6_gen(self): method tuple_gen (line 536) | def tuple_gen(self, gen): method batch_gen (line 544) | def batch_gen(self, gen, allow_partial=True): method parse (line 558) | def parse(self): class ChunkParserTest (line 570) | class ChunkParserTest(unittest.TestCase): method setUp (line 571) | def setUp(self): method generate_fake_pos (line 574) | def generate_fake_pos(self): method v4_record (line 601) | def v4_record(self, planes, i, probs, winner, best_q, best_d): method test_structsize (line 612) | def test_structsize(self): method test_parsing (line 618) | def test_parsing(self): FILE: tf/decode_training.py class Board (line 277) | class Board: method __init__ (line 278) | def __init__(self): method clear_board (line 281) | def clear_board(self): method describe (line 287) | def describe(self): class TrainingStep (line 295) | class TrainingStep: method __init__ (line 296) | def __init__(self, version): method init_structs (line 329) | def init_structs(self): method init_move_map (line 333) | def init_move_map(self): method clear_hist (line 347) | def clear_hist(self): method update_board (line 351) | def update_board(self, hist, piece, bit_board): method describe (line 363) | def describe(self): method update_reals (line 412) | def update_reals(self, text_item): method flip_single_v1_plane (line 423) | def flip_single_v1_plane(self, plane): method display_v4 (line 429) | def display_v4(self, ply, content): function main (line 467) | def main(args): FILE: tf/lc0_az_policy_map.py function index_to_position (line 14) | def index_to_position(x): function position_to_index (line 18) | def position_to_index(p): function valid_index (line 22) | def valid_index(i): function queen_move (line 30) | def queen_move(start, direction, steps): function knight_move (line 49) | def knight_move(start, direction, steps): function make_map (line 68) | def make_map(kind='matrix'): FILE: tf/net.py function nested_getattr (line 20) | def nested_getattr(obj, attr): class Net (line 27) | class Net: method __init__ (line 29) | def __init__(self, method set_networkformat (line 57) | def set_networkformat(self, net): method set_policyformat (line 63) | def set_policyformat(self, policy): method set_headcount (line 66) | def set_headcount(self, headcount): method set_pol_headcount (line 69) | def set_pol_headcount(self, headcount): method set_valueformat (line 72) | def set_valueformat(self, value): method set_movesleftformat (line 81) | def set_movesleftformat(self, moves_left): method set_input (line 84) | def set_input(self, input_format): method set_defaultactivation (line 94) | def set_defaultactivation(self, activation): method set_smolgen_activation (line 100) | def set_smolgen_activation(self, activation): method set_ffn_activation (line 106) | def set_ffn_activation(self, activation): method activation (line 112) | def activation(self, name): method get_weight_amounts (line 134) | def get_weight_amounts(self): method fill_layer_v2 (line 146) | def fill_layer_v2(self, layer, params): method fill_layer (line 161) | def fill_layer(self, layer, weights): method fill_conv_block (line 176) | def fill_conv_block(self, convblock, weights, gammas): method fill_plain_conv (line 190) | def fill_plain_conv(self, convblock, weights): method fill_se_unit (line 195) | def fill_se_unit(self, se_unit, weights): method denorm_layer_v2 (line 201) | def denorm_layer_v2(self, layer): method denorm_layer (line 207) | def denorm_layer(self, layer, weights): method denorm_conv_block (line 210) | def denorm_conv_block(self, convblock, weights): method denorm_plain_conv (line 226) | def denorm_plain_conv(self, convblock, weights): method denorm_se_unit (line 231) | def denorm_se_unit(self, convblock, weights): method save_txt (line 242) | def save_txt(self, filename): method save_proto (line 267) | def save_proto(self, filename): method tf_name_to_pb_name (line 279) | def tf_name_to_pb_name(self, name): method get_weights_v2 (line 492) | def get_weights_v2(self, names): method get_weights (line 536) | def get_weights(self): method filters (line 564) | def filters(self): method blocks (line 569) | def blocks(self): method print_stats (line 572) | def print_stats(self): method parse_proto (line 578) | def parse_proto(self, filename): method parse_txt (line 595) | def parse_txt(self, filename): method fill_net_v2 (line 615) | def fill_net_v2(self, all_weights): method fill_net (line 701) | def fill_net(self, weights): function print_pb_stats (line 744) | def print_pb_stats(obj, parent=None): function main (line 761) | def main(argv): FILE: tf/shufflebuffer.py class ShuffleBuffer (line 23) | class ShuffleBuffer: method __init__ (line 24) | def __init__(self, elem_size, elem_count): method extract (line 42) | def extract(self): method insert_or_replace (line 56) | def insert_or_replace(self, item): class ShuffleBufferTest (line 83) | class ShuffleBufferTest(unittest.TestCase): method test_extract (line 84) | def test_extract(self): method test_wrong_size (line 95) | def test_wrong_size(self): method test_insert_or_replace (line 103) | def test_insert_or_replace(self): FILE: tf/tfprocess.py function square_relu (line 34) | def square_relu(x): class Gating (line 38) | class Gating(tf.keras.layers.Layer): method __init__ (line 40) | def __init__(self, name=None, additive=True, init_value=None, **kwargs): method build (line 47) | def build(self, input_shape): method call (line 56) | def call(self, inputs): function ma_gating (line 61) | def ma_gating(inputs, name): class ApplySqueezeExcitation (line 67) | class ApplySqueezeExcitation(tf.keras.layers.Layer): method __init__ (line 69) | def __init__(self, **kwargs): method build (line 72) | def build(self, input_dimens): method call (line 75) | def call(self, inputs): class ApplyPolicyMap (line 85) | class ApplyPolicyMap(tf.keras.layers.Layer): method __init__ (line 87) | def __init__(self, **kwargs): method call (line 91) | def call(self, inputs): class ApplyAttentionPolicyMap (line 97) | class ApplyAttentionPolicyMap(tf.keras.layers.Layer): method __init__ (line 99) | def __init__(self, **kwargs): method call (line 103) | def call(self, logits, pp_logits): class Metric (line 112) | class Metric: method __init__ (line 114) | def __init__(self, short_name, long_name, suffix='', **kwargs): method assign (line 121) | def assign(self, value): method accumulate (line 125) | def accumulate(self, value): method merge (line 132) | def merge(self, other): method get (line 137) | def get(self): method reset (line 142) | def reset(self): class TFProcess (line 147) | class TFProcess: method __init__ (line 149) | def __init__(self, cfg): method init (line 355) | def init(self, train_dataset, test_dataset, validation_dataset=None): method init_net (line 380) | def init_net(self): method replace_weights (line 631) | def replace_weights(self, proto_filename, ignore_errors=False): method restore (line 718) | def restore(self): method process_loop (line 723) | def process_loop(self, batch_size, test_batches, batch_splits=1): method read_weights (line 747) | def read_weights(self): method process_inner_loop (line 751) | def process_inner_loop(self, x, y, z, q, m): method strategy_process_inner_loop (line 790) | def strategy_process_inner_loop(self, x, y, z, q, m): method apply_grads (line 799) | def apply_grads(self, grads, effective_batch_splits): method strategy_apply_grads (line 815) | def strategy_apply_grads(self, grads, effective_batch_splits): method merge_grads (line 824) | def merge_grads(self, grads, new_grads): method strategy_merge_grads (line 828) | def strategy_merge_grads(self, grads, new_grads): method train_step (line 831) | def train_step(self, steps, batch_size, batch_splits): method process (line 914) | def process(self, batch_size, test_batches, batch_splits): method calculate_swa_summaries (line 992) | def calculate_swa_summaries(self, test_batches, steps): method calculate_test_summaries_inner_loop (line 1004) | def calculate_test_summaries_inner_loop(self, x, y, z, q, m): method strategy_calculate_test_summaries_inner_loop (line 1041) | def strategy_calculate_test_summaries_inner_loop(self, x, y, z, q, m): method calculate_test_summaries (line 1050) | def calculate_test_summaries(self, test_batches, steps): method calculate_swa_validations (line 1082) | def calculate_swa_validations(self, steps): method calculate_test_validations (line 1093) | def calculate_test_validations(self, steps): method compute_update_ratio (line 1118) | def compute_update_ratio(self, before_weights, after_weights, steps): method update_swa (line 1147) | def update_swa(self): method save_swa_weights (line 1154) | def save_swa_weights(self, filename): method save_leelaz_weights (line 1162) | def save_leelaz_weights(self, filename): method batch_norm (line 1169) | def batch_norm(self, input, name, scale=False): method squeeze_excitation (line 1195) | def squeeze_excitation(self, inputs, channels, name): method conv_block (line 1211) | def conv_block(self, method residual_block (line 1228) | def residual_block(self, inputs, channels, name): method split_heads (line 1257) | def split_heads(inputs, batch_size: int, num_heads: int, depth: int): method scaled_dot_product_attention (line 1264) | def scaled_dot_product_attention(self, method mha (line 1294) | def mha(self, inputs, emb_size: int, d_model: int, num_heads: int, method ffn (line 1332) | def ffn(self, inputs, emb_size: int, dff: int, initializer, name: str): method encoder_layer (line 1347) | def encoder_layer(self, inputs, emb_size: int, d_model: int, method smolgen_weights (line 1389) | def smolgen_weights(self, method create_residual_body (line 1417) | def create_residual_body(self, inputs): method create_encoder_body (line 1429) | def create_encoder_body(self, inputs, embedding_size): method apply_promotion_logits (line 1470) | def apply_promotion_logits(self, queries, keys, attn_wts): method construct_net (line 1523) | def construct_net(self, inputs, name=''): FILE: tf/train.py function get_chunks (line 32) | def get_chunks(data_prefix): function get_all_chunks (line 36) | def get_all_chunks(path): function get_latest_chunks (line 50) | def get_latest_chunks(path, num_chunks, allow_less, sort_key_fn): function identity_function (line 77) | def identity_function(name): function game_number_for_name (line 81) | def game_number_for_name(name): function get_input_mode (line 87) | def get_input_mode(cfg): function main (line 109) | def main(cmd): FILE: tf/update_steps.py function main (line 12) | def main(cmd):