Home
last modified time | relevance | path

Searched defs:InvertPermutationOp (Results 1 – 2 of 2) sorted by relevance

/aosp_15_r20/external/tensorflow/tensorflow/compiler/tf2xla/kernels/
H A Dtranspose_op.cc121 class InvertPermutationOp : public XlaOpKernel { class
123 explicit InvertPermutationOp(OpKernelConstruction* ctx) : XlaOpKernel(ctx) {} in InvertPermutationOp() function in tensorflow::__anon5623cd890111::InvertPermutationOp
/aosp_15_r20/external/tensorflow/tensorflow/core/kernels/
H A Dtranspose_op.cc42 class InvertPermutationOp : public OpKernel { class
44 explicit InvertPermutationOp(OpKernelConstruction* context) in InvertPermutationOp() function in tensorflow::InvertPermutationOp