OpenGrok
Home
Sort by relevance
Sort by last modified time
Full Search
Definition
Symbol
File Path
History
|
|
Help
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