Merge pull request #45610 from TomWildenhain-Microsoft:patch-1
PiperOrigin-RevId: 356317758 Change-Id: Ie1e790c288c5a3200fde3d8a6483b0e287af2888
This commit is contained in:
commit
879fd3bc67
@ -823,7 +823,7 @@ class _FunctionConverterData(_ConverterData):
|
|||||||
if idx in map_index_to_variable:
|
if idx in map_index_to_variable:
|
||||||
data = map_index_to_variable[idx].numpy()
|
data = map_index_to_variable[idx].numpy()
|
||||||
else:
|
else:
|
||||||
data = val_tensor.numpy()
|
data = np.array(val_tensor.numpy())
|
||||||
self._tensor_data[tensor_name] = _TensorData(
|
self._tensor_data[tensor_name] = _TensorData(
|
||||||
numpy=data,
|
numpy=data,
|
||||||
dtype=dtypes.as_dtype(data.dtype).as_datatype_enum,
|
dtype=dtypes.as_dtype(data.dtype).as_datatype_enum,
|
||||||
|
Loading…
Reference in New Issue
Block a user