Searched refs:UnaryVariantOp (Results 1 – 2 of 2) sorted by relevance
23 REGISTER_VARIANT(UnaryVariantOp, CPU, "Conj", CONJ_VARIANT_UNARY_OP);28 UnaryVariantOp<GPUDevice, CONJ_VARIANT_UNARY_OP>);
316 class UnaryVariantOp : public OpKernel {318 explicit UnaryVariantOp(OpKernelConstruction* ctx) : OpKernel(ctx) {} in UnaryVariantOp() function