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()
244 TransposePermsVec perm(ndims); in DoMatrixTransposeImpl()
Dtranspose_functor_gpu.cu.cc101 TransposePermsVec new_perm; in run()
123 if (new_perm == TransposePermsVec({0, 2, 1})) { in run()
128 } else if (new_perm == TransposePermsVec({2, 1, 0})) { in run()
Dtranspose_util_test.cc29 internal::TransposePermsVec new_perm; in TestDimensionReduction()