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)):
|