Searched refs:transpose_fn (Results 1 – 2 of 2) sorted by relevance
| /external/tensorflow/tensorflow/core/kernels/ |
| D | transpose_functor_cpu.cc | 42 auto transpose_fn = [=, &in_strides, &out_strides, &perm](int64 begin, in TransposeSimple() local
65 device.parallelFor(in.NumElements(), cost, std::move(transpose_fn)); in TransposeSimple()
|
| /external/tensorflow/tensorflow/python/ops/ |
| D | array_ops.py | 1676 transpose_fn = (
1687 ret = transpose_fn(a, perm, name=name)
1695 ret = transpose_fn(a, perm, name=name)
|