Searched refs:PlasmaToTensorOp (Results 1 – 3 of 3) sorted by relevance
238 class PlasmaToTensorOp : public tf::AsyncOpKernel { class240 explicit PlasmaToTensorOp(tf::OpKernelConstruction* context) : tf::AsyncOpKernel(context) { in PlasmaToTensorOp() function in PlasmaToTensorOp252 ~PlasmaToTensorOp() override { in ~PlasmaToTensorOp()387 PlasmaToTensorOp<CPUDevice>);390 PlasmaToTensorOp<GPUDevice>);