Searched refs:axis_order (Results 1 – 3 of 3) sorted by relevance
| /external/tensorflow/tensorflow/contrib/labeled_tensor/python/ops/ |
| D | core.py | 679 def transpose(labeled_tensor, axis_order=None, name=None): argument
700 if axis_order is None:
701 axis_order = list(reversed(original_order))
702 elif sorted(axis_order) != sorted(original_order):
706 (axis_order, original_order))
709 permutation = [axis_names.index(n) for n in axis_order]
715 permuted_axes = [labeled_tensor.axes[n] for n in axis_order]
791 axis_order, = axis_order_list
793 axis_order = None
794 return axis_order
[all …]
|
| D | core_test.py | 467 axis_order = ['w', 'x', 'y', 'z']
469 lt = core.LabeledTensor(array_ops.ones((1, 1, 1, 1)), axis_order)
470 core.check_axis_order(lt, axis_order)
472 lt = core.LabeledTensor(array_ops.ones((1, 1, 1)), axis_order[1:])
473 core.check_axis_order(lt, axis_order)
475 lt = core.LabeledTensor(array_ops.ones((1, 1, 1)), axis_order[:-1])
476 core.check_axis_order(lt, axis_order)
479 axis_order = ['w', 'x', 'y', 'z']
480 lt = core.LabeledTensor(array_ops.ones((1, 1, 1, 1)), axis_order)
484 core.check_axis_order(lt, axis_order[:-1])
[all …]
|
| /external/tensorflow/tensorflow/python/ops/ |
| D | special_math_ops.py | 492 axis_order = {}
495 axis_order[ax] = len(axis_order)
497 axis_order[ax] = len(axis_order)
509 sorted_idx = sorted(axes_, key=axis_order.get)
526 for j, ax in enumerate(sorted(idx_all, key=axis_order.get)):
|