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

  /external/tensorflow/tensorflow/compiler/tf2xla/kernels/
transpose_op.cc 33 class TransposeOp : public XlaOpKernel {
35 explicit TransposeOp(OpKernelConstruction* ctx) : XlaOpKernel(ctx) {}
92 REGISTER_XLA_OP(Name("Transpose").CompileTimeConstInput("perm"), TransposeOp);
  /external/tensorflow/tensorflow/core/kernels/
transpose_op.h 24 class TransposeOp : public OpKernel {
26 explicit TransposeOp(OpKernelConstruction* ctx) : OpKernel(ctx) {}
36 class TransposeCpuOp : public TransposeOp {
38 explicit TransposeCpuOp(OpKernelConstruction* ctx) : TransposeOp(ctx) {}
46 class MklTransposeCpuOp : public TransposeOp {
48 explicit MklTransposeCpuOp(OpKernelConstruction* ctx) : TransposeOp(ctx) {}
56 class TransposeGpuOp : public TransposeOp {
58 explicit TransposeGpuOp(OpKernelConstruction* ctx) : TransposeOp(ctx) {}
66 class TransposeSyclOp : public TransposeOp {
68 explicit TransposeSyclOp(OpKernelConstruction* ctx) : TransposeOp(ctx) {
    [all...]

Completed in 218 milliseconds