Searched refs:moveaxis (Results 1 – 6 of 6) sorted by relevance
| /external/tensorflow/tensorflow/python/kernel_tests/boosted_trees/ |
| D | stats_ops_test.py | 94 stats_summary = np.moveaxis(stats_summary, 0, 1)
98 return np.moveaxis(indices, 0, 1), values, stats_summary.shape
124 dense_summary = np.moveaxis(dense_summary, 0, 1)
133 summary_indices = np.moveaxis(summary_indices, 0, 1)
271 stats_summary = np.moveaxis(stats_summaries, 0, 1)
307 stats_summary = np.moveaxis(stats_summaries, 0, 1)
342 stats_summary = np.moveaxis(stats_summaries, 0, 1)
448 stats_summary = np.moveaxis(stats_summaries, 0, 1)
483 stats_summary = np.moveaxis(stats_summaries, 0, 1)
517 stats_summary = np.moveaxis(stats_summaries, 0, 1)
[all …]
|
| /external/tensorflow/tensorflow/python/keras/engine/ |
| D | training_gpu_test.py | 97 data = np.moveaxis(data_channels_first, 1, -1)
99 labels = np.moveaxis(labels_channels_first[index], 1, -1)
|
| /external/tensorflow/tensorflow/python/ops/numpy_ops/ |
| D | np_array_ops.py | 328 a = moveaxis(a, (axis1, axis2), (-2, -1))
830 def moveaxis(a, source, destination): # pylint: disable=missing-docstring function
1694 tensor = moveaxis(tensor, dims, range(len(dims)))
1719 updates = moveaxis(updates, range_(batch_start, batch_size),
1732 tensor = moveaxis(tensor, range(len(dims)), dims)
|
| D | np_array_ops_test.py | 1062 expected = np.moveaxis(*args)
1063 raw_ans = np_array_ops.moveaxis(*args)
|
| D | np_math_ops.py | 1119 res = np_array_ops.moveaxis(res, 0, axis)
|
| /external/tensorflow/third_party/py/numpy/tf_numpy_api/ |
| D | tensorflow.experimental.numpy.pbtxt | 636 name: "moveaxis"
|