Searched refs:ssm_shape (Results 1 – 2 of 2) sorted by relevance
115 TensorShape ssm_shape(ssm.shape()); in Add()116 if (!shape.IsSameSize(ssm_shape)) { in Add()118 "Mismatching shapes: existing tensor = ", ssm_shape.DebugString(), in Add()
171 TensorShape ssm_shape(ssm.shape()); in LoadShard() local174 status_ = RegisterTensorSlice(ssm.name(), ssm_shape, ssm.type(), fname, in LoadShard()