Searched refs:BiasOp (Results 1 – 2 of 2) sorted by relevance
29 class BiasOp : public XlaOpKernel { class31 explicit BiasOp(OpKernelConstruction* ctx) : XlaOpKernel(ctx) { in BiasOp() function in tensorflow::__anon2f669e060111::BiasOp75 REGISTER_XLA_OP(Name("BiasAdd"), BiasOp);76 REGISTER_XLA_OP(Name("BiasAddV1"), BiasOp);
87 class BiasOp : public BinaryOp<T> { class89 explicit BiasOp(OpKernelConstruction* context) : BinaryOp<T>(context) { in BiasOp() function in tensorflow::BiasOp208 BiasOp<CPUDevice, type>); \211 BiasOp<CPUDevice, type>);297 class BiasOp<GPUDevice, T> : public BinaryOp<T> { class300 explicit BiasOp(OpKernelConstruction* context) : BinaryOp<T>(context) { in BiasOp() function in tensorflow::BiasOp348 BiasOp<GPUDevice, type>); \351 BiasOp<GPUDevice, type>);