Home
last modified time | relevance | path

Searched refs:TransposePermsVec (Results 1 – 3 of 3) sorted by relevance

/external/tensorflow/tensorflow/core/kernels/
Dtranspose_functor.h71 typedef gtl::InlinedVector<int32, 8> TransposePermsVec; typedef
80 TransposePermsVec* new_perm, in ReduceTransposeDimensions()
90 TransposePermsVec new_dim_position(shape.dims(), -1); in ReduceTransposeDimensions()
242 TransposePermsVec perm(ndims); in DoMatrixTransposeImpl()
Dtranspose_functor_gpu.cu.cc100 TransposePermsVec new_perm; in run()
122 if (new_perm == TransposePermsVec({0, 2, 1})) { in run()
127 } else if (new_perm == TransposePermsVec({2, 1, 0})) { in run()
Dtranspose_util_test.cc29 internal::TransposePermsVec new_perm; in TestDimensionReduction()