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

  /external/tensorflow/tensorflow/core/kernels/
transpose_op.h 56 class TransposeGpuOp : public TransposeOp {
58 explicit TransposeGpuOp(OpKernelConstruction* ctx) : TransposeOp(ctx) {}
transpose_op.cc 254 Status TransposeGpuOp::DoTranspose(OpKernelContext* ctx, const Tensor& in,
274 TransposeGpuOp); \

Completed in 67 milliseconds