Home
last modified time | relevance | path

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

/aosp_15_r20/external/tensorflow/tensorflow/compiler/tf2xla/kernels/
H A Ddepthtospace_op.cc26 class DepthToSpaceOp : public XlaOpKernel { class
28 explicit DepthToSpaceOp(OpKernelConstruction* ctx) : XlaOpKernel(ctx) { in DepthToSpaceOp() function in tensorflow::__anon44cd0dce0111::DepthToSpaceOp
/aosp_15_r20/external/tensorflow/tensorflow/core/kernels/
H A Ddepthtospace_op.cc44 class DepthToSpaceOp : public OpKernel { class
46 explicit DepthToSpaceOp(OpKernelConstruction* context) : OpKernel(context) { in DepthToSpaceOp() function in tensorflow::DepthToSpaceOp