Searched defs:ssm (Results 1 – 4 of 4) sorted by relevance
114 const SavedSliceMeta& ssm = sts_.meta().tensor(index); in Add() local131 SavedSliceMeta* ssm = sts_.mutable_meta()->add_tensor(); in Add() local137 SavedSliceMeta* ssm = sts_.mutable_meta()->mutable_tensor(index); in Add() local
171 const SavedSliceMeta& ssm = sts.meta().tensor(0); in CheckEntries() local189 const SavedSliceMeta& ssm = sts.meta().tensor(1); in CheckEntries() local205 const SavedSliceMeta& ssm = sts.meta().tensor(2); in CheckEntries() local221 const SavedSliceMeta& ssm = sts.meta().tensor(3); in CheckEntries() local
171 for (const SavedSliceMeta& ssm : sts.meta().tensor()) { in LoadShard() local
98 vk::ShaderModuleMap ssm; in step() local