Searched refs:proto_values (Results 1 – 1 of 1) sorted by relevance
46 def SlowAppendFloat16ArrayToTensorProto(tensor_proto, proto_values): argument48 [ExtractBitsFromFloat16(x) for x in proto_values])51 def _MediumAppendFloat16ArrayToTensorProto(tensor_proto, proto_values): argument55 np.asarray(proto_values, dtype=np.float16).view(np.uint16))63 def SlowAppendBFloat16ArrayToTensorProto(tensor_proto, proto_values): argument65 [ExtractBitsFromBFloat16(x) for x in proto_values])68 def FastAppendBFloat16ArrayToTensorProto(tensor_proto, proto_values): argument71 proto_values, dtype=dtypes.bfloat16.as_numpy_dtype).view(np.uint16))122 def SlowAppendFloat32ArrayToTensorProto(tensor_proto, proto_values): argument123 tensor_proto.float_val.extend([x.item() for x in proto_values])[all …]