Searched defs:axis2 (Results 1 – 4 of 4) sorted by relevance
| /external/tensorflow/tensorflow/python/ops/numpy_ops/ |
| D | np_array_ops.py | 310 def diagonal(a, offset=0, axis1=0, axis2=1): # pylint: disable=missing-docstring argument
795 def swapaxes(a, axis1, axis2): # pylint: disable=missing-docstring argument
|
| D | np_math_ops.py | 1305 def trace(a, offset=0, axis1=0, axis2=1, dtype=None): # pylint: disable=missing-docstring argument
|
| /external/pytorch/torch/_numpy/ |
| D | _funcs_impl.py | 1028 def diagonal(a: ArrayLike, offset=0, axis1=0, axis2=1): argument
1038 axis2=1, argument
1422 def swapaxes(a: ArrayLike, axis1, axis2): argument
|
| /external/tensorflow/tensorflow/core/grappler/optimizers/ |
| D | constant_folding_test.cc | 2619 Output axis2 = ops::Const(scope.WithOpName("axis2"), 1, {}); in TEST_F() local
|