Searched refs:_TransposeAndTest (Results 1 – 1 of 1) sorted by relevance
| /external/tensorflow/tensorflow/compiler/xla/python/ |
| D | xla_client_test.py | 856 def _TransposeAndTest(array): function
863 _TransposeAndTest(array_fun([[1, 2, 3], [4, 5, 6]]))
864 _TransposeAndTest(array_fun([[1, 2, 3], [4, 5, 6]], order="F"))
865 _TransposeAndTest(array_fun([[1, 2], [4, 5]]))
866 _TransposeAndTest(array_fun([[1, 2], [4, 5]], order="F"))
870 def _TransposeAndTest(array, permutation): function
876 _TransposeAndTest(NumpyArrayF32([[1, 2, 3], [4, 5, 6]]), [0, 1])
877 _TransposeAndTest(NumpyArrayF32([[1, 2, 3], [4, 5, 6]]), [1, 0])
878 _TransposeAndTest(NumpyArrayF32([[1, 2], [4, 5]]), [0, 1])
879 _TransposeAndTest(NumpyArrayF32([[1, 2], [4, 5]]), [1, 0])
[all …]
|