Searched refs:tensor_meta (Results 1 – 2 of 2) sorted by relevance
148 protobuf::io::CodedInputStream* input, TensorProto* tensor_meta) { in ParseTensorSubmessage() argument158 TensorShape shape(tensor_meta->tensor_shape()); in ParseTensorSubmessage()159 Tensor t(allocator_, tensor_meta->dtype(), shape); in ParseTensorSubmessage()169 tensor_meta->set_dtype(static_cast<DataType>(static_cast<int>(v))); in ParseTensorSubmessage()170 if (!DataTypeCanUseMemcpy(tensor_meta->dtype())) return false; in ParseTensorSubmessage()175 !ReadNestedMessage(input, tensor_meta->mutable_tensor_shape())) in ParseTensorSubmessage()184 tensor_meta->set_version_number(static_cast<int32>(v)); in ParseTensorSubmessage()192 !tensor_meta->has_tensor_shape()) { in ParseTensorSubmessage()198 TensorShape shape(tensor_meta->tensor_shape()); in ParseTensorSubmessage()199 Tensor t(allocator_, tensor_meta->dtype(), shape); in ParseTensorSubmessage()
96 TensorProto* tensor_meta);