Searched refs:ResizeNearestNeighborOp (Results 1 – 7 of 7) sorted by relevance
23 class ResizeNearestNeighborOp : public XlaOpKernel {25 explicit ResizeNearestNeighborOp(OpKernelConstruction* ctx);
584 ResizeNearestNeighborOp::ResizeNearestNeighborOp(OpKernelConstruction* ctx) in ResizeNearestNeighborOp() function in tensorflow::ResizeNearestNeighborOp598 void ResizeNearestNeighborOp::Compile(XlaOpKernelContext* ctx) { in Compile()603 ResizeNearestNeighborOp);
39 class ResizeNearestNeighborOp : public OpKernel { class41 explicit ResizeNearestNeighborOp(OpKernelConstruction* context) in ResizeNearestNeighborOp() function in tensorflow::ResizeNearestNeighborOp347 ResizeNearestNeighborOp<CPUDevice, T>); \365 ResizeNearestNeighborOp<GPUDevice, T>); \
214 TypeID::get<TF::ResizeNearestNeighborOp>(), in IsOpAllowedTf2XlaFallback()
1281 : RewritePattern(ResizeNearestNeighborOp::getOperationName(), in LowerResizeNearestNeighbor()1297 auto op = cast<ResizeNearestNeighborOp>(src_op); in matchAndRewrite()
1680 auto tf_resize_op = cast<TF::ResizeNearestNeighborOp>(op); in matchAndRewrite()
2292 auto tfl_resize_op = cast<TFL::ResizeNearestNeighborOp>(op); in matchAndRewrite()