Searched refs:tensor_descr (Results 1 – 1 of 1) sorted by relevance
1218 TensorDescription* tensor_descr = no->mutable_tensor_description(); in GenerateRunMetadata() local1219 tensor_descr->set_dtype(properties.dtype()); in GenerateRunMetadata()1220 *tensor_descr->mutable_shape() = properties.shape(); in GenerateRunMetadata()1225 tensor_descr->mutable_allocation_description()->set_requested_bytes( in GenerateRunMetadata()1227 tensor_descr->mutable_allocation_description()->set_allocated_bytes( in GenerateRunMetadata()