Searched refs:SqueezeOp (Results 1 – 3 of 3) sorted by relevance
425 REGISTER_KERNEL_BUILDER(Name("Squeeze").Device(DEVICE_CPU), SqueezeOp);431 SqueezeOp);444 SqueezeOp);451 SqueezeOp);461 SqueezeOp);
188 class SqueezeOp : public OpKernel {190 explicit SqueezeOp(OpKernelConstruction* ctx) : OpKernel(ctx) { in SqueezeOp() function
160 class SqueezeOp : public XlaOpKernel { class162 explicit SqueezeOp(OpKernelConstruction* ctx) : XlaOpKernel(ctx) { in SqueezeOp() function in tensorflow::__anon812f91ee0111::SqueezeOp219 REGISTER_XLA_OP(Name("Squeeze"), SqueezeOp);