Home | History | Annotate | Download | only in util

Lines Matching defs:ssm

113     const SavedSliceMeta& ssm = sts_.meta().tensor(index);
114 CHECK_EQ(name, ssm.name()) << ProtoShortDebugString(ssm);
115 TensorShape ssm_shape(ssm.shape());
121 if (dt != ssm.type()) {
123 "Mismatching types: existing type = ", DataTypeString(ssm.type()),
130 SavedSliceMeta* ssm = sts_.mutable_meta()->add_tensor();
131 ssm->set_name(name);
132 shape.AsProto(ssm->mutable_shape());
133 ssm->set_type(dt);
136 SavedSliceMeta* ssm = sts_.mutable_meta()->mutable_tensor(index);
137 slice.AsProto(ssm->add_slice());
145 TensorShape saved_shape(ssm->shape());