Searched refs:InvertPermutationOp (Results 1 - 2 of 2) sorted by relevance

/external/tensorflow/tensorflow/compiler/tf2xla/kernels/
H A Dtranspose_op.cc96 // inv = InvertPermutationOp(T<int32> p) takes a permutation of
103 class InvertPermutationOp : public XlaOpKernel { class in namespace:tensorflow::__anon25468
105 explicit InvertPermutationOp(OpKernelConstruction* ctx) : XlaOpKernel(ctx) {} function in class:tensorflow::__anon25468::InvertPermutationOp
137 InvertPermutationOp); member in namespace:tensorflow::__anon25468
/external/tensorflow/tensorflow/core/kernels/
H A Dtranspose_op.cc34 // inv = InvertPermutationOp(T<int32/int64> p) takes a permutation of
42 class InvertPermutationOp : public OpKernel { class in namespace:tensorflow
44 explicit InvertPermutationOp(OpKernelConstruction* context) function in class:tensorflow::InvertPermutationOp
76 InvertPermutationOp<int32>); member in namespace:tensorflow
79 InvertPermutationOp<int64>); member in namespace:tensorflow
86 InvertPermutationOp<int32>); member in namespace:tensorflow
92 InvertPermutationOp<int64>); member in namespace:tensorflow
100 InvertPermutationOp<int32>); member in namespace:tensorflow
106 InvertPermutationOp<int64>); member in namespace:tensorflow

Completed in 287 milliseconds