Searched refs:tensor_descr (Results 1 – 1 of 1) sorted by relevance
1042 TensorDescription* tensor_descr = no->mutable_tensor_description(); in GenerateRunMetadata() local1043 tensor_descr->set_dtype(properties.dtype()); in GenerateRunMetadata()1044 *tensor_descr->mutable_shape() = properties.shape(); in GenerateRunMetadata()1049 tensor_descr->mutable_allocation_description()->set_requested_bytes( in GenerateRunMetadata()1051 tensor_descr->mutable_allocation_description()->set_allocated_bytes( in GenerateRunMetadata()