Searched defs:axis1 (Results 1 – 6 of 6) 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
1037 axis1=0, argument
1422 def swapaxes(a: ArrayLike, axis1, axis2): argument
|
| /external/pytorch/aten/src/ATen/native/ |
| D | TensorShape.cpp | 3925 Tensor swapaxes(const Tensor& self, int64_t axis0, int64_t axis1) { in swapaxes()
3929 Tensor& swapaxes_(Tensor& self, int64_t axis0, int64_t axis1) { in swapaxes_()
|
| /external/tensorflow/tensorflow/core/grappler/optimizers/ |
| D | constant_folding_test.cc | 2618 Output axis1 = ops::Const(scope.WithOpName("axis1"), 0, {}); in TEST_F() local
|
| /external/pytorch/torch/csrc/jit/runtime/static/ |
| D | generated_ops.cpp | 5169 const auto axis1 = p_node->Input(2).toInt(); in __anon7ebb59391f602() local
|