Searched refs:reshaped_tensor (Results 1 – 3 of 3) sorted by relevance
| /external/pytorch/torch/_inductor/ |
| D | decomposition.py | 424 reshaped_tensor = tensor.view(new_shape)
425 return reshaped_tensor
|
| /external/tensorflow/tensorflow/python/client/ |
| D | session_test.py | 1737 reshaped_tensor = array_ops.reshape(some_tensor, new_shape)
1740 sess.run(reshaped_tensor, feed_dict={some_tensor: [1.0, 2.0, 3.0]})
1746 sess.run(reshaped_tensor, feed_dict={new_shape: [3, 7]})
|
| /external/tensorflow/tensorflow/python/ops/ |
| D | array_ops.py | 1873 def _apply_mask_1d(reshaped_tensor, mask, axis=None): argument
1876 return gather(reshaped_tensor, indices, axis=axis)
|