HomeSort by relevance Sort by last modified time
    Searched refs:InvertPermutationOp (Results 1 - 2 of 2) sorted by null

  /external/tensorflow/tensorflow/compiler/tf2xla/kernels/
transpose_op.cc 96 // inv = InvertPermutationOp(T<int32> p) takes a permutation of
103 class InvertPermutationOp : public XlaOpKernel {
105 explicit InvertPermutationOp(OpKernelConstruction* ctx) : XlaOpKernel(ctx) {}
137 InvertPermutationOp);
  /external/tensorflow/tensorflow/core/kernels/
transpose_op.cc 34 // inv = InvertPermutationOp(T<int32/int64> p) takes a permutation of
42 class InvertPermutationOp : public OpKernel {
44 explicit InvertPermutationOp(OpKernelConstruction* context)
76 InvertPermutationOp<int32>);
79 InvertPermutationOp<int64>);
86 InvertPermutationOp<int32>);
92 InvertPermutationOp<int64>);
100 InvertPermutationOp<int32>);
106 InvertPermutationOp<int64>);

Completed in 89 milliseconds